Update app.py

app.py

@@ -88,11 +88,15 @@ def extract_db_to_excel():
     data = [{"page_content": doc.page_content, "metadata": json.dumps(doc.metadata)} for doc in documents]
     df = pd.DataFrame(data)
     
-    excel_path = "database_output.xlsx"
-    df.to_excel(excel_path, index=False)
+    # Save DataFrame to an in-memory Excel file
+    output = io.BytesIO()
+    with pd.ExcelWriter(output, engine='openpyxl') as writer:
+        df.to_excel(writer, index=False)
+    output.seek(0)
     
-    return f"Database extracted to {excel_path}"
+    return output
 
+# Modify the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Chat with your PDF documents")
     
@@ -111,8 +115,8 @@ with gr.Blocks() as demo:
     submit_button.click(ask_question, inputs=[question_input], outputs=answer_output)
     
     extract_button = gr.Button("Extract Database to Excel")
-    extract_output = gr.Textbox(label="Extraction Status")
-    extract_button.click(extract_db_to_excel, inputs=[], outputs=extract_output)
+    excel_output = gr.File(label="Download Excel File")
+    extract_button.click(extract_db_to_excel, inputs=[], outputs=excel_output)
 
 if __name__ == "__main__":
     demo.launch()