fix error 503

app/services/mindmap_service.py

@@ -1,33 +1,47 @@
-# app/services/mindmap_service.py
 import asyncio
 import json
 import logging
+import random
+import re
+import time
 
-from app.config import GEMINI_API_KEY, GEMINI_MODEL
+from app.config import GEMINI_API_KEY
 
-import google.genai as genai
-from google.api_core.exceptions import GoogleAPIError
+try:
+    import google.genai as genai
+    try:
+        from google.genai import errors as genai_errors
+    except Exception:
+        genai_errors = None
+except Exception:
+    genai = None
+    genai_errors = None
+    logging.warning("[mindmap_service] google.genai module not found; mindmap generation disabled")
+
+try:
+    from google.api_core.exceptions import GoogleAPIError
+except Exception:
+    GoogleAPIError = Exception
 
-_MINDMAP_MODEL = GEMINI_MODEL
+_MINDMAP_MODEL = "gemini-2.5-flash"
 _gemini_client = None
 
-if not GEMINI_API_KEY:
-    logging.warning("[mindmap_service] GEMINI_API_KEY is not set; mindmap generation disabled")
-elif not _MINDMAP_MODEL:
-    logging.warning("[mindmap_service] GEMINI_MODEL is not set; mindmap generation disabled")
+if not genai:
+    logging.warning("[mindmap_service] google.genai not available, mindmap generation will be disabled")
+elif not GEMINI_API_KEY:
+    logging.warning("[mindmap_service] GEMINI_API_KEY is not set, mindmap generation will be disabled")
 else:
     try:
         _gemini_client = genai.Client(api_key=GEMINI_API_KEY)
-        logging.info(f"[mindmap_service] Initialized google.genai client; model={_MINDMAP_MODEL}")
+        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
 
 
 async def generate_mindmap(text: str) -> dict:
-    """
-    Sinh cấu trúc mindmap JSON từ văn bản.
-    Fallback: trả {} nếu không có model hoặc lỗi.
+    """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 {}
@@ -58,19 +72,47 @@ Cấu trúc JSON bắt buộc:
     ]
   }}
 }}
-
 Văn bản:
 \"\"\"{text}\"\"\"
 """
 
     loop = asyncio.get_event_loop()
 
+    MAX_RETRIES = 3
+    BASE_DELAY = 1.0
+
     def call():
-        resp = _gemini_client.models.generate_content(
-            model=_MINDMAP_MODEL,
-            contents=prompt,
-        )
-        return resp.text or ""
+        last_exc = None
+        for attempt in range(1, MAX_RETRIES + 1):
+            try:
+                resp = _gemini_client.models.generate_content(
+                    model=_MINDMAP_MODEL,
+                    contents=prompt,
+                )
+                return resp.text or ""
+            except Exception as e:
+                last_exc = e
+                is_server_error = False
+                try:
+                    if genai_errors and isinstance(e, genai_errors.ServerError):
+                        is_server_error = True
+                except Exception:
+                    is_server_error = False
+
+                msg = str(e)
+                if "503" in msg or "UNAVAILABLE" in msg or is_server_error:
+                    if attempt < MAX_RETRIES:
+                        delay = BASE_DELAY * (2 ** (attempt - 1))
+                        delay = delay + random.uniform(0, 0.5 * delay)
+                        logging.warning(f"[mindmap_service] model overloaded (attempt {attempt}/{MAX_RETRIES}), retrying after {delay:.2f}s")
+                        time.sleep(delay)
+                        continue
+                logging.exception(f"[mindmap_service] generate_mindmap call failed on attempt {attempt}: {e}")
+                break
+
+        if last_exc:
+            raise last_exc
+        return ""
 
     try:
         raw = await loop.run_in_executor(None, call)
@@ -88,4 +130,34 @@ Văn bản:
     except Exception as e:
         logging.exception(f"[mindmap_service] generate_mindmap failed: {e}")
 
-    return {}
+    # fallback: build a minimal mindmap from the text (best-effort)
+    try:
+        # use first sentence as root label (shortened)
+        sentences = re.split(r"(?<=[.!?])\s+", text.strip())
+        root_label = sentences[0] if sentences else "Topic"
+        # shorten to ~8 words
+        root_label = " ".join(root_label.split()[:8]).strip()
+
+        children = []
+        # take up to 3 next sentences as child labels
+        for s in sentences[1:4]:
+            label = " ".join(s.split()[:6]).strip()
+            if label:
+                children.append({
+                    "label": label,
+                    "colorHex": "#F59E2B",
+                    "children": []
+                })
+
+        fallback = {
+            "root": {
+                "label": root_label or "Topic",
+                "colorHex": "#6200EE",
+                "children": children,
+            }
+        }
+        logging.info("[mindmap_service] Returning fallback mindmap after errors")
+        return fallback
+    except Exception:
+        return {}
+

app/services/summary_service.py

@@ -1,37 +1,45 @@
 import asyncio
 import logging
+import random
+import re
+import time
 
-from app.config import GEMINI_API_KEY, GEMINI_MODEL
+from app.config import GEMINI_API_KEY
 
 try:
     import google.genai as genai
+    try:
+        from google.genai import errors as genai_errors
+    except Exception:
+        genai_errors = None
 except Exception:
     genai = None
+    genai_errors = None
     logging.warning("[summary_service] google.genai module not found; summaries disabled")
 
-from google.api_core.exceptions import GoogleAPIError
+try:
+    from google.api_core.exceptions import GoogleAPIError
+except Exception:
+    GoogleAPIError = Exception
 
-_SUMMARY_MODEL = GEMINI_MODEL
+_SUMMARY_MODEL = "gemini-2.5-flash"
 _gemini_client = None
 
 if not genai:
-    logging.warning("[summary_service] google.genai not available; summaries disabled")
+    logging.warning("[summary_service] google.genai not available, summary will be empty")
 elif not GEMINI_API_KEY:
-    logging.warning("[summary_service] GEMINI_API_KEY is not set; summaries disabled")
-elif not _SUMMARY_MODEL:
-    logging.warning("[summary_service] GEMINI_MODEL is not set; summaries disabled")
+    logging.warning("[summary_service] GEMINI_API_KEY is not set, summary will be empty")
 else:
     try:
         _gemini_client = genai.Client(api_key=GEMINI_API_KEY)
-        logging.info(f"[summary_service] Initialized google.genai client; model={_SUMMARY_MODEL}")
+        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
 
 
 async def generate_summary(text: str) -> str:
-    """
-    Tạo tóm tắt ngắn gọn 3-5 câu, một đoạn văn duy nhất.
+    """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:
@@ -52,20 +60,65 @@ Văn bản:
 
     loop = asyncio.get_event_loop()
 
+    MAX_RETRIES = 3
+    BASE_DELAY = 1.0
+
     def call():
-        resp = _gemini_client.models.generate_content(
-            model=_SUMMARY_MODEL,
-            contents=prompt,
-        )
-        return (resp.text or "").strip()
+        last_exc = None
+        for attempt in range(1, MAX_RETRIES + 1):
+            try:
+                resp = _gemini_client.models.generate_content(
+                    model=_SUMMARY_MODEL,
+                    contents=prompt,
+                )
+                return (resp.text or "").strip()
+            except Exception as e:
+                last_exc = e
+                is_server_error = False
+                try:
+                    if genai_errors and isinstance(e, genai_errors.ServerError):
+                        is_server_error = True
+                except Exception:
+                    is_server_error = False
+
+                msg = str(e)
+                if "503" in msg or "UNAVAILABLE" in msg or is_server_error:
+                    if attempt < MAX_RETRIES:
+                        delay = BASE_DELAY * (2 ** (attempt - 1))
+                        # add jitter
+                        delay = delay + random.uniform(0, 0.5 * delay)
+                        logging.warning(f"[summary_service] model overloaded (attempt {attempt}/{MAX_RETRIES}), retrying after {delay:.2f}s")
+                        time.sleep(delay)
+                        continue
+                # non-retryable or out of retries
+                logging.exception(f"[summary_service] generate_summary call failed on attempt {attempt}: {e}")
+                break
+
+        # propagate last exception to outer handler
+        if last_exc:
+            raise last_exc
+        return ""
 
     try:
         result = await loop.run_in_executor(None, call)
-        # clean backticks nếu model có lỡ bọc
-        return result.replace("```", "").strip()
+        result = result.replace("```", "").strip()
+        if result:
+            return result
     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 ""
+    # fallback: return a very small extracted summary (first 1-2 sentences) or empty
+    try:
+        sentences = re.split(r"(?<=[.!?])\s+", text.strip())
+        if not sentences:
+            return ""
+        fallback = " ".join(sentences[:2]).strip()
+        # keep fallback reasonably short
+        if len(fallback) > 400:
+            fallback = fallback[:400].rsplit(" ", 1)[0] + "..."
+        logging.info("[summary_service] Returning fallback summary after errors")
+        return fallback
+    except Exception:
+        return ""