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.6 (Add Admin Routes)
+Phiên bản: 9.8 (Revert to Strict Prompt + Full Features)
 Tác giả: Hoàng Tấn Thiên
 """
 
@@ -43,13 +43,12 @@ except ImportError:
     create_client = None
 
 # ===== CẤU HÌNH =====
-# Keys mặc định của server (dự phòng)
 SERVER_GEMINI_KEYS = os.getenv("GEMINI_API_KEYS", "").split(",")
 GEMINI_MODELS = os.getenv("GEMINI_MODELS", "gemini-2.5-flash,gemini-1.5-flash,gemini-1.5-pro").split(",")
 SUPABASE_URL = os.getenv("SUPABASE_URL", "")
 SUPABASE_KEY = os.getenv("SUPABASE_KEY", "")
 MAX_THREADS = int(os.getenv("MAX_THREADS", "8"))
-ADMIN_SECRET_KEY = os.getenv("ADMIN_SECRET_KEY", "admin123") # Key bảo mật cho Admin
+ADMIN_SECRET_KEY = os.getenv("ADMIN_SECRET_KEY", "admin123")
 
 # Setup Supabase
 supabase = None
@@ -59,7 +58,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.6")
+app = FastAPI(title="HT_MATH_WEB API", version="9.8")
 
 app.add_middleware(
     CORSMiddleware,
@@ -69,14 +68,12 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# --- SETUP STATIC FILES ---
 os.makedirs("uploads", exist_ok=True)
 app.mount("/uploads", StaticFiles(directory="uploads"), name="uploads")
 
-# ===== KEY MANAGER CLASS =====
+# ===== KEY MANAGER =====
 class ApiKeyManager:
     def __init__(self, keys: List[str]):
-        # Lọc bỏ key rỗng và khoảng trắng
         self.api_keys = [k.strip() for k in keys if k.strip()]
         self.current_index = 0
         self.total_keys = len(self.api_keys)
@@ -86,7 +83,6 @@ class ApiKeyManager:
         return self.api_keys[self.current_index]
     
     def rotate_key(self) -> Optional[str]:
-        """Chuyển sang key tiếp theo"""
         if not self.api_keys: return None
         self.current_index = (self.current_index + 1) % self.total_keys
         print(f"[KeyManager] Rotated to key index: {self.current_index}/{self.total_keys}")
@@ -95,10 +91,7 @@ class ApiKeyManager:
     def get_key_count(self) -> int:
         return len(self.api_keys)
 
-# Global manager cho keys của server
 server_key_manager = ApiKeyManager(SERVER_GEMINI_KEYS)
-
-# Rate limiter đơn giản
 ip_rate_limits = {}
 RATE_LIMIT_DURATION = 2 
 
@@ -109,10 +102,10 @@ def check_rate_limit(request: Request):
     if client_ip in ip_rate_limits:
         elapsed = now - ip_rate_limits[client_ip]
         if elapsed < RATE_LIMIT_DURATION:
-            pass # Có thể log warning
+            pass
     ip_rate_limits[client_ip] = now
 
-# ===== PROMPTS =====
+# ===== PROMPTS (KHÔI PHỤC BẢN 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
@@ -124,10 +117,6 @@ DIRECT_GEMINI_PROMPT_TEXT_ONLY = r"""**TRÍCH XUẤT VĂN BẢN THUẦN TÚY**
 DIRECT_GEMINI_PROMPT_LATEX = r"""**TRÍCH XUẤT SANG MARKDOWN VỚI CÔNG THỨC LATEX ($...$)**
 Bạn là một trợ lý kỹ thuật có nhiệm vụ duy nhất là chuyển đổi hình ảnh chứa nội dung toán học sang định dạng Markdown.
 Sự chính xác tuyệt đối trong việc tuân thủ các quy tắc dưới đây là yêu cầu bắt buộc để đầu ra tương thích với công cụ xử lý tự động.
-⚠️ YÊU CẦU CỐT LÕI - KHÔNG ĐƯỢC BỎ SÓT:
-- Đọc kỹ từng pixel, trích xuất TOÀN BỘ nội dung từ trên xuống dưới.
-- KHÔNG bỏ qua bất kỳ bài tập, hình vẽ, hoặc ghi chú nhỏ nào.
-- Nếu ảnh bị cắt ngang chữ, hãy cố gắng đoán và hoàn thiện từ đó dựa trên ngữ cảnh.
 **QUY TẮC BẤT DI BẤT DỊCH:**
 1.  **CHỈ MARKDOWN:** Toàn bộ đầu ra phải là văn bản Markdown thuần túy. CẤM TUYỆT ĐỐI chứa các lệnh như `\documentclass`, `\usepackage`, `\begin{document}`.
 2.  **QUY TẮC DẤU ĐÔ LA ($) - QUAN TRỌNG NHẤT:**
@@ -210,17 +199,18 @@ 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()
-        if len(tail_a) >= min_overlap_chars and tail_a == head_b:
+        # So sánh lỏng hơn một chút để tránh lỗi do khoảng trắng thừa
+        if len(tail_a) >= min_overlap_chars and tail_a.replace(" ","") == head_b.replace(" ",""):
             best_overlap_idx = i
             break
             
     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:
-    # Chuẩn hóa khoảng trắng LaTeX
     return re.sub(r'\$\s+(.*?)\s+\$', lambda m: f'${m.group(1).strip()}$', text)
 
 def hash_password(password: str) -> str:
@@ -234,7 +224,7 @@ def verify_password(password: str, hashed: str) -> bool:
 @app.get("/")
 @app.get("/health")
 async def root():
-    return {"status": "ok", "version": "9.6"}
+    return {"status": "ok", "version": "9.8"}
 
 @app.get("/api/models")
 async def get_models():
@@ -281,46 +271,33 @@ async def logout(request: Request):
     except: pass
     return {"status": "success"}
 
-# --- ADMIN API (BỔ SUNG) ---
-
+# --- ADMIN API ---
 @app.get("/admin/users")
 async def admin_get_users(secret: str = ""):
-    if secret != ADMIN_SECRET_KEY:
-        raise HTTPException(status_code=403, detail="Forbidden")
-    if not supabase:
-        raise HTTPException(status_code=500, detail="DB Error")
-    
+    if secret != ADMIN_SECRET_KEY: raise HTTPException(status_code=403, detail="Forbidden")
+    if not supabase: raise HTTPException(status_code=500, detail="DB Error")
     try:
         res = supabase.table("users").select("*").order("created_at", desc=True).execute()
         return {"users": res.data}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+    except Exception as e: raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/admin/approve-user")
 async def admin_approve_user(email: str = Form(...), secret: str = Form(...)):
-    if secret != ADMIN_SECRET_KEY:
-         raise HTTPException(status_code=403, detail="Forbidden")
-    if not supabase:
-         raise HTTPException(status_code=500, detail="DB Error")
-         
+    if secret != ADMIN_SECRET_KEY: raise HTTPException(status_code=403, detail="Forbidden")
+    if not supabase: raise HTTPException(status_code=500, detail="DB Error")
     try:
         supabase.table("users").update({"status": "active"}).eq("email", email).execute()
         return {"success": True}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+    except Exception as e: raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/admin/delete-user")
 async def admin_delete_user(email: str = Form(...), secret: str = Form(...)):
-    if secret != ADMIN_SECRET_KEY:
-         raise HTTPException(status_code=403, detail="Forbidden")
-    if not supabase:
-         raise HTTPException(status_code=500, detail="DB Error")
-         
+    if secret != ADMIN_SECRET_KEY: raise HTTPException(status_code=403, detail="Forbidden")
+    if not supabase: raise HTTPException(status_code=500, detail="DB Error")
     try:
         supabase.table("users").delete().eq("email", email).execute()
         return {"success": True}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+    except Exception as e: raise HTTPException(status_code=500, detail=str(e))
 
 # --- IMAGE UPLOAD ---
 @app.post("/api/upload-image")
@@ -336,10 +313,16 @@ async def upload_image(file: UploadFile = File(...)):
 # --- CORE PROCESSING ---
 
 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 cơ chế xoay vòng Key khi gặp lỗi 429"""
+    """Gửi ảnh lên Gemini với settings tắt bộ lọc an toàn để tránh mất nội dung"""
     
-    # Thử tối đa số lần = số lượng key * 2 (để chắc chắn đã quay vòng hết các key)
-    # Hoặc giới hạn cứng là 10 lần để tránh loop vô hạn
+    # Cấu hình tắt bộ lọc an toàn (QUAN TRỌNG ĐỂ KHÔNG BỊ MẤT CHỮ)
+    safety_settings = [
+        {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
+        {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
+        {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
+        {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
+    ]
+
     max_attempts = min(manager.get_key_count() * 2, 10) if manager.get_key_count() > 0 else 3
 
     for i in range(max_attempts):
@@ -349,30 +332,33 @@ async def process_image_with_gemini(image: Image.Image, model_id: str, prompt: s
         try:
             genai.configure(api_key=api_key)
             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
+            # Khởi tạo model kèm safety_settings
+            model = genai.GenerativeModel(
+                model_id, 
+                generation_config=generation_config,
+                safety_settings=safety_settings
+            )
+            
             response = model.generate_content([prompt, image])
             if response.text:
                 return response.text.strip()
                 
         except Exception as e:
             error_str = str(e)
-            # Nếu gặp lỗi Quota (429) hoặc lỗi Model quá tải (503) -> Xoay Key
             if "429" in error_str or "quota" in error_str.lower() or "503" in error_str:
-                print(f"[KeyManager] Key {api_key[:8]}... exhausted/error. Rotating...")
+                print(f"[KeyManager] Key {api_key[:8]}... exhausted. Rotating...")
                 manager.rotate_key()
-                time.sleep(1) # Nghỉ 1s trước khi thử key mới
+                time.sleep(1)
                 continue
             
-            # Các lỗi khác (400, v.v) thì không retry bằng key khác làm gì
             print(f"[Gemini Error] {error_str}")
             if i == max_attempts - 1: raise e
             
     return ""
 
 async def process_large_image(image: Image.Image, model: str, prompt: str, semaphore: asyncio.Semaphore, manager: ApiKeyManager) -> str:
-    # Tăng kích thước chunk để giảm số request
+    # Chunk height lớn (4096) để đọc 1 lần hết trang A4 tiêu chuẩn (200dpi)
     CHUNK_HEIGHT = 4096 
     OVERLAP_HEIGHT = 256
     
@@ -392,6 +378,7 @@ 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)
@@ -413,21 +400,18 @@ async def convert_file(
     file: UploadFile = File(...),
     model: str = Form("gemini-2.5-flash"),
     mode: str = Form("latex"),
-    api_keys: str = Form(None) # Nhận danh sách keys từ client
+    api_keys: str = Form(None) 
 ):
     check_rate_limit(request)
     
-    # Ưu tiên dùng keys từ User gửi lên
     if api_keys and len(api_keys.strip()) > 0:
         user_keys = api_keys.split(',')
         manager = ApiKeyManager(user_keys)
-        # print(f"Using {manager.get_key_count()} user provided keys.")
     else:
-        # Fallback về keys của server
         manager = server_key_manager
         
     if manager.get_key_count() == 0:
-        raise HTTPException(status_code=400, detail="Chưa có API Key. Vui lòng nạp file Key hoặc cấu hình Server.")
+        raise HTTPException(status_code=400, detail="Chưa có API Key.")
     
     prompt = DIRECT_GEMINI_PROMPT_LATEX if mode == "latex" else DIRECT_GEMINI_PROMPT_TEXT_ONLY
     
@@ -441,7 +425,6 @@ async def convert_file(
         if file_ext == ".pdf":
             doc = fitz.open(stream=file_content, filetype="pdf")
             async def process_page_wrapper(page, idx):
-                # Render 200 DPI cho nhanh
                 pix = page.get_pixmap(dpi=200) 
                 img = Image.open(io.BytesIO(pix.tobytes("png")))
                 text = await process_large_image(img, model, prompt, global_semaphore, manager)
@@ -463,8 +446,6 @@ async def convert_file(
         return {"success": True, "result": clean_latex_formulas(final_text)}
         
     except Exception as e:
-        # import traceback
-        # traceback.print_exc()
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/api/export-docx")