Update app.py

app.py

@@ -119,16 +119,16 @@ def handle_file_save(history, file_format):
 
 # Handler function for Gradio app
 def save_handler(message, history, system_message, max_tokens, temperature, top_p, file_format):
-    response = respond(message, history, system_message, max_tokens, temperature, top_p)
-    saved_file = handle_file_save(history, file_format)
-    return saved_file
+    new_history = history + [(message, next(respond(message, history, system_message, max_tokens, temperature, top_p)))]
+    saved_file = handle_file_save(new_history, file_format)
+    return saved_file, new_history
 
 # Gradio interface
 demo = gr.Interface(
     fn=save_handler,
     inputs=[
         gr.Textbox(value="", label="Message"),
-        gr.State(),
+        gr.State([]),  # Initialize state as an empty list
         gr.Textbox(value="", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
@@ -144,7 +144,7 @@ demo = gr.Interface(
             label="Save as",
         ),
     ],
-    outputs=gr.File(label="Download Chat History"),
+    outputs=[gr.File(label="Download Chat History"), gr.State()],
     css=css,
     theme="allenai/gradio-theme",
 )