Update app.py

app.py

@@ -24,9 +24,8 @@ with gr.Blocks() as demo:
         current_chat_id += 1
         chat_name = f"chat history {current_chat_id}"
         chat_histories[chat_name] = chat_history
-        # Update display_chat_names with all chat names
-        display_chat_names.value = "\n".join(chat_histories.keys())
-        return f"Saved as {chat_name}"
+        chat_names = "\n".join(chat_histories.keys())
+        return f"Saved as {chat_name}", chat_names
 
     def load_chat(entered_chat_name):
         # Fetch the specific chat history based on provided name
@@ -37,7 +36,7 @@ with gr.Blocks() as demo:
             return "Chat not found!", []
 
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    save_button.click(save_chat, [chatbot])
+    save_button.click(save_chat, [chatbot], outputs=[chatbot, display_chat_names])  # Added display_chat_names as an output
     load_button.click(load_chat, [enter_chat_name], outputs=[chatbot])
 
 demo.launch()