Create app.py

app.py

@@ -0,0 +1,94 @@
+import gradio as gr
+import easyocr
+import google.generativeai as genai
+import os, json, re
+from PIL import Image, ImageEnhance, ImageFilter
+
+# 🔐 API key
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+
+# OCR
+reader = easyocr.Reader(['en'])
+model = genai.GenerativeModel("gemini-3-flash-preview")
+
+
+def preprocess_image(path):
+    img = Image.open(path).convert("L")
+    img = ImageEnhance.Contrast(img).enhance(2)
+    img = img.filter(ImageFilter.SHARPEN)
+    return img
+
+
+def extract_text(image):
+    result = reader.readtext(image, detail=0)
+    text = "\n".join(result)
+
+    text = re.sub(r'\.{2,}', ' ', text)
+    text = re.sub(r'\s+', ' ', text)
+
+    return text
+
+
+def build_prompt(text):
+    return f"""
+You are a strict JSON generator.
+
+Convert menu text into VALID JSON.
+
+RULES:
+- Output ONLY JSON
+- Detect categories
+- Each item has "name" and "sizes"
+- 1 price → Regular
+- 2 prices → infer meaning
+- Fix OCR errors
+
+MENU:
+{text}
+"""
+
+
+def extract_json(text):
+    try:
+        start = text.index("{")
+        end = text.rindex("}") + 1
+        return json.loads(text[start:end])
+    except:
+        return None
+
+
+def validate_schema(data):
+    if not data or "categories" not in data:
+        return False
+    return True
+
+
+def generate_json(text):
+    for _ in range(3):
+        prompt = build_prompt(text)
+        response = model.generate_content(prompt)
+
+        parsed = extract_json(response.text)
+
+        if validate_schema(parsed):
+            return parsed
+
+    return {"error": "Failed to generate valid JSON"}
+
+
+def process(image):
+    img = preprocess_image(image)
+    text = extract_text(img)
+    result = generate_json(text)
+
+    return json.dumps(result, indent=2)
+
+
+app = gr.Interface(
+    fn=process,
+    inputs=gr.Image(type="filepath"),
+    outputs=gr.Code(language="json"),
+    title="🚀 Production Menu Parser (OCR + Gemini)",
+)
+
+app.launch()