Update app.py

app.py

@@ -5,6 +5,9 @@ from PIL import Image
 import gradio as gr
 import io
 
+# Uncomment and edit this path if you’re on Windows
+# pytesseract.pytesseract.tesseract_cmd = r"C:\Program Files\Tesseract-OCR\tesseract.exe"
+
 # IBS Café schema columns
 COLUMNS = [
     "Parent Category", "Category", "Store Item Name", "Item Code", "Master Item Name", "EAN Code",
@@ -48,7 +51,6 @@ def parse_menu_text(text):
         if prices:
             name = re.sub(price_pattern, '', line).strip(" -:–")
             if '/' in line and len(prices) > 1:
-                # multi-size e.g., 149/199
                 for i, price in enumerate(prices):
                     size_label = f" ({['Regular', 'Large', 'XL'][i]})" if i < 3 else f" (Option {i+1})"
                     store_name = f"{name}{size_label}"
@@ -83,28 +85,31 @@ def ocr_and_extract(image):
     rows = parse_menu_text(text)
     
     df = pd.DataFrame(rows, columns=COLUMNS)
-    csv_buffer = io.StringIO()
-    df.to_csv(csv_buffer, index=False)
     
+    # Save Excel file in memory
+    excel_buffer = io.BytesIO()
+    with pd.ExcelWriter(excel_buffer, engine='openpyxl') as writer:
+        df.to_excel(writer, index=False, sheet_name="Menu")
+    excel_buffer.seek(0)
+
     json_output = {"rows": rows, "needs_review": []}
-    return text, csv_buffer.getvalue(), json_output
+    return text, excel_buffer, json_output
 
 # Gradio UI
 with gr.Blocks(title="Menu → IBS Schema Extractor") as demo:
-    gr.Markdown("## 🧾 Menu OCR to IBS Café Schema")
-    gr.Markdown("Upload a menu image and extract structured data in CSV + JSON formats.")
+    gr.Markdown("## 🧾 Menu OCR → IBS Café Excel Generator")
+    gr.Markdown("Upload a menu image and extract structured data in Excel (.xlsx) + JSON formats.")
     
     with gr.Row():
         image_input = gr.Image(type="filepath", label="Upload Menu Image")
     
     extract_btn = gr.Button("Extract")
-    
+
     with gr.Tab("Extracted Text"):
         text_output = gr.Textbox(label="OCR Text", lines=10)
     
-    with gr.Tab("CSV Output"):
-        csv_output = gr.Textbox(label="CSV Data", lines=10)
-        csv_file = gr.File(label="Download CSV")
+    with gr.Tab("Excel Output"):
+        excel_file = gr.File(label="Download Excel (.xlsx)")
     
     with gr.Tab("JSON Output"):
         json_output = gr.JSON(label="Structured JSON")
@@ -112,7 +117,7 @@ with gr.Blocks(title="Menu → IBS Schema Extractor") as demo:
     extract_btn.click(
         ocr_and_extract,
         inputs=[image_input],
-        outputs=[text_output, csv_output, json_output]
+        outputs=[text_output, excel_file, json_output]
     )
 
 demo.launch()