Update app.py

app.py

@@ -405,15 +405,23 @@ def process_message(messages):
     html_output += '</div>'  # 关闭最外层div
     return html_output
 
-def o3_chat(model_name, api_key, base_url, question, image):
+def o3_chat(model_name, client_type, api_key, base_url, question, image):
     print("done!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!")
     # 初始化组件
-    client = AzureOpenAI(
-        api_key=os.getenv("AZURE_OPENAI_API_KEY"),  # 从环境变量读取
-        api_version="2025-04-01-preview",          # 使用最新API版本
-        azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT")  # 终结点URL
-    )
-    # client = OpenAI(api_key=api_key, base_url=base_url)
+    # client = AzureOpenAI(
+    #     api_key=os.getenv("AZURE_OPENAI_API_KEY"),  # 从环境变量读取
+    #     api_version="2025-04-01-preview",          # 使用最新API版本
+    #     azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT")  # 终结点URL
+    # )
+
+    if client_type == "Azure":
+        client = AzureOpenAI(
+            api_key=api_key,  # 从环境变量读取
+            api_version="2025-04-01-preview",          # 使用最新API版本
+            azure_endpoint=base_url  # 终结点URL
+        )
+    else:
+        client = OpenAI(api_key=api_key, base_url=base_url)
     executor = PythonExecutor()
 
     prompt_template = json.load(open("./prompt_template_vis.json", "r", encoding="utf-8"))
@@ -448,6 +456,7 @@ def create_demo():
                     choices=["gpt-4.1"],
                     value="gpt-4.1"
                 )
+                client_type = gr.Dropdown(label="Client Type Selection", choices=["Azure", "OpenAI"], value="Azure")
                 api_key = gr.Textbox(label="OpenAI API Key (optional)", type="password", value=os.getenv("AZURE_OPENAI_API_KEY"))
                 base_url = gr.Textbox(label="Base URL (optional)", value=os.getenv("AZURE_OPENAI_ENDPOINT"))
                 image_input = gr.Image(label="Upload Image", type="pil")
@@ -479,7 +488,7 @@ def create_demo():
 
         submit_btn.click(
             fn=handle_submit,
-            inputs=[model_name, api_key, base_url, question, image_input],
+            inputs=[model_name, client_type, api_key, base_url, question, image_input],
             outputs=[output, extracted_images, extracted_json, current_images, current_json]
         )