Fix bug early enrich again

app/api/transcribe.py

@@ -123,25 +123,28 @@ async def _run_sync_pipeline(tmp_wav: str, note_id: str):
 
 
 async def _create_placeholder_note(note_id: str, duration: float):
-    """
-    Tạo note NGAY LẬP TỨC để:
-    - SSE không trả not_found
-    - enrich có object để update
-    """
-    await NoteServiceClient().create_audio_note({
-    "note_id": note_id,
-    "type": "audio",
-    "status": "processing",
-    "raw_text": "",
-    "metadata": {
-        "audio": {
-            "duration": duration,
-            "chunks": [],
-            "asr_model": "PhoWhisper-base",
+    client = NoteServiceClient()
+    res = await client.create_audio_note(
+        {
+            "note_id": note_id,
+            "type": "audio",
+            "status": "processing",
+            "raw_text": "",
+            "metadata": {
+                "audio": {
+                    "duration": duration,
+                    "chunks": [],
+                    "asr_model": "PhoWhisper-base",
+                }
+            },
+            # ❌ KHÔNG generate ở đây
         }
-    },
-    # 🔥 KHÔNG generate ở đây
-})
+    )
+
+    if res is None:
+        # 🔥 FAIL FAST
+        raise RuntimeError("Failed to create placeholder note")
+
 
 
 # ============================================================

app/jobs/transcribe_job.py

@@ -2,11 +2,24 @@ import asyncio
 import tempfile
 import os
 import requests
+import httpx
+import time
 
 from app.core.asr_engine import load_model, transcribe_file, transcribe_file_chunks
 from app.services.note_client import NoteServiceClient
 from app.core.audio_utils import get_audio_info
 
+def run_async(coro):
+    try:
+        loop = asyncio.get_running_loop()
+    except RuntimeError:
+        loop = None
+
+    if loop and loop.is_running():
+        return asyncio.create_task(coro)
+    else:
+        return asyncio.run(coro)
+
 
 def download_audio(audio_url: str) -> str:
     r = requests.get(audio_url, timeout=30)
@@ -62,18 +75,32 @@ def transcribe_job(audio_url: str, note_id: str, user_id: str | None = None):
             "generate": ["normalize", "keywords", "summary", "mindmap"],
         }
 
-        client = NoteServiceClient()
-        asyncio.run(
-            client.update_note(
-                note_id,
-                {
-                    "status": note_status,
-                    "raw_text": text,
-                    "metadata": payload["metadata"],
-                },
-            )
+        generate_tasks = (
+            ["normalize", "keywords", "summary", "mindmap"]
+            if note_status == "transcribed"
+            else []
         )
 
+        time.sleep(1.0)
+        
+        client = NoteServiceClient()
+        try:
+            run_async(
+                client.update_note(
+                    note_id,
+                    {
+                        "status": note_status,
+                        "raw_text": text,
+                        "metadata": payload["metadata"],
+                        **({"generate": generate_tasks} if generate_tasks else {}),
+                    },
+                )
+            )
+        except httpx.HTTPStatusError as e:
+            if e.response.status_code == 404:
+                run_async(client.create_audio_note(payload))
+            else:
+                raise
 
     finally:
         if wav_path and os.path.exists(wav_path):