Update app.py

app.py

@@ -6,11 +6,13 @@ import json
 import time
 
 togetherai_base_url = "https://api.together.xyz/v1"
+
 supported_models = ["mistralai/Mixtral-8x7B-Instruct-v0.1",
                     "mistralai/Mixtral-8x22B-Instruct-v0.1",
                     "Qwen/Qwen1.5-1.8B-Chat",
                     "Qwen/Qwen1.5-14B-Chat",
-                    "Qwen/Qwen1.5-7B-Chat"
+                    "Qwen/Qwen1.5-7B-Chat",
+                    "gpt-3.5-turbo-0125"
                    ]
 
 multilingual_embeddings = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="jost/multilingual-e5-base-politics-de")
@@ -86,21 +88,31 @@ def predict(
         prompt = prompt_template.format(impersonation_template=impersonation_template, answer_option_template=answer_option_template, statement=political_statement[3:], rag_template=rag_template)
         print(prompt)
 
-    client = OpenAI(base_url=togetherai_base_url, api_key=togetherai_api_key)
-    
-    response1 = client.chat.completions.create(
-        model=model1,
-        messages=[{"role": "user", "content": prompt},],
-        temperature=temperature,
-        max_tokens=1000).choices[0].message.content
-        
-    response2 = client.chat.completions.create(
-        model=model2,
-        messages=[{"role": "user", "content": prompt},],
-        temperature=temperature,
-        max_tokens=1000).choices[0].message.content
+    responses = []
+    for model in [model1, model2]:
+        if model == "gpt-3.5-turbo-0125":
+            client = OpenAI(api_key=openai_api_key)
+            
+            response = client.chat.completions.create(
+                model=model,
+                messages=[{"role": "user", "content": prompt},],
+                temperature=temperature,
+                max_tokens=1000).choices[0].message.content
+
+            responses.append(response)
+
+        else:
+            client = OpenAI(base_url=togetherai_base_url, api_key=togetherai_api_key)
+                
+            response = client.chat.completions.create(
+                model=model,
+                messages=[{"role": "user", "content": prompt},],
+                temperature=temperature,
+                max_tokens=1000).choices[0].message.content
+
+            responses.append(response)
  
-    return response1, response2, prompt
+    return responses[0], responses[1], prompt
 
 def update_political_statement_options(test_type):
     # Append an index starting from 1 before each statement
@@ -196,7 +208,7 @@ def main():
 
             with gr.Row():
                 with gr.Accordion("Prompt Details", open=False):
-                    prompt_display = gr.Textbox(interactive=False, placeholder="Prompt used in the last submission will appear here.")
+                    prompt_display = gr.Textbox(show_label=False, interactive=False, placeholder="Prompt used in the last query will appear here.")
 
         with gr.Tab("Settings"):
             with gr.Row():