Update app.py

app.py

@@ -5,26 +5,6 @@ chat_histories = []
 current_chat = []
 saved_chats_choices = []
 
-class App:
-    def __init__(self):
-        with gr.Blocks() as self.demo:
-            self.chatbot = gr.Chatbot()
-            self.msg = gr.Textbox()
-            self.save_button = gr.Button("Save Chat History")
-            self.saved_chats_dropdown = gr.Dropdown(choices=saved_chats_choices, label="Saved Chat Histories:")
-
-            self.msg.submit(respond, [self.msg, self.chatbot], [self.msg, self.chatbot])
-            self.save_button.click(save_chat, [self.chatbot, self], [self.chatbot, self.saved_chats_dropdown])
-            self.saved_chats_dropdown.change(load_chat, [self.chatbot])
-
-    def update_dropdown(self, choices):
-        self.saved_chats_dropdown.set_choices(choices)
-
-    def launch(self):
-        self.demo.launch()
-
-app = App()
-
 def respond(message, chat_history):
     global current_chat
     message = message.lower()
@@ -33,16 +13,16 @@ def respond(message, chat_history):
     return "", chat_history
 
 # Save the current chat history
-def save_chat(chat_history, app_instance):
+def save_chat(chat_history, saved_chats_dropdown):
     global chat_histories, current_chat, saved_chats_choices
     title = f"Chat History {len(chat_histories) + 1}"
     chat_histories.append({'title': title, 'content': current_chat.copy()})
     current_chat.clear()
     saved_chats_choices.append(title)
-    
+    print(saved_chats_choices)
+    saved_chats_dropdown = gr.Dropdown(saved_chats_choices, label="Saved Chat Histories:",interactive=True)
     # Manually update the dropdown choices
-    app_instance.update_dropdown(saved_chats_choices)
-    return chat_history, saved_chats_choices
+    return chat_history, saved_chats_dropdown
 
 # Load a chat history when selected
 def load_chat(selection):
@@ -51,4 +31,14 @@ def load_chat(selection):
             return chat['content']
     return []
 
-app.launch()
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    save_button = gr.Button("Save Chat History")
+    saved_chats_dropdown = gr.Dropdown(saved_chats_choices, label="Saved Chat Histories:",interactive=True)
+
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    save_button.click(save_chat, [chatbot, saved_chats_dropdown], [chatbot, saved_chats_dropdown])
+    saved_chats_dropdown.change(load_chat, [chatbot])
+
+demo.launch()