Update app.py

app.py

@@ -30,9 +30,13 @@ try:
     print("model_logic.py loaded successfully.")
 except ImportError:
     print("Warning: Local modules (build_logic.py, model_logic.py) not found. Using dummy functions.")
-    def get_available_providers(): return ["DummyProvider"]
-    def get_models_for_provider(p): return ["dummy-model"]
-    def get_default_model_for_provider(p): return "dummy-model"
+    def get_available_providers(): return ["DummyProvider", "Groq"] # Added Groq for testing
+    def get_models_for_provider(p):
+        if p == 'Groq': return ["llama3-8b-8192", "gemma-7b-it"]
+        return ["dummy-model"]
+    def get_default_model_for_provider(p):
+        if p == 'Groq': return "llama3-8b-8192"
+        return "dummy-model"
     # The dummy function already accepts the api_key argument ('a')
     def generate_stream(p, m, a, msgs):
         yield f"Using dummy model. API Key provided: {'Yes' if a else 'No'}. This is a dummy response as local modules were not found."
@@ -724,9 +728,31 @@ with gr.Blocks(theme=custom_theme, css=custom_css) as demo:
                 load_space_button = gr.Button("🔄 Load Existing Space", variant="secondary")
 
             with gr.Accordion("🤖 AI Model Settings", open=True):
-                provider_select = gr.Dropdown(label="AI Provider", choices=get_available_providers(), value=get_default_model_for_provider(get_available_providers()[0] if get_available_providers() else 'Groq'))
-                model_select = gr.Dropdown(label="AI Model", choices=[])
-                # --- NEW UI ELEMENT ---
+                # --- MODIFIED: Set up default provider and model logic on load ---
+                available_providers = get_available_providers()
+                default_provider = 'Groq'
+                # Fallback if 'Groq' is not an option
+                if default_provider not in available_providers:
+                    default_provider = available_providers[0] if available_providers else None
+
+                # Get initial models and the default model for the selected provider
+                initial_models = get_models_for_provider(default_provider) if default_provider else []
+                initial_model = get_default_model_for_provider(default_provider) if default_provider else None
+                # Fallback for the model as well
+                if initial_model not in initial_models:
+                    initial_model = initial_models[0] if initial_models else None
+
+                provider_select = gr.Dropdown(
+                    label="AI Provider",
+                    choices=available_providers,
+                    value=default_provider
+                )
+                model_select = gr.Dropdown(
+                    label="AI Model",
+                    choices=initial_models,
+                    value=initial_model
+                )
+                # --- END MODIFICATION ---
                 provider_api_key_input = gr.Textbox(label="Model Provider API Key (Optional)", type="password", placeholder="sk_... (overrides backend settings)")
                 system_prompt_input = gr.Textbox(label="System Prompt", lines=10, value=DEFAULT_SYSTEM_PROMPT, elem_id="system-prompt")
 
@@ -781,7 +807,6 @@ with gr.Blocks(theme=custom_theme, css=custom_css) as demo:
     # --- Event Listeners ---
     provider_select.change(update_models_dropdown, inputs=provider_select, outputs=model_select)
 
-    # --- UPDATED chat_inputs LIST ---
     chat_inputs = [chat_message_input, chatbot_display, hf_api_key_input, provider_api_key_input, provider_select, model_select, system_prompt_input, owner_name_input, space_name_input]
     chat_outputs = [
         chat_message_input, chatbot_display, status_output,