Update app.py

app.py

@@ -55,6 +55,27 @@ def init_db():
 
 init_db()
 
+# --- Dynamic Model Dropdown Refresher Logic ---
+GROQ_MODELS = [
+    "llama-3.1-8b-instant", 
+    "llama-3.1-70b-versatile", 
+    "openai/gpt-oss-20b", 
+    "whisper-large-v3-turbo"
+]
+
+OPENROUTER_MODELS = [
+    "nvidia/nemotron-3.5-content-safety:free",
+    "qwen/qwen3.7-plus",
+    "microsoft/mai-transcribe-1.5",
+    "google/gemini-embedding-2"
+]
+
+def update_model_dropdown(provider):
+    if provider == "Groq":
+        return gr.Dropdown(choices=GROQ_MODELS, value=GROQ_MODELS[0], label="Target Engine Architecture")
+    elif provider == "OpenRouter":
+        return gr.Dropdown(choices=OPENROUTER_MODELS, value=OPENROUTER_MODELS[0], label="Target Engine Architecture")
+
 # --- Common LLM API Request Orchestrator ---
 def call_llm(provider, api_key, model_choice, system_prompt, user_message, chat_history_format=None):
     messages = [{"role": "system", "content": system_prompt}]
@@ -80,15 +101,16 @@ def call_llm(provider, api_key, model_choice, system_prompt, user_message, chat_
 
     elif provider == "OpenRouter":
         headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+        
+        # Maps user-facing selection directly to their OpenRouter global endpoints
         openrouter_model_map = {
-            "llama-3.1-8b-instant": "meta-llama/llama-3.1-8b-instruct",
-            "llama-3.1-70b-versatile": "meta-llama/llama-3.1-70b-instruct",
-            "openai/gpt-oss-20b": "openai/gpt-3.5-turbo",
-            "whisper-large-v3-turbo": "meta-llama/llama-3.1-8b-instruct"
+            "nvidia/nemotron-3.5-content-safety:free": "nvidia/nemotron-3.5-content-safety:free",
+            "qwen/qwen3.7-plus": "qwen/qwen3.7-plus",
+            "microsoft/mai-transcribe-1.5": "microsoft/mai-transcribe-1.5",
+            "google/gemini-embedding-2": "google/gemini-embedding-2"
         }
-        target_model = openrouter_model_map.get(model_choice, "meta-llama/llama-3.1-8b-instruct")
+        target_model = openrouter_model_map.get(model_choice, "nvidia/nemotron-3.5-content-safety:free")
         
-        # FIXED: Corrected structural syntax for payload dictionary assignment
         payload = {
             "model": target_model, 
             "messages": messages, 
@@ -199,11 +221,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             gr.Markdown("### 🔑 Global Control Tower")
             provider_select = gr.Dropdown(choices=["Groq", "OpenRouter"], value="Groq", label="API Gateway Provider")
             token_input = gr.Textbox(label="User API Secret Key", type="password", placeholder="gsk_... or sk-or-...")
+            
+            # This component gets updated dynamically by the event handler below
             model_select = gr.Dropdown(
-                choices=["llama-3.1-8b-instant", "llama-3.1-70b-versatile", "openai/gpt-oss-20b", "whisper-large-v3-turbo"], 
-                value="llama-3.1-8b-instant", 
+                choices=GROQ_MODELS, 
+                value=GROQ_MODELS[0], 
                 label="Target Engine Architecture"
             )
+            
             gr.Markdown("✨ **Global Platform Database State Schema:**\n- `phase` (Discovery, Data Prep, Modeling, Validation)\n- `task_name` (Structural workflow description string)\n- `owner` (Assigned personnel scientist name)\n- `timeline` (Expected operational delivery time frames)\n- `priority` (High, Medium, Low)")
             
         with gr.Column(scale=2):
@@ -214,7 +239,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     gr.Markdown("Chat with this engine normally, or tell it to log a brand-new task assignment milestone directly into the active SQLite infrastructure.")
                     
                     chatbot_viewport = gr.Chatbot(type="messages", label="Operational History Workspace")
-                    chat_input = gr.Textbox(placeholder="Say hello, or submit task details to log (e.g., Log a task in Validation phase named Run final audit owned by M. Reddy for Week 8 with High priority)", label="Your Message")
+                    chat_input = gr.Textbox(placeholder="Say hello, or submit task details to log...", label="Your Message")
                     send_btn = gr.Button("Submit Message", variant="primary")
                     
                     send_btn.click(
@@ -242,5 +267,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                         outputs=[output_data_table, sql_status_display]
                     )
 
+    # --- REACTIVE EVENT LISTENER ---
+    # Whenever the provider dropdown changes, change the choices of the model select dropdown
+    provider_select.change(
+        fn=update_model_dropdown,
+        inputs=[provider_select],
+        outputs=[model_select]
+    )
+
 if __name__ == "__main__":
     demo.launch()