Advanced settings

app.py

@@ -37,14 +37,14 @@ os.makedirs(os.path.dirname(PREFERENCES_FILE), exist_ok=True)
 
 # Define model pricing information (approximate costs per 100 image API calls)
 MODEL_PRICING = {
-    "google/gemini-2.0-flash-001": "$0.50",
-    "gpt-4.1-mini": "$2.00",
-    "gpt-4.1": "$5.00",
-    "anthropic/claude-3.7-sonnet": "$4.00",
-    "google/gemini-2.5-pro-preview-03-25": "$1.50",
-    "google/gemini-2.5-flash-preview:thinking": "$0.75",
-    "gpt-4.1-nano": "$1.00",
-    "openai/chatgpt-4o-latest": "$4.50",
+    "google/gemini-2.0-flash-001": "$0.03",
+    "gpt-4.1-mini": "$0.07",
+    "gpt-4.1": "$0.35",
+    "anthropic/claude-3.7-sonnet": "$0.70",
+    "google/gemini-2.5-pro-preview-03-25": "$1.20",
+    "google/gemini-2.5-flash-preview:thinking": "$0.35",
+    "gpt-4.1-nano": "$0.02",
+    "openai/chatgpt-4o-latest": "$0.75",
     "meta-llama/llama-4-maverick": "$0.25"
 }
 
@@ -239,15 +239,21 @@ def create_demo():
                 all_images = gr.State([])
                 all_results = gr.State([])
         
-        # Define functions within the demo function to avoid scope issues
-        
-        # Handle checkbox change to update model dropdown
-        def toggle_models(show_all):
+       # Handle checkbox change to update model dropdown
+        def toggle_models(show_all, current_model):
             if show_all:
-                return gr.Dropdown.update(choices=all_models)
+                return gr.Dropdown.update(choices=all_models, value=current_model)
             else:
-                return gr.Dropdown.update(choices=preferred_models)
-        
+                # Check if current model is in preferred models list
+                is_preferred = any(current_model == value for _, value in preferred_models)
+                
+                if is_preferred:
+                    # Keep the current model if it's in preferred models
+                    return gr.Dropdown.update(choices=preferred_models, value=current_model)
+                else:
+                    # Reset to default model if current model is not in preferred models
+                    return gr.Dropdown.update(choices=preferred_models, value=default_model)
+
         # Update model info when model selection changes
         def update_model_info(model_value):
             # Find display name
@@ -268,7 +274,7 @@ def create_demo():
         # Connect checkbox to toggle model choices
         show_all_models.change(
             fn=toggle_models,
-            inputs=[show_all_models],
+            inputs=[show_all_models, model_choice],
             outputs=[model_choice]
         )