Update app.py

app.py

@@ -32,17 +32,14 @@ def api_call(prompt, thread_id, model_name, chat_history):
     response = requests.post(url, headers=headers, json=data)
     return response.json()["message"]
 
-# Initialize the chat history
-chat_history = []
-
 # Gradio interface
 with gr.Blocks() as demo:
     model_name = gr.Dropdown(choices=["OpenAI", "Claude"], label="Select Model")
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label="Your Message")
+    state = gr.State([])  # Initialize state for chat history
 
     def respond(message, model_name, chat_history):
-        global chat_history
         # Append the user's message to the chat history immediately
         chat_history.append((message, None))
         # Get the bot's response
@@ -52,13 +49,12 @@ with gr.Blocks() as demo:
         return "", chat_history
 
     def reset_chat_history(model_name):
-        global chat_history
-        chat_history = []
-        return [], chat_history
+        # Reset the chat history when the model_name changes
+        return [], []
 
     # Link the reset function to the model_name dropdown
-    model_name.change(fn=reset_chat_history, inputs=model_name, outputs=[chatbot, gr.State(chat_history)])
+    model_name.change(fn=reset_chat_history, inputs=model_name, outputs=[chatbot, state])
 
-    msg.submit(fn=respond, inputs=[msg, model_name, gr.State(chat_history)], outputs=[msg, chatbot])
+    msg.submit(fn=respond, inputs=[msg, model_name, state], outputs=[msg, chatbot])
 
 demo.launch(share=True)