Upload 3 files

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y \
+    libglib2.0-0 \
+    libgl1 \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,38 @@
+from fastapi import FastAPI, UploadFile, File
+from doctr.io import DocumentFile
+from doctr.models import ocr_predictor
+from PIL import Image
+import tempfile
+import io
+import os
+
+app = FastAPI()
+
+print("Loading OCR model...")
+model = ocr_predictor(pretrained=True)
+print("OCR ready.")
+
+@app.get("/")
+def root():
+    return {"status": "OCR running"}
+
+@app.post("/ocr")
+async def ocr_images(file: UploadFile = File(...)):
+    try:
+        contents = await file.read()
+        pil_image = Image.open(io.BytesIO(contents)).convert("RGB")
+
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp:
+            pil_image.save(tmp.name)
+            temp_path = tmp.name
+
+        doc = DocumentFile.from_images([temp_path])
+        result = model(doc)
+        text = result.render()
+
+        os.remove(temp_path)
+
+        return {"success": True, "text": text}
+
+    except Exception as e:
+        return {"success": False, "error": str(e)}