test v2.1 new model

mistralocr_app_demo.py

@@ -400,8 +400,8 @@ Return ONLY the JSON object, without any surrounding text or markdown formatting
                         # Fallback: return the raw text wrapped in a basic JSON structure
                         pretty_text = json.dumps({"error": "Failed to parse Gemini JSON response", "raw_output": raw_json_text}, indent=2, ensure_ascii=False)
 
-                elif structure_model == "gpt-4o-mini":
-                    print(f"    - Using OpenAI GPT-4o mini...")
+                elif structure_model.startswith("gpt-"):
+                    print(f"    - Using OpenAI model: {structure_model}...")
                     if not openai_client:
                         print("    - ⚠️ OpenAI client not initialized. Skipping.")
                         return json.dumps({"error": "OpenAI client not initialized. Check API key and library installation."}, indent=2, ensure_ascii=False)
@@ -1101,9 +1101,17 @@ def create_gradio_interface():
                     )
                     structure_model = gr.Dropdown(
                         label="結構化模型 (用於圖片 OCR)", 
-                        choices=["pixtral-12b-latest", "gemini-2.0-flash", "gpt-4o-mini", "gpt-4o"], # Added gpt-4o
+                        choices=[
+                            ("pixtral-12b-latest", "pixtral-12b-latest"),
+                            ("gemini-2.0-flash", "gemini-2.0-flash"),
+                            ("gpt-4o-mini", "gpt-4o-mini"),
+                            ("gpt-4o", "gpt-4o"),
+                            ("gpt-4.1-nano (不建議使用，可能格式錯誤)", "gpt-4.1-nano"),
+                            ("gpt-4.1-mini", "gpt-4.1-mini"),
+                            ("gpt-4.1", "gpt-4.1")
+                        ], 
                         value="gemini-2.0-flash",
-                        info="選擇用於結構化圖片 OCR 結果的模型。需要對應的 API Key。" 
+                        info="選擇用於結構化圖片 OCR 結果的模型。需要對應的 API Key。"
                     )
                     structure_text_only = gr.Checkbox(
                         label="僅用文字進行結構化 (節省 Token)",
@@ -1116,8 +1124,11 @@ def create_gradio_interface():
                             "gemini-2.0-flash", 
                             "gemini-2.5-pro-exp-03-25", 
                             "gemini-2.0-flash-lite",
-                            "gpt-4o", # Added OpenAI models
-                            "gpt-4o-mini" 
+                            "gpt-4o", 
+                            "gpt-4o-mini",
+                            "gpt-4.1-nano",
+                            "gpt-4.1-mini",
+                            "gpt-4.1" 
                         ], 
                         value="gemini-2.0-flash"
                     )