first commit

.gitignore

@@ -0,0 +1 @@
+.env

Dockerfile

@@ -0,0 +1,16 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.11-slim
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,44 @@
+from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
+from PIL import Image
+import io
+import os
+import google.generativeai as genai
+from fastapi.responses import FileResponse
+from pdf_generator import create_receipt
+
+
+genai.configure(api_key=os.environ.get("GOOGLE_API_KEY"))
+
+app = FastAPI()
+
+class OCRResponse(BaseModel):
+    extracted_text: str
+
+@app.post("/ocr-image/", response_model=OCRResponse)
+async def ocr_image(file: UploadFile = File(...)):
+    
+    if not file.content_type.startswith("image/"):
+        raise HTTPException(status_code=400, detail="Uploaded file must be an image.")
+
+    try:
+        image_data = await file.read()
+        image = Image.open(io.BytesIO(image_data))
+
+    
+        model = genai.GenerativeModel('gemini-2.0-flash')
+
+        prompt = "Extract and Identify all text from this image and generate a receipt without any additional information."
+        contents = [prompt, image]
+
+        response = model.generate_content(contents)
+
+        extracted_text = response.text
+
+        create_receipt(extracted_text, file_name="result/receipt.pdf")
+        return FileResponse(path="result/receipt.pdf", filename="receipt.pdf", media_type='application/pdf')
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"OCR failed: {str(e)}")
+

pdf_generator.py

@@ -0,0 +1,20 @@
+from reportlab.lib.pagesizes import A5
+from reportlab.pdfgen import canvas
+
+def create_receipt(text, file_name):
+    pdf = canvas.Canvas(file_name, pagesize=A5)
+    width, height = A5
+
+    pdf.setFont("Helvetica-Bold", 16)
+    pdf.drawCentredString(width / 2, height - 50, "Receipt Summary")
+
+    pdf.setFont("Helvetica", 12)
+    text_object = pdf.beginText(50, height - 100)
+
+    for line in text.split("\n"):
+        text_object.textLine(line.strip())
+
+    pdf.drawText(text_object)
+
+    pdf.save()
+

requirements.txt

@@ -0,0 +1,8 @@
+google-generativeai
+requests==2.31.0
+Pillow==10.4.0
+fastapi==0.100.0
+uvicorn==0.23.2
+python-multipart==0.0.6
+pydantic==2.11.4
+reportlab

result/receipt.pdf

@@ -0,0 +1,74 @@
+%PDF-1.3
+%���� ReportLab Generated PDF document http://www.reportlab.com
+1 0 obj
+<<
+/F1 2 0 R /F2 3 0 R
+>>
+endobj
+2 0 obj
+<<
+/BaseFont /Helvetica /Encoding /WinAnsiEncoding /Name /F1 /Subtype /Type1 /Type /Font
+>>
+endobj
+3 0 obj
+<<
+/BaseFont /Helvetica-Bold /Encoding /WinAnsiEncoding /Name /F2 /Subtype /Type1 /Type /Font
+>>
+endobj
+4 0 obj
+<<
+/Contents 8 0 R /MediaBox [ 0 0 419.5276 595.2756 ] /Parent 7 0 R /Resources <<
+/Font 1 0 R /ProcSet [ /PDF /Text /ImageB /ImageC /ImageI ]
+>> /Rotate 0 /Trans <<
+
+>> 
+  /Type /Page
+>>
+endobj
+5 0 obj
+<<
+/PageMode /UseNone /Pages 7 0 R /Type /Catalog
+>>
+endobj
+6 0 obj
+<<
+/Author (anonymous) /CreationDate (D:20251105143021+01'00') /Creator (ReportLab PDF Library - www.reportlab.com) /Keywords () /ModDate (D:20251105143021+01'00') /Producer (ReportLab PDF Library - www.reportlab.com) 
+  /Subject (unspecified) /Title (untitled) /Trapped /False
+>>
+endobj
+7 0 obj
+<<
+/Count 1 /Kids [ 4 0 R ] /Type /Pages
+>>
+endobj
+8 0 obj
+<<
+/Filter [ /ASCII85Decode /FlateDecode ] /Length 374
+>>
+stream
+GarW5>t`'h(kqGQ/'a\hUaDpXRs2G<;#D0a95e;:%PsQN]P3,tpSMNN;'[E?PNSI+&.&r(*7iO>G6P/;-"Ma>".?&IMC5K_ae3:.f7Q7_ZTh_:,%R:AE]0ISjXSNP.\[^ADcDm641kf62=E1pCf)ImOoJd2N)TlSr$3gb%WP'\%-hKT?b%B#;Km5H,UcoopEO,ZI-5=!?F1`G8(%j4Tlj'g6>ro^anm-:m5PY,6@H<=-JR*nM5_+S@'/,WOn57k44mrl?`o6%ksA*qd5]$#U=X3R65r"o-GHN>AkIG]6B&]+eRh=amg8H7LVhUGp2FH\qQm&=O+h@-L99m&CO4GcEG7&1V:]E?+]ocZnP>U2:-JiF+4Q4.DZ~>endstream
+endobj
+xref
+0 9
+0000000000 65535 f 
+0000000073 00000 n 
+0000000114 00000 n 
+0000000221 00000 n 
+0000000333 00000 n 
+0000000536 00000 n 
+0000000604 00000 n 
+0000000900 00000 n 
+0000000959 00000 n 
+trailer
+<<
+/ID 
+[<03cbf99983fa0b0281f96429a1239e03><03cbf99983fa0b0281f96429a1239e03>]
+% ReportLab generated PDF document -- digest (http://www.reportlab.com)
+
+/Info 6 0 R
+/Root 5 0 R
+/Size 9
+>>
+startxref
+1423
+%%EOF