Update app.py

app.py

@@ -4,8 +4,6 @@ import pytesseract
 import requests
 import pandas as pd
 import gradio as gr
-import io
-from io import BytesIO
 
 # ──────────────────────────────────────────────────────────────
 # 1. Utility: Detect rectangular contours (approximate book covers)
@@ -100,7 +98,7 @@ def query_openlibrary(title_text: str, author_text: str = None):
 def process_image(image_file):
     """
     Gradio passes a PIL image or numpy array. Convert to OpenCV BGR, detect covers → OCR → OpenLibrary.
-    Return a DataFrame and a (filename, BytesIO) tuple for CSV.
+    Return a DataFrame and CSV bytes (as raw bytes).
     """
     img = np.array(image_file)[:, :, ::-1].copy()  # PIL to OpenCV BGR
     boxes = detect_book_regions(img)
@@ -128,19 +126,15 @@ def process_image(image_file):
                 }
             )
 
+    # Build DataFrame
     if not records:
         df_empty = pd.DataFrame(columns=["title", "author_name", "publisher", "first_publish_year"])
-        # Build an empty CSV bytes buffer
-        empty_csv = df_empty.to_csv(index=False).encode()
-        buffer = io.BytesIO(empty_csv)
-        buffer.name = "books.csv"
-        return df_empty, buffer
+        csv_bytes = df_empty.to_csv(index=False).encode()
+        return df_empty, csv_bytes
 
     df = pd.DataFrame(records)
     csv_bytes = df.to_csv(index=False).encode()
-    buffer = io.BytesIO(csv_bytes)
-    buffer.name = "books.csv"
-    return df, buffer
+    return df, csv_bytes
 
 # ──────────────────────────────────────────────────────────────
 # 5. Build the Gradio Interface
@@ -168,8 +162,9 @@ def build_interface():
         download_file = gr.File(label="Download CSV")
 
         def on_run(image):
-            df, file_buffer = process_image(image)
-            return df, file_buffer
+            df, csv_bytes = process_image(image)
+            # Return DataFrame plus a dict that gr.File understands:
+            return df, {"name": "books.csv", "data": csv_bytes}
 
         run_button.click(
             fn=on_run,
@@ -181,4 +176,5 @@ def build_interface():
 
 if __name__ == "__main__":
     demo_app = build_interface()
+    # You can add share=True if you want a public link; otherwise this is fine:
     demo_app.launch()