Update app.py

app.py

@@ -8,25 +8,11 @@ from typing import List, Dict, Any
 
 MCP_SERVER_URL = "https://mcp-1st-birthday-project-conductor-core-mcp.hf.space/execute"
 
-# --- HARDCODED DEFAULTS (The Safety Net) ---
-# These populate the dropdowns if config.ini is missing or empty.
-DEFAULT_PROVIDER_MODELS = {
-    "Google Gemini": [
-        "gemini-2.0-flash-lite",
-        "gemini-1.5-pro-latest",
-        "gemini-1.5-flash-latest",
-        "gemini-1.0-pro"
-    ],
-    "Anthropic": [
-        "claude-3-opus-20240229",
-        "claude-3-sonnet-20240229",
-        "claude-3-haiku-20240307"
-    ],
-    "OpenAI": [
-        "gpt-4-turbo",
-        "gpt-4o",
-        "gpt-3.5-turbo"
-    ],
+# HARDCODED PROVIDER MAPPING (Directive 1.1, 1.2, 1.3)
+PROVIDER_MODEL_MAPPING = {
+    "Google Gemini": [],
+    "Anthropic": [],
+    "OpenAI": [],
     "Custom": []
 }
 
@@ -45,41 +31,40 @@ def classify_model_by_provider(model_label: str) -> str:
 
 def load_client_config():
     config = {}
+    coder_models = {}
+    strategist_models = {}
     
-    # 1. Start with the Hardcoded Defaults
-    # We use deep copy logic to ensure we don't mess up the original global dict
-    coder_by_provider = {k: v[:] for k, v in DEFAULT_PROVIDER_MODELS.items()}
-    strategist_by_provider = {k: v[:] for k, v in DEFAULT_PROVIDER_MODELS.items()}
-    
-    # 2. Try to load from config.ini to ADD extra models
     if os.path.exists('config.ini'):
         parser = ConfigParser()
         parser.read('config.ini')
         try:
             if 'CODER_MODELS' in parser:
-                for label, _ in parser['CODER_MODELS'].items():
-                    provider = classify_model_by_provider(label)
-                    # Only add if not already in the list to avoid duplicates
-                    if label not in coder_by_provider[provider]:
-                        coder_by_provider[provider].append(label)
-            
+                for label, entry in parser['CODER_MODELS'].items():
+                    coder_models[label] = label
             if 'STRATEGIST_MODELS' in parser:
-                for label, _ in parser['STRATEGIST_MODELS'].items():
-                    provider = classify_model_by_provider(label)
-                    if label not in strategist_by_provider[provider]:
-                        strategist_by_provider[provider].append(label)
+                for label, entry in parser['STRATEGIST_MODELS'].items():
+                    strategist_models[label] = label
         except Exception as e:
             print(f"Config parsing warning: {e}")
 
-    config['coder_by_provider'] = coder_by_provider
-    config['strategist_by_provider'] = strategist_by_provider
+    # Build provider-grouped structure
+    coder_by_provider = {p: [] for p in PROVIDER_MODEL_MAPPING.keys()}
+    strategist_by_provider = {p: [] for p in PROVIDER_MODEL_MAPPING.keys()}
+    
+    for label in coder_models.keys():
+        provider = classify_model_by_provider(label)
+        coder_by_provider[provider].append(label)
     
-    # Flatten lists for easy searching later
-    config['coder_models'] = [m for models in coder_by_provider.values() for m in models]
-    config['strategist_models'] = [m for models in strategist_by_provider.values() for m in models]
+    for label in strategist_models.keys():
+        provider = classify_model_by_provider(label)
+        strategist_by_provider[provider].append(label)
     
-    config['coder_default'] = os.getenv('CODER_MODEL_DEFAULT_LABEL', 'gemini-2.0-flash-lite')
-    config['strategist_default'] = os.getenv('STRATEGIST_MODEL_DEFAULT_LABEL', 'gemini-2.0-flash-lite')
+    config['coder_by_provider'] = coder_by_provider
+    config['strategist_by_provider'] = strategist_by_provider
+    config['coder_models'] = list(coder_models.keys())
+    config['strategist_models'] = list(strategist_models.keys())
+    config['coder_default'] = os.getenv('CODER_MODEL_DEFAULT_LABEL', '')
+    config['strategist_default'] = os.getenv('STRATEGIST_MODEL_DEFAULT_LABEL', '')
     
     return config
 
@@ -296,36 +281,11 @@ 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" 
-    
-    # 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" 
-        # 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)
+        return DEMO_BLUEPRINT, DEMO_INTENT, gr.update(value=demo_model_label)
 
     def start_execution_sequence():
         return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
@@ -356,8 +316,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 = initial_coder_model_val
-        strategist_reset_value = initial_strategist_model_val
+        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
         
         return (
             "",  # blueprint_input
@@ -383,14 +343,14 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             with gr.Column():
                 gr.Markdown("### Coder Agent")
                 coder_provider = gr.Dropdown(
-                    choices=list(DEFAULT_PROVIDER_MODELS.keys()),
+                    choices=list(PROVIDER_MODEL_MAPPING.keys()),
                     value=initial_coder_provider,
                     label="Provider",
                     interactive=True
                 )
                 coder_model = gr.Dropdown(
                     choices=CLIENT_CONFIG['coder_by_provider'].get(initial_coder_provider, []),
-                    value=initial_coder_model_val,
+                    value=coder_default if coder_default else None,
                     label="Model",
                     interactive=True
                 )
@@ -432,14 +392,14 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             with gr.Column():
                 gr.Markdown("### Strategist Agent")
                 strategist_provider = gr.Dropdown(
-                    choices=list(DEFAULT_PROVIDER_MODELS.keys()),
+                    choices=list(PROVIDER_MODEL_MAPPING.keys()),
                     value=initial_strategist_provider,
                     label="Provider",
                     interactive=True
                 )
                 strategist_model = gr.Dropdown(
                     choices=CLIENT_CONFIG['strategist_by_provider'].get(initial_strategist_provider, []),
-                    value=initial_strategist_model_val,
+                    value=strategist_default if strategist_default else None,
                     label="Model",
                     interactive=True
                 )