Upload 4 files

Dockerfile

@@ -0,0 +1,23 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements and install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application code
+COPY app.py .
+
+# Expose port
+EXPOSE 7860
+
+# Run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+

README.md

@@ -1,10 +1,45 @@
----
-title: Ht Math Web Backend
-emoji: 🔥
-colorFrom: indigo
-colorTo: blue
-sdk: docker
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# HT_MATH_WEB Backend
+
+Backend API cho ứng dụng chuyển đổi PDF/Ảnh sang Markdown với LaTeX.
+
+## Cấu hình trên Hugging Face Spaces
+
+1. Tạo một Space mới trên Hugging Face
+2. Chọn SDK: **Docker**
+3. Upload các file: `app.py`, `requirements.txt`
+4. Thêm các biến môi trường trong Settings:
+
+### Biến môi trường cần thiết:
+
+```
+GEMINI_API_KEYS=your_key1,your_key2,your_key3
+GEMINI_MODELS=gemini-1.5-flash,gemini-1.5-pro
+SUPABASE_URL=https://your-project.supabase.co
+SUPABASE_KEY=your-supabase-anon-key
+MAX_THREADS=3
+```
+
+### Tạo Dockerfile (nếu cần):
+
+```dockerfile
+FROM python:3.10-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY app.py .
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+```
+
+## API Endpoints
+
+- `GET /` - Health check
+- `GET /api/models` - Lấy danh sách models
+- `POST /api/register` - Đăng ký user
+- `POST /api/login` - Đăng nhập
+- `POST /api/convert` - Chuyển đổi file PDF/ảnh
+- `POST /api/convert-base64` - Chuyển đổi ảnh từ base64
+

app.py

@@ -0,0 +1,402 @@
+"""
+Backend API cho HT_MATH_WEB - Chạy trên Hugging Face Spaces
+Tác giả: Hoàng Tấn Thiên
+"""
+
+import os
+import io
+import json
+import base64
+import tempfile
+import time
+import re
+from typing import List, Optional
+from fastapi import FastAPI, File, UploadFile, HTTPException, Depends, Form
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse, StreamingResponse
+from pydantic import BaseModel
+import fitz  # PyMuPDF
+from PIL import Image
+import google.generativeai as genai
+try:
+    from supabase import create_client, Client
+    SUPABASE_AVAILABLE = True
+except ImportError:
+    SUPABASE_AVAILABLE = False
+    Client = None
+    create_client = None
+
+import hashlib
+import secrets
+
+# ===== CẤU HÌNH =====
+# Các biến môi trường (sẽ được set trên Hugging Face Spaces)
+GEMINI_API_KEYS = os.getenv("GEMINI_API_KEYS", "").split(",")  # Danh sách API keys cách nhau bởi dấu phẩy
+GEMINI_MODELS = os.getenv("GEMINI_MODELS", "gemini-1.5-flash,gemini-1.5-pro").split(",")  # Danh sách models
+SUPABASE_URL = os.getenv("SUPABASE_URL", "")
+SUPABASE_KEY = os.getenv("SUPABASE_KEY", "")
+MAX_THREADS = int(os.getenv("MAX_THREADS", "3"))
+
+# Khởi tạo Supabase client
+supabase = None
+if SUPABASE_AVAILABLE and SUPABASE_URL and SUPABASE_KEY:
+    try:
+        supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
+    except Exception as e:
+        print(f"Warning: Không thể kết nối Supabase: {e}")
+
+# Khởi tạo FastAPI
+app = FastAPI(title="HT_MATH_WEB API", version="6.0")
+
+# CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Trong production, chỉ định domain cụ thể
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# ===== PROMPTS (Giữ nguyên từ desktop app) =====
+DIRECT_GEMINI_PROMPT_TEXT_ONLY = r"""**TRÍCH XUẤT VĂN BẢN THUẦN TÚY**
+Bạn là một chuyên gia trong việc trích xuất nội dung văn bản từ ảnh và PDF.
+NHIỆM VỤ CỦA BẠN:
+1.  **Trích xuất toàn bộ văn bản** từ hình ảnh/PDF được cung cấp.
+2.  **Giữ nguyên định dạng văn bản gốc**, bao gồm các ký tự toán học, mà **KHÔNG** chuyển đổi chúng sang LaTeX. Ví dụ, biểu thức `x^2 + y^2 = r^2` phải được giữ nguyên, không đổi thành `$x^2 + y^2 = r^2$`.
+3.  **Chuyển đổi sang định dạng Markdown** cơ bản cho tiêu đề và bảng biểu.
+4.  Các đề mục, tiêu đề, Định nghĩa, Ví dụ, Bài tập, trắc nghiệm, Lưu ý đều in đậm (dùng markdown `**text**`).
+5.  Không trích xuất Header, Footer, hoặc số trang.
+6.  Giữ nguyên cấu trúc đoạn văn, bảng biểu, và danh sách so với file gốc.
+QUY TẮC QUAN TRỌNG:
+- **KHÔNG SỬ DỤNG LATEX**. Mọi công thức toán học phải được giữ ở dạng văn bản thuần túy như trong tài liệu gốc.
+PHẢN HỒI:
+- Chỉ trả về văn bản Markdown đã được trích xuất.
+- KHÔNG đưa ra bất kỳ giải thích nào hoặc tự ý thêm vào nội dung.
+"""
+
+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.
+**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:**
+    -   **TẤT CẢ, KHÔNG NGOẠI LỆ,** mọi thứ liên quan đến toán học phải được bọc trong cặp dấu đô la `$ ... $`. Điều này bao gồm biến đơn lẻ, số, hàm số, phương trình, ký hiệu, và các biểu thức dù đơn giản hay phức tạp. Đằng sau dấu `$`mở và đằng trước dấu `$` đóng không được phép có khoảng trắng. Ví dụ $[nội dung]$
+    -   **Bảng ví dụ về các lỗi sai phổ biến và cách sửa đúng (BẮT BUỘC TUÂN THEO):**
+        | Sai (Không chấp nhận) | Đúng (Yêu cầu) |
+        | :--- | :--- |
+        | `Vậy f(x)=2x-13.` | `Vậy $f(x)=2x-13$.` |
+        | `Cho M(x)=0.` | `Cho $M(x)=0$.` |
+        | `Thay b=a-15 vào` | `Thay $b=a-15$ vào` |
+        | `a=2; b=-13` | `$a=2$; $b=-13$` |
+        | `Đa thức M(x) có nghiệm x=0.` | `Đa thức $M(x)$ có nghiệm $x=0$.` |
+3.  **QUY TẮC VỀ KHOẢNG TRẮNG (CỰC KỲ QUAN TRỌNG):**
+    -   **LUÔN LUÔN** phải có một khoảng trắng (space) đứng **ngay trước** dấu `$` mở đầu một công thức, trừ khi dấu `$` đó nằm ở đầu một dòng mới.
+    -   **Ví dụ:**
+        -   **SAI:** `...suy ra⇒$b=a-15$`
+        -   **ĐÚNG:** `...suy ra ⇒ $b=a-15$`
+        -   **SAI:** `...với x=0.`
+        -   **ĐÚNG:** `...với $x=0$.`
+4.  **GIỮ NGUYÊN CẤU TRÚC:**
+    -   Giữ nguyên bố cục đoạn văn, các mục danh sách, tiêu đề so với ảnh gốc.
+    -   Các tiêu đề như "Bài 7:", "Dạng 4:", "PHẦN I." phải được **in đậm**.
+    -   Bảng biểu phải được giữ nguyên số hàng, số cột.
+    "**GIỮ NGUYÊN NGẮT DÒNG:** Cố gắng giữ lại các dấu xuống dòng và ngắt đoạn một cách chính xác nhất có thể so với văn bản gốc."
+5. HƯớNG DẫN ĐịNH DạNG:**
+- Các đề mục, tiêu đề, Định nghĩa, Ví dụ, Bài tập, trắc nghiệm, Lưu ý đều in đậm (dùng markdown)
+- Không chuyển Header và Footer, chỉ số trang của ảnh hoặc file pdf.
+- Giữ nguyên format gốc của văn bản (căn lề, khoảng cách, xuống dòng), không được tự ý thay đổi nội dung, hình thức.
+6. Xử lý CÔNG THứC TOáN HọC:**
+    - Chuyển đổi TẤT CẢ công thức toán học sang định dạng LaTeX sử dụng $...$
+    - Đảm bảo các biểu thức, biến số (x,y,z,t,m,n,a,b,c,d,e)đều được đặt trong LaTeX.
+    - Đoạn thẳng (ví dụ: $AB$, $CD$,...)
+    - Điểm trong hình học (ví dụ: từ B kẻ BC thì viết từ $B$ kẻ $BC$)
+    - Đơn vị diện tích: $cm^{2}$
+    - Đơn vị độ: ${90^{0}}$
+    - Tam giác: $\Delta$ không dùng $\Triangle$
+    - Song song: "//" không dùng "\\"
+    - Thay "\cong" bằng dấu "="
+    - Thay "\overline" bằng "\widehat"
+    - Tích phân có cận: $\int\limits_{0}^{2}{y^{2}dx}$
+    - Phân số: hiển thị phân số ở dạng LaTeX sử dụng $\frac{tử số}{mẫu số}$, ví dụ: $\frac{1}{2}$, $\frac{3}{4}$, $\frac{a}{b}$
+    - Số thập phân dùng dấu "," để ký hiệu. Ví dụ: $1,3$
+7. **Xử lý BảNG:**
+    - Bảo toàn cấu trúc bảng gốc - KHÔNG đổi hàng thành cột hoặc cột thành hàng.
+    - Số lượng hàng và cột trong bảng Markdown phải giống hệt như trong tài liệu gốc.
+    - Giữ nguyên thứ tự của các hàng và cột.
+    - Đặc biệt đối với bảng thống kê, bảng dữ liệu: PHẢI giữ nguyên cấu trúc với đúng số cột và hàng.
+    - Sử dụng định dạng bảng Markdown chuẩn với dấu | phân cách cột và dòng --- sau tiêu đề.
+    - Ví dụ, với bảng có dạng:
+    | Chiều cao (cm) | [150;158) | [158;161) | [161;164) | [164;167) |
+    |---------------|-----------|-----------|-----------|-----------|
+    | Số học sinh   |     5     |    12     |    15     |     8     |
+    Phải giữ nguyên cấu trúc này, KHÔNG được chuyển thành:
+    | Chiều cao (cm) | Số học sinh |
+    |---------------|------------|
+    | [150;158)     |      5     |
+    | [158;161)     |     12     |
+    | [161;164)     |     15     |
+    | [164;167)     |      8     |
+8. **PHảN HỒI:**
+- Chỉ trả về văn bản Markdown đã được trích xuất.
+- KHÔNG đưa ra bất kỳ giải thích nào hoặc tự ý thêm vào nội dung.
+**Dữ liệu đầU VàO:**
+Hình ảnh/PDF gốc
+**ĐầU RA MONG MUốN:**
+Văn bản Markdown tiếng Việt hoàn chỉnh với công thức LaTeX và đã được sửa lỗi chính tả.
+**BắT ĐầU THựC HIệN NGAY! **
+"""
+
+# ===== MODELS =====
+class ApiKeyManager:
+    """Quản lý API keys với cơ chế xoay vòng"""
+    def __init__(self, keys: List[str]):
+        self.api_keys = [k.strip() for k in keys if k.strip()]
+        self.current_index = 0
+    
+    def get_next_key(self) -> Optional[str]:
+        if not self.api_keys:
+            return None
+        key = self.api_keys[self.current_index]
+        self.current_index = (self.current_index + 1) % len(self.api_keys)
+        return key
+    
+    def get_key_count(self) -> int:
+        return len(self.api_keys)
+
+# Khởi tạo key manager
+key_manager = ApiKeyManager(GEMINI_API_KEYS)
+
+# ===== HELPER FUNCTIONS =====
+def clean_latex_formulas(text: str) -> str:
+    """Làm sạch công thức LaTeX"""
+    return re.sub(r'\$(.*?)\$', lambda m: f'${m.group(1).strip()}$', text)
+
+def hash_password(password: str) -> str:
+    """Hash mật khẩu"""
+    return hashlib.sha256(password.encode()).hexdigest()
+
+def verify_password(password: str, hashed: str) -> bool:
+    """Xác thực mật khẩu"""
+    return hash_password(password) == hashed
+
+# ===== API ENDPOINTS =====
+
+@app.get("/")
+async def root():
+    """Health check endpoint"""
+    return {
+        "status": "ok",
+        "service": "HT_MATH_WEB API",
+        "version": "6.0",
+        "keys_loaded": key_manager.get_key_count(),
+        "models_available": GEMINI_MODELS
+    }
+
+@app.get("/api/models")
+async def get_models():
+    """Lấy danh sách models"""
+    return {"models": GEMINI_MODELS}
+
+@app.post("/api/register")
+async def register(email: str = Form(...), password: str = Form(...)):
+    """Đăng ký user mới"""
+    if not supabase:
+        raise HTTPException(status_code=500, detail="Database không khả dụng")
+    
+    try:
+        # Kiểm tra email đã tồn tại chưa
+        result = supabase.table("users").select("email").eq("email", email).execute()
+        if result.data:
+            raise HTTPException(status_code=400, detail="Email đã được sử dụng")
+        
+        # Tạo user mới
+        hashed_password = hash_password(password)
+        user_data = {
+            "email": email,
+            "password": hashed_password,
+            "created_at": time.strftime("%Y-%m-%d %H:%M:%S")
+        }
+        
+        result = supabase.table("users").insert(user_data).execute()
+        return {"success": True, "message": "Đăng ký thành công"}
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Lỗi đăng ký: {str(e)}")
+
+@app.post("/api/login")
+async def login(email: str = Form(...), password: str = Form(...)):
+    """Đăng nhập"""
+    if not supabase:
+        raise HTTPException(status_code=500, detail="Database không khả dụng")
+    
+    try:
+        result = supabase.table("users").select("*").eq("email", email).execute()
+        if not result.data:
+            raise HTTPException(status_code=401, detail="Email hoặc mật khẩu không đúng")
+        
+        user = result.data[0]
+        if not verify_password(password, user["password"]):
+            raise HTTPException(status_code=401, detail="Email hoặc mật khẩu không đúng")
+        
+        # Tạo token đơn giản (trong production nên dùng JWT)
+        token = secrets.token_urlsafe(32)
+        # Lưu token vào session (có thể lưu vào Redis hoặc database)
+        
+        return {
+            "success": True,
+            "token": token,
+            "email": email
+        }
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Lỗi đăng nhập: {str(e)}")
+
+async def process_image_with_gemini(
+    image: Image.Image,
+    model_id: str,
+    prompt: str,
+    max_retries: int = 3
+) -> str:
+    """Xử lý ảnh với Gemini API"""
+    for attempt in range(max_retries):
+        try:
+            api_key = key_manager.get_next_key()
+            if not api_key:
+                raise ValueError("Không tìm thấy API Key nào")
+            
+            genai.configure(api_key=api_key)
+            
+            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"},
+            ]
+            
+            generation_config = {
+                "temperature": 0.1,
+                "top_p": 0.95,
+                "top_k": 64,
+                "max_output_tokens": 8192,
+            }
+            
+            model = genai.GenerativeModel(
+                model_name=model_id,
+                generation_config=generation_config,
+                safety_settings=safety_settings
+            )
+            
+            response = model.generate_content([prompt, image])
+            
+            if response.text:
+                return response.text
+            else:
+                raise ValueError("Phản hồi từ API rỗng")
+        
+        except Exception as e:
+            error_str = str(e)
+            if "429" in error_str or "ResourceExhausted" in error_str or "Quota exceeded" in error_str:
+                if attempt < max_retries - 1:
+                    time.sleep(2 * (attempt + 1))
+                    continue
+            raise HTTPException(status_code=500, detail=f"Lỗi xử lý ảnh: {error_str}")
+    
+    raise HTTPException(status_code=500, detail="Thất bại sau nhiều lần thử")
+
+@app.post("/api/convert")
+async def convert_file(
+    file: UploadFile = File(...),
+    model: str = Form("gemini-1.5-flash"),
+    mode: str = Form("latex")  # "latex" hoặc "text_only"
+):
+    """Chuyển đổi PDF/ảnh sang Markdown"""
+    if key_manager.get_key_count() == 0:
+        raise HTTPException(status_code=500, detail="Chưa cấu hình API Key")
+    
+    if model not in GEMINI_MODELS:
+        raise HTTPException(status_code=400, detail=f"Model không hợp lệ. Models khả dụng: {GEMINI_MODELS}")
+    
+    prompt = DIRECT_GEMINI_PROMPT_LATEX if mode == "latex" else DIRECT_GEMINI_PROMPT_TEXT_ONLY
+    
+    try:
+        # Đọc file
+        file_content = await file.read()
+        file_ext = os.path.splitext(file.filename)[1].lower()
+        
+        results = []
+        
+        if file_ext == ".pdf":
+            # Xử lý PDF
+            doc = fitz.open(stream=file_content, filetype="pdf")
+            for page_num in range(len(doc)):
+                page = doc[page_num]
+                pix = page.get_pixmap(dpi=250)
+                img = Image.open(io.BytesIO(pix.tobytes(output="png")))
+                
+                text = await process_image_with_gemini(img, model, prompt)
+                results.append(text)
+            
+            doc.close()
+        elif file_ext in [".png", ".jpg", ".jpeg", ".bmp"]:
+            # Xử lý ảnh
+            img = Image.open(io.BytesIO(file_content))
+            text = await process_image_with_gemini(img, model, prompt)
+            results.append(text)
+        else:
+            raise HTTPException(status_code=400, detail="Định dạng file không được hỗ trợ")
+        
+        # Tổng hợp kết quả
+        final_text = "\n\n".join(results)
+        cleaned_text = clean_latex_formulas(final_text)
+        
+        return {
+            "success": True,
+            "result": cleaned_text,
+            "pages_processed": len(results)
+        }
+    
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Lỗi xử lý file: {str(e)}")
+
+@app.post("/api/convert-base64")
+async def convert_base64(
+    image_data: str = Form(...),  # Base64 encoded image
+    model: str = Form("gemini-1.5-flash"),
+    mode: str = Form("latex")
+):
+    """Chuyển đổi ảnh từ base64"""
+    if key_manager.get_key_count() == 0:
+        raise HTTPException(status_code=500, detail="Chưa cấu hình API Key")
+    
+    if model not in GEMINI_MODELS:
+        raise HTTPException(status_code=400, detail=f"Model không hợp lệ")
+    
+    prompt = DIRECT_GEMINI_PROMPT_LATEX if mode == "latex" else DIRECT_GEMINI_PROMPT_TEXT_ONLY
+    
+    try:
+        # Decode base64
+        if "," in image_data:
+            image_data = image_data.split(",")[1]  # Remove data:image/png;base64, prefix
+        
+        image_bytes = base64.b64decode(image_data)
+        img = Image.open(io.BytesIO(image_bytes))
+        
+        text = await process_image_with_gemini(img, model, prompt)
+        cleaned_text = clean_latex_formulas(text)
+        
+        return {
+            "success": True,
+            "result": cleaned_text
+        }
+    
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Lỗi xử lý ảnh: {str(e)}")
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)
+

requirements.txt

@@ -0,0 +1,9 @@
+fastapi==0.104.1
+uvicorn[standard]==0.24.0
+python-multipart==0.0.6
+pillow==10.1.0
+PyMuPDF==1.23.8
+google-generativeai==0.3.2
+supabase==2.0.3
+pydantic==2.5.0
+