fix: update model list

app.py

@@ -3,21 +3,32 @@ import requests
 
 prefix = "https://smartfeed-custom-tools.hf.space/gradio_api/file="
 
-extract_models = [
+extract_model_tag_list = [
     "o3",
+    "o3(poe)",
     "gpt-5",
-    "o4-mini",
-    "o4-mini-high",
+    "o4-mini(openrouter)",
+    "o4-mini-high(openrouter)",
 ]
 
+extract_model_map = {
+    "o3": "o3-2025-04-16",
+    "o3(poe)": "o3(poe)",
+    "gpt-5": "gpt-5-2025-08-07",
+    "o4-mini(openrouter)": "openai/o4-mini",
+    "o4-mini-high(openrouter)": "openai/o4-mini-high",
+}
+
 
 def change_image_style(
     image_url,
     style_image_url,
-    extract_model,
+    extract_model_tag,
     extract_prompt,
     prompt_prefix
 ):
+    extract_model = extract_model_map.get(extract_model_tag, "o3")
+
     data = {
         "image_url": image_url,
         "style_image_url": style_image_url,
@@ -90,7 +101,7 @@ with gr.Blocks() as demo:
             with gr.Column(visible=False) as clear_button:
                 remove_and_reupload = gr.ClearButton(value="Remove and upload new ones", components=files, size="sm")
             input_image = gr.Image(label="Input Image", type="filepath", interactive=True)
-            extract_model = gr.Dropdown(choices=extract_models, label="Extract Model", value="o3")
+            extract_model = gr.Dropdown(choices=extract_model_tag_list, label="Extract Model", value="o3")
             extract_prompt = gr.Textbox(lines=2, label="Extracted Prompt", value="Accurately extract everything from this photo into a prompt especially the hairstyle excluding the backgrounds. I want to replicate it perfectly. need to be a complete paragraph")
             prompt_prefix = gr.Textbox(lines=1, label="Prompt Prefix", value="change the backgrounds to match this:")