Update logic generate mindmap, summary and nlp

app/jobs/enrichment_job.py

@@ -1,3 +1,26 @@
+# from app.services.storage import get_note, update_note
+# from app.services.summary_service import generate_summary
+# from app.services.mindmap_service import generate_mindmap
+
+# async def run_enrichment(note_id: str, tasks: list):
+#     note = get_note(note_id)
+#     if not note:
+#         return
+
+#     text = note.get("normalized_text") or note["raw_text"]
+
+#     update_note(note_id, status="processing")
+#     updates = {}
+
+#     if "summary" in tasks:
+#         updates["summary"] = await generate_summary(text)
+
+#     if "mindmap" in tasks:
+#         updates["mindmap"] = await generate_mindmap(text)
+
+#     update_note(note_id, data=updates, status="ready")
+
+import logging
 from app.services.storage import get_note, update_note
 from app.services.summary_service import generate_summary
 from app.services.mindmap_service import generate_mindmap
@@ -5,6 +28,7 @@ from app.services.mindmap_service import generate_mindmap
 async def run_enrichment(note_id: str, tasks: list):
     note = get_note(note_id)
     if not note:
+        logging.warning(f"[enrichment] Note not found: {note_id}")
         return
 
     text = note.get("normalized_text") or note["raw_text"]
@@ -12,10 +36,18 @@ async def run_enrichment(note_id: str, tasks: list):
     update_note(note_id, status="processing")
     updates = {}
 
-    if "summary" in tasks:
-        updates["summary"] = await generate_summary(text)
-
-    if "mindmap" in tasks:
-        updates["mindmap"] = await generate_mindmap(text)
+    try:
+        if "summary" in tasks:
+            try:
+                updates["summary"] = await generate_summary(text)
+            except Exception as e:
+                logging.exception(f"[enrichment] generate_summary failed for note_id={note_id}: {e}")
 
-    update_note(note_id, data=updates, status="ready")
+        if "mindmap" in tasks:
+            try:
+                updates["mindmap"] = await generate_mindmap(text)
+            except Exception as e:
+                logging.exception(f"[enrichment] generate_mindmap failed for note_id={note_id}: {e}")
+    finally:
+        # Dù thành công hay thất bại, vẫn set status=ready để client ngừng poll
+        update_note(note_id, data=updates, status="ready")

app/services/mindmap_service.py

@@ -1,56 +1,89 @@
-import asyncio, json
+# app/services/mindmap_service.py
+import asyncio
+import json
+import logging
+
 from app.config import GEMINI_API_KEY
-import google.generativeai as genai
+
+import google.genai as genai
+from google.api_core.exceptions import GoogleAPIError
+
+_MINDMAP_MODEL = "gemini-1.5-flash"
+_gemini_client = None
 
 if GEMINI_API_KEY:
-    genai.configure(api_key=GEMINI_API_KEY)
-    _model = genai.GenerativeModel("gemini-pro")
+    try:
+        _gemini_client = genai.Client(api_key=GEMINI_API_KEY)
+        logging.info(f"[mindmap_service] Initialized google.genai client with model={_MINDMAP_MODEL}")
+    except Exception as e:
+        logging.exception(f"[mindmap_service] Failed to init google.genai client: {e}")
+        _gemini_client = None
 else:
-    _model = None
+    logging.warning("[mindmap_service] GEMINI_API_KEY is not set, mindmap generation will be disabled")
 
 
 async def generate_mindmap(text: str) -> dict:
-    if not _model:
+    """
+    Sinh cấu trúc mindmap JSON từ văn bản.
+    Fallback: trả {} nếu không có model hoặc lỗi.
+    """
+    if not _gemini_client:
         return {}
 
     prompt = f"""
-Bạn là chuyên gia tạo Sơ đồ tư duy. Hãy phân tích văn bản sau và tạo cấu trúc JSON Mindmap.
+Bạn là chuyên gia tạo Sơ đồ tư duy. Hãy phân tích văn bản sau và tạo CẤU TRÚC JSON Mindmap.
+
 Yêu cầu:
 1. Xác định Ý chính làm Root.
 2. Phân tách ý phụ thành nhánh con (tối đa 3 cấp).
 3. Nhãn (label) ngắn gọn (< 7 từ).
-4. Màu sắc (colorHex): Root="#6200EE", Con="#F59E2B", "#2ECF9A", "#2F9BFF".
+4. Màu sắc (colorHex):
+   - Root: "#6200EE"
+   - Các nhánh con: sử dụng một trong các màu: "#F59E2B", "#2ECF9A", "#2F9BFF"
+5. CHỈ TRẢ VỀ JSON, không giải thích thêm.
 
-Cấu trúc JSON bắt buộc (Chỉ trả về JSON):
+Cấu trúc JSON bắt buộc:
 {{
-    "root": {{
-        "label": "Chủ đề",
-        "colorHex": "#6200EE",
-        "children": [
-            {{
-                "label": "Ý 1",
-                "colorHex": "#F59E2B",
-                "children": []
-            }}
-        ]
-    }}
+  "root": {{
+    "label": "Chủ đề",
+    "colorHex": "#6200EE",
+    "children": [
+      {{
+        "label": "Ý 1",
+        "colorHex": "#F59E2B",
+        "children": []
+      }}
+    ]
+  }}
 }}
 
 Văn bản:
-{text}
+\"\"\"{text}\"\"\"
 """
 
     loop = asyncio.get_event_loop()
 
     def call():
-        r = _model.generate_content(prompt)
-        return r.text
-
-    raw = await loop.run_in_executor(None, call)
+        resp = _gemini_client.models.generate_content(
+            model=_MINDMAP_MODEL,
+            contents=prompt,
+        )
+        return resp.text or ""
 
-    start = raw.find("{")
-    end = raw.rfind("}")
-    if start != -1 and end != -1:
-        return json.loads(raw[start:end+1])
+    try:
+        raw = await loop.run_in_executor(None, call)
+        start = raw.find("{")
+        end = raw.rfind("}")
+        if start != -1 and end != -1:
+            try:
+                return json.loads(raw[start:end + 1])
+            except Exception as e:
+                logging.warning(f"[mindmap_service] Failed to parse mindmap JSON: {e}")
+        else:
+            logging.warning("[mindmap_service] Mindmap response has no JSON block")
+    except GoogleAPIError as e:
+        logging.error(f"[mindmap_service] Gemini API error: {e}")
+    except Exception as e:
+        logging.exception(f"[mindmap_service] generate_mindmap failed: {e}")
 
     return {}

app/services/summary_service.py

@@ -1,35 +1,62 @@
 import asyncio
+import logging
+
 from app.config import GEMINI_API_KEY
-import google.generativeai as genai
+
+import google.genai as genai
+from google.api_core.exceptions import GoogleAPIError
+
+_SUMMARY_MODEL = "gemini-1.5-flash"
+_gemini_client = None
 
 if GEMINI_API_KEY:
-    genai.configure(api_key=GEMINI_API_KEY)
-    _model = genai.GenerativeModel("gemini-pro")
+    try:
+        _gemini_client = genai.Client(api_key=GEMINI_API_KEY)
+        logging.info(f"[summary_service] Initialized google.genai client with model={_SUMMARY_MODEL}")
+    except Exception as e:
+        logging.exception(f"[summary_service] Failed to init google.genai client: {e}")
+        _gemini_client = None
 else:
-    _model = None
+    logging.warning("[summary_service] GEMINI_API_KEY is not set, summary will be empty")
 
 
 async def generate_summary(text: str) -> str:
-    if not _model:
+    """
+    Tạo tóm tắt ngắn gọn 3-5 câu, một đoạn văn duy nhất.
+    Fallback: trả "" nếu không có model hoặc lỗi.
+    """
+    if not _gemini_client:
         return ""
 
     prompt = f"""
-    Bạn là chuyên gia tóm tắt. Hãy tóm tắt văn bản sau thành **một đoạn văn duy nhất**.
-    Yêu cầu:
-    1. Viết khoảng 3-5 câu, tổng hợp đầy đủ chủ đề và các ý chính.
-    2. Viết liền mạch, KHÔNG xuống dòng, KHÔNG dùng gạch đầu dòng hay đánh số.
-    3. Chỉ dựa trên thông tin được cung cấp, tuyệt đối KHÔNG tự thêm thông tin bên ngoài.
-    4. Trả về văn bản thuần (plain text).
-
-    Văn bản:
-    \"\"\"{text}\"\"\"
-    """
+Bạn là chuyên gia tóm tắt. Hãy tóm tắt văn bản sau thành một đoạn văn duy nhất.
+
+Yêu cầu:
+1. Viết khoảng 3-5 câu, tổng hợp đầy đủ chủ đề và các ý chính.
+2. Viết liền mạch, KHÔNG xuống dòng, KHÔNG dùng gạch đầu dòng hay đánh số.
+3. Chỉ dựa trên thông tin được cung cấp, tuyệt đối KHÔNG tự thêm thông tin bên ngoài.
+4. Trả về VĂN BẢN THUẦN (plain text), không bọc trong ``` hoặc JSON.
+
+Văn bản:
+\"\"\"{text}\"\"\"
+"""
 
     loop = asyncio.get_event_loop()
 
     def call():
-        r = _model.generate_content(prompt)
-        return r.text.strip()
-
-    result = await loop.run_in_executor(None, call)
-    return result.replace("```", "").strip()
+        resp = _gemini_client.models.generate_content(
+            model=_SUMMARY_MODEL,
+            contents=prompt,
+        )
+        return (resp.text or "").strip()
+
+    try:
+        result = await loop.run_in_executor(None, call)
+        # clean backticks nếu model có lỡ bọc
+        return result.replace("```", "").strip()
+    except GoogleAPIError as e:
+        logging.error(f"[summary_service] Gemini API error: {e}")
+    except Exception as e:
+        logging.exception(f"[summary_service] generate_summary failed: {e}")
+
+    return ""