Update app.py

app.py

@@ -281,11 +281,36 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
     if strategist_default:
         initial_strategist_provider = classify_model_by_provider(strategist_default)
 
+    # TASK 1: Set Default Models on Launch
+    # We override the config default if specific models are requested
+    default_model_label = "gemini-2.0-flash-lite" # Assumes this key exists in config
+    
+    # Helper to safely get a default value
+    def get_safe_default(provider, preferred_label):
+        models = CLIENT_CONFIG['coder_by_provider'].get(provider, [])
+        if preferred_label in models:
+            return preferred_label
+        return models[0] if models else None
+
+    initial_coder_model_val = get_safe_default(initial_coder_provider, default_model_label)
+    initial_strategist_model_val = get_safe_default(initial_strategist_provider, default_model_label)
+
     def load_demo_data():
         # TASK 2: "Load Demo" Button Model Switch
         # Switch to a high-end model for the demo
-        demo_model_label = "gemini-2.5-pro" 
-        return DEMO_BLUEPRINT, DEMO_INTENT, gr.update(value=demo_model_label)
+        demo_model_label = "gemini-2.5-pro" # Assumes this key exists or will be handled
+        # Fallback if exact label not found, try to find a 'pro' model
+        models = CLIENT_CONFIG['coder_by_provider'].get("Google Gemini", [])
+        target_model = demo_model_label
+        if demo_model_label not in models:
+             # Try to find a pro model
+             pro_models = [m for m in models if 'pro' in m.lower()]
+             if pro_models:
+                 target_model = pro_models[0]
+             elif models:
+                 target_model = models[0]
+
+        return DEMO_BLUEPRINT, DEMO_INTENT, gr.update(value=target_model)
 
     def start_execution_sequence():
         return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
@@ -316,8 +341,8 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
         coder_models_for_reset = CLIENT_CONFIG['coder_by_provider'].get(initial_coder_provider, [])
         strategist_models_for_reset = CLIENT_CONFIG['strategist_by_provider'].get(initial_strategist_provider, [])
         
-        coder_reset_value = coder_models_for_reset[0] if coder_models_for_reset else None
-        strategist_reset_value = strategist_models_for_reset[0] if strategist_models_for_reset else None
+        coder_reset_value = initial_coder_model_val
+        strategist_reset_value = initial_strategist_model_val
         
         return (
             "",  # blueprint_input
@@ -350,7 +375,7 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
                 )
                 coder_model = gr.Dropdown(
                     choices=CLIENT_CONFIG['coder_by_provider'].get(initial_coder_provider, []),
-                    value=coder_default if coder_default else None,
+                    value=initial_coder_model_val,
                     label="Model",
                     interactive=True
                 )
@@ -399,7 +424,7 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
                 )
                 strategist_model = gr.Dropdown(
                     choices=CLIENT_CONFIG['strategist_by_provider'].get(initial_strategist_provider, []),
-                    value=strategist_default if strategist_default else None,
+                    value=initial_strategist_model_val,
                     label="Model",
                     interactive=True
                 )