Update app.py

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.responses import JSONResponse, HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
+from fastapi.middleware.cors import CORSMiddleware
 import fitz  # PyMuPDF
 from transformers import AutoModel, AutoTokenizer
 from PIL import Image
@@ -20,6 +21,15 @@ import uvicorn
 
 app = FastAPI()
 
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allow all origins for simplicity
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
 model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, device_map='cuda', use_safetensors=True)
@@ -91,7 +101,7 @@ def cleanup_old_files():
     current_time = time.time()
     for folder in [UPLOAD_FOLDER, RESULTS_FOLDER]:
         for file_path in Path(folder).glob('*'):
-            if current_time - file_path.stat().st_mtime > 3600:  # 1 hour
+            if current_time - file_path.stat().st_mtime > 3600:  # 1 hour
                 file_path.unlink()
 
 cleanup_old_files()