Upload 3 files

Dockerfile.txt

@@ -0,0 +1,11 @@
+FROM python:3.11
+
+WORKDIR /app
+
+COPY requirements.txt .
+
+RUN pip install -r requirements.txt
+
+COPY . .
+
+CMD uvicorn app:app --host 0.0.0.0 --port 7860

app.py

@@ -0,0 +1,48 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from google.cloud import vision
+import base64
+import json
+import os
+
+app = FastAPI()
+
+class OCRRequest(BaseModel):
+    image_base64: str
+
+# Arabic digits -> English digits
+translation = str.maketrans(
+    "٠١٢٣٤٥٦٧٨٩",
+    "0123456789"
+)
+
+# Load Google credentials from HF Secret
+creds_json = os.environ["GOOGLE_CREDS_JSON"]
+
+with open("google_creds.json", "w", encoding="utf-8") as f:
+    f.write(creds_json)
+
+os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "google_creds.json"
+
+client = vision.ImageAnnotatorClient()
+
+@app.post("/ocr")
+async def ocr(req: OCRRequest):
+
+    image = vision.Image(
+        content=base64.b64decode(req.image_base64)
+    )
+
+    response = client.text_detection(image=image)
+
+    text = ""
+
+    if response.text_annotations:
+        text = response.text_annotations[0].description
+
+    text = text.translate(translation)
+
+    return {
+        "success": True,
+        "text": text
+    }

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+google-cloud-vision
+pydantic