add GPT-5

app.py

@@ -24,27 +24,26 @@ AUTHORIZED_USER_IDS = set(authorized_users_str.split(',') if authorized_users_st
 
 # Define model pricing information (approximate costs per 100 image API calls)
 MODEL_PRICING = {
-    "google/gemini-2.0-flash-001": "$0.03",
+    "google/gemini-2.5-flash": "$0.08",
     "gpt-4.1-mini": "$0.07",
     "gpt-4.1": "$0.35",
-    "anthropic/claude-3.7-sonnet": "$0.70",
-    "google/gemini-2.5-pro-preview": "$1.20",
-    "google/gemini-2.5-flash-preview-05-20:thinking": "$0.35",
+    "anthropic/claude-sonnet-4": "$0.70",
+    "google/gemini-2.5-pro": "$1.20",
     "gpt-4.1-nano": "$0.02",
     "openai/chatgpt-4o-latest": "$0.75",
     "meta-llama/llama-4-maverick": "$0.04",
     "meta-llama/llama-4-maverick:free": "Free",
-    "openai/gpt-5-chat": "N/A"
+    "openai/gpt-5-chat": "N/A",
+    "openai/gpt-5-mini": "N/A"
 }
 
 # Define preferred and additional models directly in the function
 preferred_models_auth = [
-    ("Gemini 2.0 Flash (cheap)", "google/gemini-2.0-flash-001"),
+    ("Gemini 2.5 Flash", "google/gemini-2.5-flash"),
     ("GPT-4.1 Mini", "gpt-4.1-mini"),
-    ("GPT-4.1 (Recommended)", "gpt-4.1"),
-    ("Claude 3.7 Sonnet", "anthropic/claude-3.7-sonnet"),
-    ("Gemini 2.5 Pro", "google/gemini-2.5-pro-preview"),
-    ("Gemini 2.5 Flash Thinking (Recommended)","google/gemini-2.5-flash-preview-05-20:thinking"),
+    ("GPT-4.1", "gpt-4.1"),
+    ("Claude Sonnet 4", "anthropic/claude-sonnet-4"),
+    ("Gemini 2.5 Pro", "google/gemini-2.5-pro"),
     ("openai/gpt-5-chat", "GPT-5-chat")
 ]
 
@@ -52,6 +51,7 @@ additional_models = [
     ("GPT-4.1 Nano", "gpt-4.1-nano"),
     ("ChatGPT Latest", "openai/chatgpt-4o-latest"),
     ("Llama 4 Maverick", "meta-llama/llama-4-maverick")
+    ("GPT-5-mini", "openai/gpt-5-mini")
 ]
 
 # Calculate all models once
@@ -167,10 +167,10 @@ def create_demo():
                     is_authorized = profile.username in AUTHORIZED_USER_IDS
                     if is_authorized:
                 
-                        text = f"""**Current Model**: Gemini 2.0 Flash (cheap) 
-                    **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.0-flash-001"]}"""
+                        text = f"""**Current Model**: Gemini 2.5 Flash
+                    **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.5-flash"]}"""
                         
-                        return gr.update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.0-flash-001"),text,f"Logged in as: {profile.username}"
+                        return gr.update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.5-flash"),text,f"Logged in as: {profile.username}"
                     else:
                         # Default model value
                         default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
@@ -296,7 +296,7 @@ def create_demo():
                     return gr.Dropdown(choices=preferred_choices, value=current_model)
                 else:
                     # Reset to default model if current model is not in preferred models
-                    return gr.Dropdown(choices=preferred_choices, value="google/gemini-2.0-flash-001")
+                    return gr.Dropdown(choices=preferred_choices, value="google/gemini-2.5-flash")
 
         # Update model info when model selection changes
         def update_model_info(model_value):