Update app.py

app.py

@@ -100,7 +100,7 @@ def update_messages_with_text(messages, generated_content):
     messages.append(message_item)
     return messages
 
-def call_chatgpt_api(messages, client, max_tokens=10000, stop=None, temperature=0.6):
+def call_chatgpt_api(model_name, messages, client, max_tokens=10000, stop=None, temperature=0.6):
     """Call ChatGPT API with the given messages"""
     try:
         response = client.chat.completions.create(
@@ -133,12 +133,13 @@ def call_chatgpt_api(messages, client, max_tokens=10000, stop=None, temperature=
         print(f"API Error: {str(e)}")
         return None, None
 
-def evaluate_single_data(data, client, executor, prompt_template, prompt_type):
+def evaluate_single_data(model_name, data, client, executor, prompt_template, prompt_type):
 
     messages = process_prompt_init(data["question"], data['image'], prompt_template, prompt_type)
     
     # 生成初始响应
     response_text, pred_stop_reason = call_chatgpt_api(
+        model_name,
         messages, 
         client,
         max_tokens=10000,
@@ -240,7 +241,7 @@ def process_message(messages):
     html_output += '</div>'  # 关闭最外层div
     return html_output
 
-def o3_chat(api_key, base_url, question, image):
+def o3_chat(model_name, api_key, base_url, question, image):
     print("done!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!")
     # 初始化组件
     client = OpenAI(api_key=api_key, base_url=base_url)
@@ -256,7 +257,7 @@ def o3_chat(api_key, base_url, question, image):
     }
     
     # 评估单个数据点
-    messages = evaluate_single_data(data, client, executor, prompt_template, prompt_type)
+    messages = evaluate_single_data(model_name, data, client, executor, prompt_template, prompt_type)
     html_output = process_message(messages)
     
     # 将消息转换为JSON字符串，用于下载
@@ -273,6 +274,11 @@ def create_demo():
         
         with gr.Row():
             with gr.Column(scale=1):
+                model_name = gr.Dropdown(
+                    label="Model Selection",
+                    choices=["GPT-4.1", "GPT-4o", "o4-mini", "gemini-2.5-pro-preview-05-06", "claude-3-7-sonnet-latest", "claude-3-7-sonnet-thinking"],
+                    value="GPT-4.1"
+                )
                 api_key = gr.Textbox(label="OpenAI API Key", type="password", value="sk-kBQuM0gvNBhOHmKz43b3iQut01bsOgg8Pv76eMKguu6jvncm")
                 base_url = gr.Textbox(label="Base URL (optional)", value="https://api.claudeshop.top/v1")
                 image_input = gr.Image(label="Upload Image", type="pil")
@@ -285,7 +291,7 @@ def create_demo():
         # 处理提交
         submit_btn.click(
             fn=o3_chat,
-            inputs=[api_key, base_url, question, image_input],
+            inputs=[model_name, api_key, base_url, question, image_input],
             outputs=[output]
         )