Update app.py

app.py

@@ -33,8 +33,9 @@ except ImportError:
     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"
+    # The dummy function already accepts the api_key argument ('a')
     def generate_stream(p, m, a, msgs):
-        yield "Error: Local modules not found. This is a dummy response."
+        yield f"Using dummy model. API Key provided: {'Yes' if a else 'No'}. This is a dummy response as local modules were not found."
     def build_logic_create_space(*args, **kwargs): return "Error: build_logic not found."
     def build_logic_get_api_token(key): return (key or os.getenv("HF_TOKEN"), None)
     def build_logic_whoami(token): return {"name": "dummy_user"}
@@ -319,18 +320,18 @@ def generate_and_stage_changes(ai_response_content, current_files_state, hf_owne
     """
     changeset = []
     current_files_dict = {f["filename"]: f for f in current_files_state if not f.get("is_structure_block")}
-    
+
     # 1. Parse proposed files from AI response
     parsing_result = _parse_chat_stream_logic(ai_response_content, existing_files_state=current_files_state)
     proposed_files = parsing_result.get("parsed_code_blocks", [])
-    
+
     # 2. Parse HF_ACTION commands from AI response
     action_pattern = re.compile(r"### HF_ACTION:\s*(?P<command_line>[^\n]+)")
     for match in action_pattern.finditer(ai_response_content):
         cmd_parts = shlex.split(match.group("command_line").strip())
         if not cmd_parts: continue
         command, args = cmd_parts[0].upper(), cmd_parts[1:]
-        
+
         # Add actions to the changeset
         if command == "DELETE_FILE" and args:
             changeset.append({"type": "DELETE_FILE", "path": args[0]})
@@ -345,24 +346,24 @@ def generate_and_stage_changes(ai_response_content, current_files_state, hf_owne
             if '--sdk' in args: sdk = args[args.index('--sdk') + 1]
             if '--private' in args: private = args[args.index('--private') + 1].lower() == 'true'
             changeset.append({"type": "CREATE_SPACE", "repo_id": repo_id, "sdk": sdk, "private": private})
-    
+
     # 3. Compare proposed files with current files to determine CREATE/UPDATE
     for file_block in proposed_files:
         if file_block.get("is_structure_block"): continue
-        
+
         filename = file_block["filename"]
         if filename not in current_files_dict:
             changeset.append({"type": "CREATE_FILE", "path": filename, "content": file_block["code"], "lang": file_block["language"]})
         elif file_block["code"] != current_files_dict[filename]["code"]:
             changeset.append({"type": "UPDATE_FILE", "path": filename, "content": file_block["code"], "lang": file_block["language"]})
-            
+
     # 4. Format the changeset into a human-readable Markdown string
     if not changeset:
         return [], "The AI did not propose any specific changes to files or the space.", parsing_result
 
     md_summary = ["### 📋 Proposed Changes Plan\n"]
     md_summary.append("The AI has proposed the following changes. Please review and confirm.")
-    
+
     for change in changeset:
         if change["type"] == "CREATE_FILE":
             md_summary.append(f"- **➕ Create File:** `{change['path']}`")
@@ -376,15 +377,15 @@ def generate_and_stage_changes(ai_response_content, current_files_state, hf_owne
             md_summary.append(f"- **🔒 Set Privacy:** Set `{change['repo_id']}` to `private={change['private']}`")
         elif change["type"] == "DELETE_SPACE":
             md_summary.append(f"- **💥 DELETE ENTIRE SPACE:** `{change['owner']}/{change['space_name']}` **(DESTRUCTIVE ACTION)**")
-            
+
     return changeset, "\n".join(md_summary), parsing_result
 
 # --- Gradio Event Handlers ---
 
-def handle_chat_submit(user_message, chat_history, hf_api_key_input, provider_select, model_select, system_prompt, hf_owner_name, hf_repo_name):
+def handle_chat_submit(user_message, chat_history, hf_api_key_input, provider_api_key_input, provider_select, model_select, system_prompt, hf_owner_name, hf_repo_name):
     global parsed_code_blocks_state_cache
     _chat_msg_in, _chat_hist = "", list(chat_history)
-    
+
     # UI updates for streaming
     yield (
         _chat_msg_in, _chat_hist, "Initializing...",
@@ -406,7 +407,7 @@ def handle_chat_submit(user_message, chat_history, hf_api_key_input, provider_se
         gr.update(), gr.update(), gr.update(),
         [], gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
     )
-    
+
     # Prepare context for the AI
     current_sys_prompt = system_prompt.strip() or DEFAULT_SYSTEM_PROMPT
     export_result = _export_selected_logic(None, f"{hf_owner_name}/{hf_repo_name}", parsed_code_blocks_state_cache)
@@ -416,7 +417,9 @@ def handle_chat_submit(user_message, chat_history, hf_api_key_input, provider_se
 
     try:
         full_bot_response_content = ""
-        for chunk in generate_stream(provider_select, model_select, None, api_msgs):
+        # Pass the provider API key from the UI to the generation logic
+        streamer = generate_stream(provider_select, model_select, provider_api_key_input, api_msgs)
+        for chunk in streamer:
             if chunk is None: continue
             if isinstance(chunk, str) and (chunk.startswith("Error:") or chunk.startswith("API HTTP Error")):
                 full_bot_response_content = chunk; break
@@ -433,10 +436,10 @@ def handle_chat_submit(user_message, chat_history, hf_api_key_input, provider_se
             yield (_chat_msg_in, _chat_hist, _status, gr.update(), gr.update(), gr.update(), [], gr.update(), gr.update(), gr.update(), gr.update())
             return
 
-        # NEW: Instead of applying, generate and stage changes
+        # Instead of applying, generate and stage changes
         _status = "Stream complete. Generating change plan..."
         yield (_chat_msg_in, _chat_hist, _status, gr.update(), gr.update(), gr.update(), [], gr.update(), gr.update(), gr.update(), gr.update())
-        
+
         staged_changeset, summary_md, parsing_res = generate_and_stage_changes(full_bot_response_content, parsed_code_blocks_state_cache, hf_owner_name, hf_repo_name)
 
         if parsing_res["error_message"]:
@@ -479,7 +482,7 @@ def handle_confirm_changes(hf_api_key, owner_name, space_name, changeset):
         return "No changes to apply.", gr.update(), gr.update(), gr.update(), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
     status_messages = []
-    
+
     # Handle space creation first, as other ops might depend on it
     create_space_op = next((c for c in changeset if c['type'] == 'CREATE_SPACE'), None)
     if create_space_op:
@@ -489,12 +492,12 @@ def handle_confirm_changes(hf_api_key, owner_name, space_name, changeset):
             # We need to pass the full markdown for creation. Let's build it from the plan.
             # This is a simplification; a more robust solution would pass the planned files directly.
             # For now, we assume the AI provides file content for the new space.
-            
+
             planned_files_md = [f"# Space: {create_space_op['repo_id']}"]
             for change in changeset:
                 if change['type'] in ['CREATE_FILE', 'UPDATE_FILE']:
                      planned_files_md.append(f"### File: {change['path']}\n{bbb}{change.get('lang', 'plaintext')}\n{change['content']}\n{bbb}")
-            
+
             markdown_for_creation = "\n\n".join(planned_files_md)
 
             result = build_logic_create_space(
@@ -543,10 +546,10 @@ def handle_confirm_changes(hf_api_key, owner_name, space_name, changeset):
                     parsed_code_blocks_state_cache = [] # Clear everything
         except Exception as e:
             status_messages.append(f"Error applying {change['type']} for {change.get('path', '')}: {e}")
-            
+
     final_status = " | ".join(status_messages)
     _formatted, _detected, _download = _generate_ui_outputs_from_cache(owner_name, space_name)
-    
+
     # Hide the confirmation UI and clear the state
     return final_status, _formatted, _detected, _download, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), []
 
@@ -664,7 +667,7 @@ def handle_commit_file_changes(hf_api_key_ui, ui_space_name_part, ui_owner_name_
 def handle_delete_file(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, file_to_delete_path):
     if not file_to_delete_path:
         return "No file selected to delete.", gr.update(), "", "", "plaintext", gr.update(), gr.update(), gr.update()
-    
+
     status_msg = build_logic_delete_space_file(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, file_to_delete_path)
     file_list, _ = list_space_files_for_browsing(hf_api_key_ui, ui_space_name_part, ui_owner_name_part)
     global parsed_code_blocks_state_cache
@@ -676,7 +679,7 @@ def handle_delete_file(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, fi
 def handle_refresh_space_status(hf_api_key_ui, ui_owner_name, ui_space_name):
     if not ui_owner_name or not ui_space_name:
         return "Owner and Space Name must be provided to get status."
-    
+
     status, err = get_space_runtime_status(hf_api_key_ui, ui_space_name, ui_owner_name)
     if err: return f"**Error:** {err}"
     if not status: return "Could not retrieve status."
@@ -706,9 +709,9 @@ body { background: linear-gradient(to bottom right, #2c3e50, #34495e); color: #e
 
 # --- Gradio UI Definition ---
 with gr.Blocks(theme=custom_theme, css=custom_css) as demo:
-    # --- NEW: State to hold the plan ---
+    # State to hold the plan
     changeset_state = gr.State([])
-    
+
     gr.Markdown("# 🤖 AI-Powered Hugging Face Space Builder")
     gr.Markdown("Use an AI assistant to create, modify, build, and manage your Hugging Face Spaces directly from this interface.")
 
@@ -723,6 +726,8 @@ with gr.Blocks(theme=custom_theme, css=custom_css) as demo:
             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 ---
+                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")
 
     with gr.Column(scale=2):
@@ -733,7 +738,7 @@ with gr.Blocks(theme=custom_theme, css=custom_css) as demo:
             send_chat_button = gr.Button("Send", variant="primary", scale=1)
         status_output = gr.Textbox(label="Last Action Status", interactive=False, value="Ready.")
 
-    # --- NEW: Confirmation Accordion ---
+    # Confirmation Accordion
     with gr.Accordion("📝 Proposed Changes (Pending Confirmation)", visible=False) as confirm_accordion:
         changeset_display = gr.Markdown("No changes proposed.")
         with gr.Row():
@@ -776,7 +781,8 @@ 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)
 
-    chat_inputs = [chat_message_input, chatbot_display, hf_api_key_input, provider_select, model_select, system_prompt_input, owner_name_input, space_name_input]
+    # --- 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,
         detected_files_preview, formatted_space_output_display, download_button,
@@ -785,7 +791,7 @@ with gr.Blocks(theme=custom_theme, css=custom_css) as demo:
     send_chat_button.click(handle_chat_submit, inputs=chat_inputs, outputs=chat_outputs)
     chat_message_input.submit(handle_chat_submit, inputs=chat_inputs, outputs=chat_outputs)
 
-    # --- NEW: Confirmation Button Listeners ---
+    # Confirmation Button Listeners
     confirm_inputs = [hf_api_key_input, owner_name_input, space_name_input, changeset_state]
     confirm_outputs = [
         status_output, formatted_space_output_display, detected_files_preview, download_button,