__init__.py

import json
import os
import threading
from flask import Flask, request, jsonify
from aqt import mw, gui_hooks
from anki.notes import Note

app = Flask(__name__)


def create_note_type_if_not_exists():
    model_name = "Basic with videos in answers"
    model = mw.col.models.by_name(model_name)
    if not model:
        model = mw.col.models.new(model_name)

        # Add fields
        fields = ["Front", "videos", "backand"]
        for field_name in fields:
            fld = mw.col.models.new_field(field_name)
            mw.col.models.add_field(model, fld)

        # Add template
        tmpl = mw.col.models.new_template("Card 1")
        tmpl['qfmt'] = "{{Front}}"
        tmpl['afmt'] = "{{Front}}<hr id=answer>{{videos}}<br>{{backand}}"  # Simple template to show videos in answer
        mw.col.models.add_template(model, tmpl)

        mw.col.models.add(model)
        mw.col.models.save(model)
    return model


@app.route('/add_card', methods=['POST'])
def add_card():
    data = request.json
    deck_name = data.get('deck_name')
    word = data.get('word')
    part_of_speech = data.get('part_of_speech')
    variants = data.get('variants')
    tags = data.get('tags', [])  # Optional list of tags


    if not all([deck_name, word, variants]):
        return jsonify({"error": "Missing data"}), 400

    is_success = True
    error_msg = "[no error]"

    def perform_add():
        nonlocal is_success, error_msg
        #is_success = False
        #error_msg = f"error"
        #return
        # Create or get note type
        model = create_note_type_if_not_exists()

        # Get or create deck
        did = mw.col.decks.id(deck_name)

        # Create note
        note = Note(mw.col, model)
        count_str= f' <span style="color: green">({len(variants)})'  if len(variants)>1 else ""
        part_of_speech_str = f'<span style="color: gray">({part_of_speech})</span>' if part_of_speech!="other" else ""
        note["Front"] = f'<b>{word}</b> {part_of_speech_str}{count_str} '

        # Process videos: Add to media and build HTML for videos field
        videos_html = []
        for var in variants:
            file_name = var.get('file')
            if not file_name:
                is_success = False
                error_msg = f"Missing file name in variant"
                return
            if not mw.col.media.have(file_name):
                is_success = False
                error_msg = f"Video file '{file_name}' not found in Anki media collection"
                return

            # Add file to Anki media
            #media_name = os.path.basename(file_path)
            #mw.col.media.add_file(file_path)

            # Build <video> tag with start time
            #start = var.get('start', 0)
            video_tag = f'[sound:{file_name}]'
            videos_html.append(video_tag)

        note["videos"] = "".join(videos_html)

        def escape_quotes(obj):
            if isinstance(obj, str):
                return obj.replace('\\', r'\\').replace('"', r'\"')
            elif isinstance(obj, list):
                return [escape_quotes(v) for v in obj]
            elif isinstance(obj, dict):
                return {k: escape_quotes(v) for k, v in obj.items()}
            else:
                return obj
        # backand: JSON of variants
        note["Back"] = json.dumps(escape_quotes(variants))

        # Assign deck
        note.note_type()['did'] = did

        if tags and isinstance(tags, list):
            note.tags.extend(tags)

        # Add note to collection
        try:
            mw.col.add_note(note, did)
            mw.col.save()
        except Exception as e:
            is_success = False
            error_msg = f"Failed to add note: {str(e)}"

    # Run the addition on the main thread to avoid thread-safety issues
    perform_add()
    #mw.taskman.run_on_main(perform_add)

    if not is_success:
        return jsonify({"success": False, "error": error_msg}), 400 if "file name" in error_msg else 500

    return jsonify({"success": True}), 200


def run_server():
    app.run(port=5001, debug=True, use_reloader=False)


# Start the server in a background thread when the add-on loads
thread = threading.Thread(target=run_server, daemon=True)
thread.start()


# Optional: Hook to ensure note type is created on Anki startup
def on_profile_loaded():
    create_note_type_if_not_exists()


gui_hooks.profile_did_open.append(on_profile_loaded)