Update app.py

app.py

@@ -85,6 +85,10 @@ def save_conversation(history, file_format):
     file_name = save_to_file(history, file_format)
     return file_name
 
+def save_and_download(history, file_format):
+    file_name = save_conversation(history, file_format)
+    return file_name
+
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
@@ -94,10 +98,29 @@ demo = gr.ChatInterface(
         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P"),
         gr.Radio(["PDF", "DOCX", "TXT"], label="Save As"),
     ],
-    button_fn=save_conversation,
     css=css,
     theme="allenai/gradio-theme",
 )
 
+save_button = gr.Button("Save Conversation")
+output_file = gr.File(label="Download File")
+
+def handle_save(history, save_format):
+    return save_conversation(history, save_format)
+
+save_button.click(
+    handle_save,
+    inputs=[demo.history, demo.inputs[-1]],  # Passing history and format
+    outputs=output_file
+)
+
+demo = gr.Blocks()
+
+with demo:
+    with gr.Column():
+        demo.render()
+        save_button.render()
+        output_file.render()
+
 if __name__ == "__main__":
     demo.launch()