Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 api_key = os.getenv("apichatbotacess")
 
 # Define the API calling function
-def api_call(prompt, thread_id):
+def api_call(prompt, thread_id, model_name):
     url = "https://multimodalcompany-dev--talent-chatbot-beta-dev-dev.modal.run"
     headers = {
         "key": api_key,
@@ -15,6 +15,7 @@ def api_call(prompt, thread_id):
     data = {
         "prompt": prompt,
         "thread_id": thread_id,
+        "model_name": model_name,  # Add model_name to the data
         "max_new_tokens": 4096,
         "top_p": 0.95,
         "temperature": 0.7,
@@ -31,15 +32,16 @@ def api_call(prompt, thread_id):
     return response.json()["message"]
 
 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):
-        bot_message = api_call(message, thread_id)
+    def respond(message, thread_id, chat_history, model_name):
+        bot_message = api_call(message, thread_id, model_name)
         chat_history.append((message, bot_message))
         return "", chat_history
 
-    msg.submit(respond, [msg, thread_id_input, chatbot], [msg, chatbot])
+    msg.submit(respond, [msg, thread_id_input, chatbot, model_name], [msg, chatbot])
 
 demo.launch(share=True)