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.0 (Copyright Safety & Fallback OCR)
+Phiên bản: 9.1 (Strict OCR - Anti-Hallucination)
 Tác giả: Hoàng Tấn Thiên
 """
 
@@ -68,7 +68,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.0")
+app = FastAPI(title="HT_MATH_WEB API", version="9.1")
 
 app.add_middleware(
     CORSMiddleware,
@@ -122,30 +122,34 @@ def check_rate_limit(request: Request):
             print(f"[RateLimit] IP {client_ip} requesting too fast.")
     ip_rate_limits[client_ip] = now
 
-# ===== SAFE PROMPTS (ANTI-COPYRIGHT) =====
-# Đã đổi prompt để tránh trigger bản quyền (finish_reason=4)
+# ===== SAFE PROMPTS (STRICT OCR - NO CHATTER) =====
+# Đã thay đổi toàn bộ Prompt theo hướng "Máy OCR lạnh lùng"
+# Loại bỏ hoàn toàn vai trò "Trợ lý ảo" để tránh nói nhảm.
 
-DIRECT_GEMINI_PROMPT_TEXT_ONLY = r"""Bạn là trợ lý AI hỗ trợ người khiếm thị số hóa tài liệu.
-Nhiệm vụ: Nhận diện và mô tả lại nội dung văn bản trong hình ảnh một cách trung thực, chính xác.
-Yêu cầu:
-- Trích xuất nội dung văn bản để lưu trữ dữ liệu.
-- Giữ nguyên các thông số, dữ liệu toán học.
-- Định dạng Markdown rõ ràng.
-- KHÔNG thêm lời bình luận.
-"""
+DIRECT_GEMINI_PROMPT_TEXT_ONLY = r"""Đóng vai một CÔNG CỤ OCR CHUYÊN DỤNG cho văn bản hành chính Việt Nam.
+NHIỆM VỤ: Trích xuất nguyên văn (Verbatim) nội dung trong ảnh.
 
-DIRECT_GEMINI_PROMPT_LATEX = r"""Bạn là trợ lý AI hỗ trợ số hóa tài liệu giáo dục. 
-Nhiệm vụ: Phân tích hình ảnh và viết lại nội dung dưới dạng Markdown chuẩn Toán học (LaTeX).
-Mục đích: Số hóa để lưu trữ và giảng dạy, không nhằm mục đích sao chép thương mại.
+QUY TẮC BẤT DI BẤT DỊCH (CẤM VI PHẠM):
+1. TUYỆT ĐỐI KHÔNG thêm lời dẫn đầu (như "Dưới đây là...", "Chào bạn...", "Văn bản này nói về...").
+2. TUYỆT ĐỐI KHÔNG thêm lời kết luận hay nhận xét, đánh giá.
+3. KHÔNG thay đổi từ ngữ, KHÔNG diễn giải (paraphrase), KHÔNG tóm tắt. Phải giữ nguyên văn kể cả lỗi chính tả của bản gốc.
+4. Giữ nguyên cấu trúc dòng, đoạn, số hiệu, ngày tháng, quốc hiệu, tiêu ngữ.
+5. Nếu chữ quá mờ không đọc được, hãy điền ký hiệu `[...]`. KHÔNG ĐƯỢC TỰ Ý ĐOÁN MÒ HAY BỊA ĐẶT NỘI DUNG.
 
-⚠️ YÊU CẦU XỬ LÝ AN TOÀN:
-1. Nếu gặp nội dung giống sách giáo khoa hoặc tài liệu có bản quyền, hãy VIẾT LẠI (paraphrase) lời dẫn nhưng GIỮ NGUYÊN công thức toán học và số liệu.
-2. Trình bày công thức toán trong dấu `$`. Ví dụ: $x^2 + 2x = 0$.
-3. Đảm bảo cấu trúc bài tập rõ ràng (Câu 1, Câu 2...).
-4. Nếu hình ảnh mờ hoặc bị cắt, hãy cố gắng phục hồi dựa trên ngữ cảnh toán học.
+ĐẦU RA: Chỉ trả về văn bản thô. Không Markdown dư thừa nếu không cần thiết."""
 
-CHỈ TRẢ VỀ NỘI DUNG MARKDOWN.
-"""
+DIRECT_GEMINI_PROMPT_LATEX = r"""Đóng vai công cụ số hóa tài liệu Toán học chính xác tuyệt đối.
+NHIỆM VỤ: Chuyển đổi ảnh sang Markdown + LaTeX.
+
+QUY TẮC:
+1. Công thức toán học PHẢI nằm trong dấu `$`. Ví dụ: $x^2 + 1 = 0$.
+2. KHÔNG thêm lời dẫn thừa (như "Kết quả là:", "Đây là bài giải:").
+3. GIỮ NGUYÊN VĂN đề bài và lời giải. KHÔNG tóm tắt, KHÔNG viết lại theo văn phong khác.
+4. Nếu ảnh bị cắt hoặc mờ:
+   - Nếu là công thức toán có thể suy luận logic: Hãy phục hồi.
+   - Nếu là văn bản/tên riêng: Điền `[...]`.
+
+CHỈ TRẢ VỀ NỘI DUNG MARKDOWN."""
 
 # ===== STITCHING ALGORITHM =====
 def stitch_text(text_a: str, text_b: str, min_overlap_chars: int = 20) -> str:
@@ -243,7 +247,7 @@ async def root():
 
     return {
         "status": "ok",
-        "service": "HT_MATH_WEB API v9.0 (Safe Mode)",
+        "service": "HT_MATH_WEB API v9.1 (Strict Mode)",
         "keys_loaded": key_manager.get_key_count(),
         "pandoc": pandoc_status,
         "tesseract": tesseract_status
@@ -314,7 +318,8 @@ async def process_image_with_gemini(image: Image.Image, model_id: str, prompt: s
             if not api_key: raise ValueError("No API Key")
             genai.configure(api_key=api_key)
             
-            generation_config = {"temperature": 0.2, "top_p": 1.0, "max_output_tokens": 8192}
+            # Giảm temperature xuống 0.0 để model trở nên máy móc nhất có thể (Deterministic)
+            generation_config = {"temperature": 0.0, "top_p": 1.0, "max_output_tokens": 8192}
             model = genai.GenerativeModel(model_id, generation_config=generation_config)
             
             # Gọi API
@@ -337,7 +342,6 @@ async def process_image_with_gemini(image: Image.Image, model_id: str, prompt: s
                 continue
             if attempt == max_retries - 1:
                 print(f"Error Gemini: {e}")
-                # Nếu lỗi mạng/quota quá nhiều, cũng fallback luôn cho chắc
                 return await fallback_ocr_tesseract(image)
                 
     return ""