Update app.py

app.py

@@ -10,14 +10,12 @@ from PIL import Image
 from pydantic import BaseModel
 from typing import List
 
-# For simplicity, we directly use the fast functions.
-# The `image_processor.py` file now contains the optimized versions.
 from image_processor import enhance_image_fast, extract_text_from_image_fast, process_pdf_in_parallel
 
 app = FastAPI(
     title="High-Speed OCR API",
     description="An API to extract text from images and PDFs, optimized for speed.",
-    version="4.0.0-hf"
+    version="5.0.0-hf-final"
 )
 
 # CORS Middleware to allow requests from any origin
@@ -51,7 +49,6 @@ def read_root():
 
 @app.post("/ocr-image", response_model=ImageOCRResponse, tags=["OCR"])
 async def ocr_image_endpoint(file: UploadFile = File(...)):
-    """Accepts an image, enhances it, and returns the extracted text."""
     if not file.content_type.startswith("image/"):
         raise HTTPException(status_code=400, detail="File must be an image.")
     try:
@@ -61,11 +58,10 @@ async def ocr_image_endpoint(file: UploadFile = File(...)):
         text = extract_text_from_image_fast(enhanced_image)
         return {"filename": file.filename, "text": text}
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error processing image OCR: {e}")
+        raise HTTPException(status_code=500, detail=f"Error processing image OCR: {str(e)}")
 
 @app.post("/ocr-pdf", response_model=PDFOCRResponse, tags=["OCR"])
 async def ocr_pdf_endpoint(file: UploadFile = File(...)):
-    """Accepts a PDF, extracts text from each page in parallel, and returns structured results."""
     if file.content_type != "application/pdf":
         raise HTTPException(status_code=400, detail="File must be a PDF.")
     try:
@@ -77,10 +73,7 @@ async def ocr_pdf_endpoint(file: UploadFile = File(...)):
             "results": results
         }
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error processing PDF: {e}")
+        raise HTTPException(status_code=500, detail=f"Error processing PDF: {str(e)}")
 
-# This block allows running the app locally for testing
-if __name__ == "__main__":
-    # Hugging Face Spaces expects the app to run on port 7860
-    port = int(os.environ.get("PORT", 7860)) 
-    uvicorn.run("app:app", host="0.0.0.0", port=port, reload=True) 
+# The `if __name__ == "__main__":` block has been completely removed.
+# The platform will import the `app` object and run it.