Update app.py

app.py

@@ -233,7 +233,7 @@ def identify_headers_with_openrouter(pdf_path, model,LLM_prompt, pages_to_check=
 
 
 def identify_headers_and_save_excel(pdf_path, model, llm_prompt):
-    # This calls your existing header extraction function
+    # Call your existing function
     result = identify_headers_with_openrouter(pdf_path, model, llm_prompt)
     
     if not result:
@@ -241,12 +241,11 @@ def identify_headers_and_save_excel(pdf_path, model, llm_prompt):
     
     df = pd.DataFrame(result)
     
-    # Save to BytesIO
-    output = BytesIO()
-    df.to_excel(output, index=False, engine='openpyxl')
-    output.seek(0)  # reset pointer to start
+    # Save Excel to a file on disk
+    output_path = "output.xlsx"
+    df.to_excel(output_path, index=False, engine='openpyxl')
     
-    return output
+    return output_path  # return file path, not BytesIO
 
 iface = gr.Interface(
     fn=identify_headers_and_save_excel,
@@ -255,8 +254,7 @@ iface = gr.Interface(
         gr.Textbox(label="Model Type"),
         gr.Textbox(label="LLM Prompt")
     ],
-    outputs=gr.File(label="Download Excel")
+    outputs=gr.File(label="Download Excel")  # File expects a path
 )
 
-iface.launch()
-
+iface.launch()