Update app.py

app.py

@@ -86,7 +86,7 @@ def encode_image_to_data_url(img_path: str) -> str:
         b64 = base64.b64encode(f.read()).decode()
     return f"data:{mime};base64,{b64}"
 
-def call_gpt4o(image_paths):
+def call_gpt_model(model_name, image_paths):
     # 把所有圖片併在同一次對話傳給 GPT-4o
     messages = [{"role": "system", "content": system_prompt}]
     for p in image_paths:
@@ -100,23 +100,29 @@ def call_gpt4o(image_paths):
             }
         )
 
-    resp = client.chat.completions.create(
-        model=MODEL,
-        messages=messages,
-        response_format={
+    params = {
+        "model": model_name,
+        "messages": messages,
+        "response_format": {
             "type": "json_schema",
             "json_schema": product_schema
-        },
-    )
+        }
+    }
+
+    # ➕ 如果是 gpt-4o，才加入 temperature = 0.0
+    if model_name == "gpt-4o":
+        params["temperature"] = 0
+        
+    resp = client.chat.completions.create(**params)
 
     # 4o temperature=0,
 
     # API 已保證回傳一定符合 schema → 直接 loads
     return json.loads(resp.choices[0].message.content)
 
-def process(images):
+def process(images, model_name):
     paths = [img.name for img in images]
-    payload = call_gpt4o(paths)       # payload 是 {"products":[ {...}, ... ]}
+    payload = call_gpt_model(model_name, paths)       # payload 是 {"products":[ {...}, ... ]}
     items = payload.get("products", [])
 
     # (1) JSON pretty print
@@ -137,11 +143,17 @@ def process(images):
 with gr.Blocks(title="Price-Tag Parser") as demo:
     gr.Markdown("## 🏷️ 零售標價解析\n上傳一張或多張標價照片 → 取得 JSON 與 Excel")
     inp = gr.Files(label="上傳圖片 (可多選)", file_types=["image"])
+    model_selector = gr.Radio(
+        choices=["gpt-4o", "o3"],
+        value="gpt-4o",
+        label="選擇使用的模型"
+    )
+    
     btn = gr.Button("開始解析 🪄")
     out_json = gr.JSON(label="辨識結果 (JSON)")
     out_file = gr.File(label="下載 Excel", file_types=[".xlsx"])
 
-    btn.click(process, inputs=inp, outputs=[out_json, out_file])
+    btn.click(process, inputs=[inp, model_selector], outputs=[out_json, out_file])
 
 if __name__ == "__main__":
     demo.launch()