Update app.py

app.py

@@ -5,7 +5,8 @@ import cv2
 import numpy as np
 import re
 import csv
-import io
+import tempfile
+import os
 
 # Synonyms for features
 feature_synonyms = {
@@ -88,21 +89,18 @@ def ocr_extract(image):
     # Parse the extracted text
     parsed_data = parse_text(text)
 
-    # Prepare CSV data as in-memory file
-    output = io.StringIO()
-    writer = csv.DictWriter(output, fieldnames=parsed_data.keys())
+    # Create a temp file to save CSV
+    tmpfile = tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode="w", encoding="utf-8")
+    writer = csv.DictWriter(tmpfile, fieldnames=parsed_data.keys())
     writer.writeheader()
     writer.writerow(parsed_data)
-    csv_str = output.getvalue()
-    output.close()
+    tmpfile_path = tmpfile.name
+    tmpfile.close()
 
-    csv_bytes = io.BytesIO(csv_str.encode("utf-8"))
-    csv_bytes.name = "extracted_bill.csv"
-
-    # Prepare display text
+    # Prepare display text as before
     display_text = "\n".join(f"{k}: {v}" for k, v in parsed_data.items())
 
-    return display_text, csv_bytes
+    return display_text, tmpfile_path  # Return path string, not BytesIO
 
 iface = gr.Interface(
     fn=ocr_extract,