Update app.py

app.py

@@ -1,6 +1,6 @@
 """
 Backend API cho HT_MATH_WEB - Chạy trên Hugging Face Spaces (Docker Version)
-Phiên bản: 9.8 (Revert to Strict Prompt + Full Features)
+Phiên bản: 9.9 (Forced Split Mode - Fix Missing Content)
 Tác giả: Hoàng Tấn Thiên
 """
 
@@ -27,7 +27,6 @@ import google.generativeai as genai
 # --- PANDOC IMPORT ---
 try:
     import pypandoc
-    # print(f"INFO: Pandoc version detected: {pypandoc.get_pandoc_version()}")
 except ImportError:
     print("CRITICAL WARNING: pypandoc module not found.")
 except OSError:
@@ -58,7 +57,7 @@ if SUPABASE_AVAILABLE and SUPABASE_URL and SUPABASE_KEY:
     except Exception as e:
         print(f"Warning: Không thể kết nối Supabase: {e}")
 
-app = FastAPI(title="HT_MATH_WEB API", version="9.8")
+app = FastAPI(title="HT_MATH_WEB API", version="9.9")
 
 app.add_middleware(
     CORSMiddleware,
@@ -105,7 +104,7 @@ def check_rate_limit(request: Request):
             pass
     ip_rate_limits[client_ip] = now
 
-# ===== PROMPTS (KHÔI PHỤC BẢN CHUẨN MỰC) =====
+# ===== PROMPTS (CHUẨN MỰC) =====
 DIRECT_GEMINI_PROMPT_TEXT_ONLY = r"""**TRÍCH XUẤT VĂN BẢN THUẦN TÚY**
 ⚠️ YÊU CẦU BẮT BUỘC:
 - PHẢI trích xuất TOÀN BỘ nội dung xuất hiện trong ảnh/PDF
@@ -199,7 +198,7 @@ def stitch_text(text_a: str, text_b: str, min_overlap_chars: int = 20) -> str:
     for i in range(scan_window, 0, -1):
         tail_a = "\n".join(a_lines[-i:]).strip()
         head_b = "\n".join(b_lines[:i]).strip()
-        # So sánh lỏng hơn một chút để tránh lỗi do khoảng trắng thừa
+        # So sánh lỏng hơn để tìm điểm trùng
         if len(tail_a) >= min_overlap_chars and tail_a.replace(" ","") == head_b.replace(" ",""):
             best_overlap_idx = i
             break
@@ -207,7 +206,6 @@ def stitch_text(text_a: str, text_b: str, min_overlap_chars: int = 20) -> str:
     if best_overlap_idx > 0:
         return text_a + "\n" + "\n".join(b_lines[best_overlap_idx:])
     else:
-        # Nếu không tìm thấy điểm nối, nối tiếp luôn bằng 2 dòng trống để an toàn
         return text_a + "\n\n" + text_b
 
 def clean_latex_formulas(text: str) -> str:
@@ -224,7 +222,7 @@ def verify_password(password: str, hashed: str) -> bool:
 @app.get("/")
 @app.get("/health")
 async def root():
-    return {"status": "ok", "version": "9.8"}
+    return {"status": "ok", "version": "9.9"}
 
 @app.get("/api/models")
 async def get_models():
@@ -315,7 +313,7 @@ async def upload_image(file: UploadFile = File(...)):
 async def process_image_with_gemini(image: Image.Image, model_id: str, prompt: str, manager: ApiKeyManager, max_retries: int = 3) -> str:
     """Gửi ảnh lên Gemini với settings tắt bộ lọc an toàn để tránh mất nội dung"""
     
-    # Cấu hình tắt bộ lọc an toàn (QUAN TRỌNG ĐỂ KHÔNG BỊ MẤT CHỮ)
+    # Cấu hình tắt bộ lọc an toàn
     safety_settings = [
         {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
         {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
@@ -333,7 +331,6 @@ async def process_image_with_gemini(image: Image.Image, model_id: str, prompt: s
             genai.configure(api_key=api_key)
             generation_config = {"temperature": 0.0, "top_p": 1.0, "max_output_tokens": 8192}
             
-            # Khởi tạo model kèm safety_settings
             model = genai.GenerativeModel(
                 model_id, 
                 generation_config=generation_config,
@@ -358,9 +355,11 @@ async def process_image_with_gemini(image: Image.Image, model_id: str, prompt: s
     return ""
 
 async def process_large_image(image: Image.Image, model: str, prompt: str, semaphore: asyncio.Semaphore, manager: ApiKeyManager) -> str:
-    # Chunk height lớn (4096) để đọc 1 lần hết trang A4 tiêu chuẩn (200dpi)
-    CHUNK_HEIGHT = 4096 
-    OVERLAP_HEIGHT = 256
+    # THAY ĐỔI QUAN TRỌNG: Giảm CHUNK_HEIGHT xuống 1500 (Khoảng 2/3 trang A4 @ 200dpi)
+    # Điều này ÉP BUỘC code phải cắt trang A4 thành ít nhất 2 phần
+    # Giúp AI không bị quá tải và bỏ sót nửa trang sau.
+    CHUNK_HEIGHT = 1500 
+    OVERLAP_HEIGHT = 300
     
     width, height = image.size
     
@@ -378,7 +377,6 @@ async def process_large_image(image: Image.Image, model: str, prompt: str, semap
         if bottom == height: break
         y += (CHUNK_HEIGHT - OVERLAP_HEIGHT)
     
-    # Nếu phải cắt, xử lý từng phần rồi nối lại
     async def process_chunk(chunk_img, index):
         async with semaphore:
             text = await process_image_with_gemini(chunk_img, model, prompt, manager)