Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 import requests
 import os
+import random
 
 api_key = os.getenv("apichatbotacess")
 
 # Define the API calling function
-def api_call(prompt, thread_id, model_name, chat_history):
+def api_call(prompt, model_name, chat_history):
     url = "https://multimodalcompany-dev--talent-chatbot-beta-dev-dev.modal.run"
     headers = {
         "key": api_key,
@@ -13,7 +14,7 @@ def api_call(prompt, thread_id, model_name, chat_history):
     }
     data = {
         "prompt": prompt,
-        "thread_id": thread_id,
+        "thread_id": random.randint(1, 10000),  # Use a random number for thread_id
         "nameM": model_name,
         "max_new_tokens": 4096,
         "top_p": 0.95,
@@ -31,17 +32,29 @@ 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")
-    thread_id_input = gr.Textbox(label="Enter Thread ID", placeholder="Thread ID", lines=1)
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label="Your Message")
 
-    def respond(message, thread_id, chat_history, model_name):
-        bot_message = api_call(message, thread_id, model_name, chat_history)
+    def respond(message, model_name, chat_history):
+        bot_message = api_call(message, model_name, chat_history)
         chat_history.append((message, bot_message))
         return "", chat_history
 
-    msg.submit(respond, [msg, thread_id_input, gr.State([]), model_name], [msg, chatbot])
+    def reset_chat_history(model_name):
+        # Reset the chat history when the model_name changes
+        global chat_history
+        chat_history = []
+        return chat_history, []
+
+    # Link the reset function to the model_name dropdown
+    model_name.change(reset_chat_history, inputs=model_name, outputs=[chatbot, gr.State(chat_history)])
+
+    msg.submit(respond, [msg, model_name, gr.State(chat_history)], [msg, chatbot])
 
-demo.launch()
+demo.launch()