Update app.py

app.py

@@ -1,3 +1,5 @@
+### File: app.py
+```python
 import gradio as gr
 import re
 import json
@@ -83,7 +85,7 @@ If no code or actions are requested, respond conversationally and help the user
 
 def escape_html_for_markdown(text):
     if not isinstance(text, str): return ""
-    return text.replace("&", "&amp;").replace("<", "&lt;").replace(">", "&gt;")
+    return text.replace("&", "&").replace("<", "<").replace(">", ">")
 
 def _infer_lang_from_filename(filename):
     if not filename: return "plaintext"
@@ -350,30 +352,21 @@ def generate_and_stage_changes(ai_response_content, current_files_state, hf_owne
 
 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)
-    _status = "Initializing..."
+    _chat_msg_in = user_message # Keep user input for clearing later
+    _chat_hist = list(chat_history)
+    _chat_hist.append((user_message, None)) # Append user message immediately
 
+    # Initial yield: Show loading state, clear input, clear/hide changes
     yield (
-        _chat_msg_in, _chat_hist, _status,
-        gr.update(), gr.update(), gr.update(interactive=False), gr.update(value="*No changes proposed.*"),
-        [], gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
-    )
-
-    if not user_message.strip():
-        _status = "Cannot send an empty message."
-        yield (
-            _chat_msg_in, _chat_hist, _status,
-            gr.update(), gr.update(), gr.update(), gr.update(),
-            [], gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
-        )
-        return
-
-    _chat_hist.append((user_message, None))
-    _status = f"Sending to {model_select}..."
-    yield (
-        _chat_msg_in, _chat_hist, _status,
-        gr.update(), gr.update(), gr.update(), gr.update(),
-        [], gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+        "", # Clear chat_message_input
+        _chat_hist, # Update chat_history with user message
+        "Initializing...", # status_output
+        gr.update(), # Keep detected_files_preview
+        gr.update(), # Keep formatted_space_output_display
+        gr.update(interactive=False), # download_button (disable)
+        [], # changeset_state (clear)
+        "*No changes proposed.*", # changeset_display (clear text)
+        gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) # Hide confirm UI
     )
 
     current_sys_prompt = system_prompt.strip() or DEFAULT_SYSTEM_PROMPT
@@ -392,45 +385,59 @@ def handle_chat_submit(user_message, chat_history, hf_api_key_input, provider_ap
             if chunk is None: continue
             full_bot_response_content += str(chunk)
             _chat_hist[-1] = (user_message, full_bot_response_content)
-            _status = f"Streaming from {model_select}..."
+            # Yield during streaming - ONLY update chatbot and status
             yield (
-                _chat_msg_in, _chat_hist, _status,
-                gr.update(), gr.update(), gr.update(), gr.update(),
-                [], gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+                gr.update(), # Keep chat_message_input cleared
+                _chat_hist, # Update chat_history with new chunk
+                f"Streaming from {model_select}...", # status_output
+                gr.update(), # Keep detected_files_preview
+                gr.update(), # Keep formatted_space_output_display
+                gr.update(), # Keep download_button
+                gr.update(), # Keep changeset_state
+                gr.update(), # Keep changeset_display
+                gr.update(), gr.update(), gr.update() # Keep confirm UI visibility
             )
 
+        # --- Post-streaming: Parse AI output, update cache, stage changes, FINAL UI UPDATE ---
+
         if full_bot_response_content.startswith("Error:") or full_bot_response_content.startswith("API HTTP Error"):
             _status = full_bot_response_content
-            yield (_chat_msg_in, _chat_hist, _status, gr.update(), gr.update(), gr.update(), gr.update(), [], gr.update(visible=False), gr.update(visible=False), gr.update(visible=False))
-            return
-
-        _status = "Stream complete. Parsing response and staging changes..."
-        yield (_chat_msg_in, _chat_hist, _status, gr.update(), gr.update(), gr.update(), gr.update(), [], gr.update(visible=False), gr.update(visible=False), gr.update(visible=False))
+            # Update chatbot with error in the last message, update status, clear/hide changes
+            yield (
+                 gr.update(), _chat_hist, _status,
+                 gr.update(), gr.update(), gr.update(), # Keep file previews as they were
+                 [], "*Error occurred, changes plan cleared.*", # Clear changeset state and display
+                 gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) # Hide confirm UI
+            )
+            return # Exit handler
 
+        # Parse AI output and update state cache
         parsed_code_blocks_state_cache, proposed_filenames_in_turn = _parse_and_update_state_cache(full_bot_response_content, parsed_code_blocks_state_cache)
+
+        # Regenerate UI previews based on the updated cache
         _formatted, _detected, _download = _generate_ui_outputs_from_cache(hf_owner_name, hf_repo_name)
 
+        # Generate changeset and summary
         staged_changeset, summary_md = generate_and_stage_changes(full_bot_response_content, parsed_code_blocks_state_cache, hf_owner_name, hf_repo_name)
 
+        # Final yield based on whether changes were staged
         if not staged_changeset:
-            _status = summary_md
+            _status = summary_md # Will be "No changes proposed" message
             yield (
-                _chat_msg_in, _chat_hist, _status,
-                _detected, _formatted, _download,
-                [],
-                gr.update(value=summary_md),
-                gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+                gr.update(), # Keep input box cleared
+                _chat_hist, _status, # Update chatbot (final), status
+                _detected, _formatted, _download, # Update file previews
+                [], summary_md, # Clear changeset state, display summary
+                gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) # Hide confirm UI
             )
         else:
             _status = "Change plan generated. Please review and confirm below."
             yield (
-                _chat_msg_in, _chat_hist, _status,
-                _detected, _formatted, _download,
-                staged_changeset,
-                gr.update(value=summary_md),
-                gr.update(visible=True),
-                gr.update(visible=True),
-                gr.update(visible=True)
+                gr.update(), # Keep input box cleared
+                _chat_hist, _status, # Update chatbot (final), status
+                _detected, _formatted, _download, # Update file previews
+                staged_changeset, summary_md, # Set changeset state, display summary
+                gr.update(visible=True), gr.update(visible=True), gr.update(visible=True) # Show confirm UI
             )
 
     except Exception as e:
@@ -438,20 +445,26 @@ def handle_chat_submit(user_message, chat_history, hf_api_key_input, provider_ap
         print(f"Error in handle_chat_submit: {e}")
         import traceback
         traceback.print_exc()
+        # Update chatbot with error
         if _chat_hist:
-            if _chat_hist[-1] and _chat_hist[-1][0] == user_message:
-                 _chat_hist[-1] = (user_message, (full_bot_response_content + "\n\n" if full_bot_response_content and full_bot_response_content != user_message else "") + error_msg)
-            else:
-                 _chat_hist.append((user_message, error_msg))
-
+            # Find the last message matching the user's input and append the error
+            try:
+                idx = next(i for i in reversed(range(len(_chat_hist))) if _chat_hist[i] and _chat_hist[i][0] == user_message)
+                current_bot_response = _chat_hist[idx][1] or ""
+                _chat_hist[idx] = (user_message, (current_bot_response + "\n\n" if current_bot_response else "") + error_msg)
+            except (StopIteration, IndexError):
+                _chat_hist.append((user_message, error_msg)) # Fallback if user message not found
+
+        # Regenerate previews from potentially partially updated cache
         _formatted, _detected, _download = _generate_ui_outputs_from_cache(hf_owner_name, hf_repo_name)
 
+        # Final error yield: Update chatbot, status, file previews, clear/hide changes
         yield (
-            _chat_msg_in, _chat_hist, error_msg,
-            _detected, _formatted, _download,
-            [],
-            gr.update(value="*Error occurred, changes plan cleared.*"),
-            gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+            gr.update(), # Keep input box content
+            _chat_hist, error_msg, # Update chatbot, status
+            _detected, _formatted, _download, # Update file previews based on state before error
+            [], "*Error occurred, changes plan cleared.*", # Clear changeset state and display
+            gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) # Hide confirm UI
         )
 
 
@@ -540,45 +553,63 @@ def handle_load_existing_space(hf_api_key_ui, ui_owner_name, ui_space_name):
     _confirm_ui_hidden = gr.update(visible=False)
 
     outputs = [
-        _formatted_md_val, _detected_preview_val, _status_val, _file_browser_update,
-        gr.update(value=ui_owner_name), gr.update(value=ui_space_name),
-        _iframe_html_update, _download_btn_update, _build_status_clear,
-        _edit_status_clear, _runtime_status_clear,
+        formatted_space_output_display, detected_files_preview, status_output,
+        file_browser_dropdown, owner_name_input, space_name_input,
+        space_iframe_display, download_button, build_status_display,
+        edit_status_display, space_runtime_status_display,
         changeset_state, changeset_display, confirm_accordion, confirm_button, cancel_button
     ]
-    yield outputs
+    # Use gr.update for components that are not state variables
+    initial_updates = [
+        gr.update(value=_formatted_md_val), gr.update(value=_detected_preview_val), gr.update(value=_status_val), _file_browser_update,
+        gr.update(value=ui_owner_name), gr.update(value=ui_space_name),
+        _iframe_html_update, _download_btn_update, gr.update(value=_build_status_clear),
+        gr.update(value=_edit_status_clear), gr.update(value=_runtime_status_clear),
+        _changeset_clear, gr.update(value=_changeset_summary_clear), _confirm_ui_hidden, _confirm_ui_hidden, _confirm_ui_hidden
+    ]
+
+    yield initial_updates
 
     owner_to_use = ui_owner_name
     if not owner_to_use:
         token, token_err = build_logic_get_api_token(hf_api_key_ui)
         if token_err:
             _status_val = f"Load Error: {token_err}"
-            outputs[2] = _status_val; yield outputs; return
+            yield gr.update(value=_status_val), # Only update status on early error
+            return
         try:
             user_info = build_logic_whoami(token=token)
             owner_to_use = user_info.get('name')
             if not owner_to_use: raise Exception("Could not find user name from token.")
-            outputs[4] = gr.update(value=owner_to_use)
-            _status_val += f" (Auto-detected owner: {owner_to_use})"
+            yield gr.update(value=owner_to_use), gr.update(value=f"Loading Space: {owner_to_use}/{ui_space_name} (Auto-detected owner)...") # Update owner and status
         except Exception as e:
-            _status_val = f"Load Error: Error auto-detecting owner: {e}"; outputs[2] = _status_val; yield outputs; return
+            _status_val = f"Load Error: Error auto-detecting owner: {e}"
+            yield gr.update(value=_status_val), # Only update status on early error
+            return
 
     if not owner_to_use or not ui_space_name:
-        _status_val = "Load Error: Owner and Space Name are required."; outputs[2] = _status_val; yield outputs; return
+        _status_val = "Load Error: Owner and Space Name are required."
+        yield gr.update(value=_status_val),
+        return
 
     sdk, file_list, err = get_space_repository_info(hf_api_key_ui, ui_space_name, owner_to_use)
 
-    outputs[4] = gr.update(value=owner_to_use)
-    outputs[5] = gr.update(value=ui_space_name)
+    # Always update owner/space inputs even on error, as user entered them
+    # These were updated in the initial yield, but we might need to update again if owner was auto-detected
+    yield gr.update(value=owner_to_use), gr.update(value=ui_space_name), # Update owner and space textboxes
 
     if err:
         _status_val = f"Load Error: {err}"
         parsed_code_blocks_state_cache = []
         _formatted, _detected, _download = _generate_ui_outputs_from_cache(owner_to_use, ui_space_name)
-        outputs[0], outputs[1], outputs[2], outputs[7] = _formatted, _detected, _status_val, _download
-        outputs[3] = gr.update(visible=False, choices=[], value=None)
-        outputs[6] = gr.update(value=None, visible=False)
-        yield outputs; return
+        yield (
+            gr.update(value=_formatted), gr.update(value=_detected), gr.update(value=_status_val), # Update markdown, preview, status
+            gr.update(visible=False, choices=[], value=None), # Hide file browser
+            gr.update(), gr.update(), # Keep owner/space
+            gr.update(value=None, visible=False), # Hide iframe
+            _download, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update() # Rest are updates
+        )
+        return
 
     loaded_files = []
     for file_path in file_list:
@@ -592,19 +623,25 @@ def handle_load_existing_space(hf_api_key_ui, ui_owner_name, ui_space_name):
 
     _formatted, _detected, _download = _generate_ui_outputs_from_cache(owner_to_use, ui_space_name)
     _status_val = f"Successfully loaded {len(file_list)} files from {owner_to_use}/{ui_space_name}. SDK: {sdk or 'unknown'}."
-    outputs[0], outputs[1], outputs[2], outputs[7] = _formatted, _detected, _status_val, _download
 
-    outputs[3] = gr.update(visible=True, choices=sorted(file_list or []), value=None)
+    file_browser_update = gr.update(visible=True, choices=sorted(file_list or []), value=None)
 
     if owner_to_use and ui_space_name:
         sub_owner = re.sub(r'[^a-z0-9\-]+', '-', owner_to_use.lower()).strip('-') or 'owner'
         sub_repo = re.sub(r'[^a-z0-9\-]+', '-', ui_space_name.lower()).strip('-') or 'space'
         iframe_url = f"https://{sub_owner}-{sub_repo}{'.static.hf.space' if sdk == 'static' else '.hf.space'}"
-        outputs[6] = gr.update(value=f'<iframe src="{iframe_url}?__theme=light&embed=true" width="100%" height="500px"></iframe>', visible=True)
+        iframe_update = gr.update(value=f'<iframe src="{iframe_url}?__theme=light&embed=true" width="100%" height="500px"></iframe>', visible=True)
     else:
-         outputs[6] = gr.update(value=None, visible=False)
+         iframe_update = gr.update(value=None, visible=False)
 
-    yield outputs
+    # Final yield after success
+    yield (
+        gr.update(value=_formatted), gr.update(value=_detected), gr.update(value=_status_val), # Update markdown, preview, status
+        file_browser_update, # Update file browser
+        gr.update(), gr.update(), # Keep owner/space
+        iframe_update, # Update iframe
+        _download, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update() # Rest are updates
+    )
 
 def handle_build_space_button(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, space_sdk_ui, is_private_ui, formatted_markdown_content):
     global parsed_code_blocks_state_cache
@@ -614,16 +651,26 @@ def handle_build_space_button(hf_api_key_ui, ui_space_name_part, ui_owner_name_p
     _changeset_summary_clear = "*Manual build initiated, changes plan cleared.*"
     _confirm_ui_hidden = gr.update(visible=False)
 
-    yield (_build_status, _iframe_html, _file_browser_update, gr.update(value=ui_owner_name_part), gr.update(value=ui_space_name_part),
-           _changeset_clear, _changeset_summary_clear, _confirm_ui_hidden, _confirm_ui_hidden, _confirm_ui_hidden,
-           gr.update(), gr.update(), gr.update())
+    outputs = [
+        build_status_display, space_iframe_display, file_browser_dropdown,
+        owner_name_input, space_name_input,
+        changeset_state, changeset_display, confirm_accordion, confirm_button, cancel_button,
+        formatted_space_output_display, detected_files_preview, download_button
+    ]
+    # Initial yield state
+    initial_updates = [
+        gr.update(value=_build_status), _iframe_html, _file_browser_update,
+        gr.update(value=ui_owner_name_part), gr.update(value=ui_space_name_part),
+        _changeset_clear, gr.update(value=_changeset_summary_clear), _confirm_ui_hidden, _confirm_ui_hidden, _confirm_ui_hidden,
+        gr.update(), gr.update(), gr.update() # Don't update markdown/download yet
+    ]
+    yield initial_updates
 
 
     if not ui_space_name_part or "/" in ui_space_name_part:
         _build_status = f"Build Error: Invalid Space Name '{ui_space_name_part}'."
-        yield (_build_status, _iframe_html, _file_browser_update, gr.update(), gr.update(),
-               gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
-               gr.update(), gr.update(), gr.update()); return
+        yield gr.update(value=_build_status), # Only update status
+        return
 
     parsed_content = build_logic_parse_markdown(formatted_markdown_content)
     proposed_files_list = parsed_content.get("files", [])
@@ -636,9 +683,8 @@ def handle_build_space_button(hf_api_key_ui, ui_space_name_part, ui_owner_name_p
 
     if not manual_changeset:
          _build_status = "Build Error: No target space specified or no files parsed from markdown."
-         yield (_build_status, _iframe_html, _file_browser_update, gr.update(), gr.update(),
-                gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
-                gr.update(), gr.update(), gr.update()); return
+         yield gr.update(value=_build_status), # Only update status
+         return
 
 
     result_message = apply_staged_changes(hf_api_key_ui, ui_owner_name_part, ui_space_name_part, manual_changeset)
@@ -646,19 +692,27 @@ def handle_build_space_button(hf_api_key_ui, ui_space_name_part, ui_owner_name_p
 
     owner_to_use = ui_owner_name_part
     space_to_use = ui_space_name_part
-    _formatted_md, _detected_preview, _download = formatted_markdown_content, "*Loading files after build...*", gr.update(interactive=False, value=None)
+    # Keep manual markdown content as is, clear/disable preview/download initially
+    _formatted_md = formatted_markdown_content
+    _detected_preview_val = "*Loading files after build...*"
+    _download_btn_update = gr.update(interactive=False, value=None)
 
-    yield (_build_status, _iframe_html, _file_browser_update, gr.update(value=owner_to_use), gr.update(value=space_to_use),
-           _changeset_clear, _changeset_summary_clear, _confirm_ui_hidden, _confirm_ui_hidden, _confirm_ui_hidden,
-           _formatted_md, _detected_preview, _download)
+    # Yield intermediate state with build status and placeholders for reloading
+    yield (
+        gr.update(value=_build_status), _iframe_html, _file_browser_update, # Build status, iframe, file browser (initial clear state)
+        gr.update(value=owner_to_use), gr.update(value=space_to_use), # Update owner/space textboxes
+        _changeset_clear, gr.update(value=_changeset_summary_clear), _confirm_ui_hidden, _confirm_ui_hidden, _confirm_ui_hidden, # Changeset clear/hide
+        gr.update(value=_formatted_md), gr.update(value=_detected_preview_val), _download_btn_update # Update markdown/preview/download
+    )
 
+    # Reload space state to reflect actual content after build attempt
     sdk_built, file_list, err_list = get_space_repository_info(hf_api_key_ui, space_to_use, owner_to_use)
 
     if err_list:
         _build_status += f" | Error reloading file list after build: {err_list}"
         parsed_code_blocks_state_cache = []
         _file_browser_update = gr.update(visible=False, choices=[], value=None)
-        _iframe_html = gr.update(value=None, visible=False)
+        _iframe_html_update = gr.update(value=None, visible=False)
     else:
         loaded_files = []
         for file_path in file_list:
@@ -675,16 +729,21 @@ def handle_build_space_button(hf_api_key_ui, ui_space_name_part, ui_owner_name_p
             sub_owner = re.sub(r'[^a-z0-9\-]+', '-', owner_to_use.lower()).strip('-') or 'owner'
             sub_repo = re.sub(r'[^a-z0-9\-]+', '-', ui_space_name_part.lower()).strip('-') or 'space'
             iframe_url = f"https://{sub_owner}-{sub_repo}{'.static.hf.space' if sdk_built == 'static' else '.hf.space'}"
-            _iframe_html = gr.update(value=f'<iframe src="{iframe_url}?__theme=light&embed=true" width="100%" height="700px"></iframe>', visible=True)
+            _iframe_html_update = gr.update(value=f'<iframe src="{iframe_url}?__theme=light&embed=true" width="100%" height="700px"></iframe>', visible=True)
         else:
-             _iframe_html = gr.update(value=None, visible=False)
+             _iframe_html_update = gr.update(value=None, visible=False)
+
 
-    _formatted_md, _detected_preview, _download = _generate_ui_outputs_from_cache(owner_to_use, space_to_use)
+    _formatted_md, _detected_preview, _download = _generate_ui_outputs_from_cache(owner_to_use, space_to_use) # Regenerate previews from reloaded cache
 
-    yield (_build_status, _iframe_html, _file_browser_update, gr.update(value=owner_to_use), gr.update(value=space_to_use),
-           _changeset_clear, _changeset_summary_clear, _confirm_ui_hidden, _confirm_ui_hidden, _confirm_ui_hidden,
-           _formatted_md, _detected_preview, _download
-           )
+
+    # Final yield after reloading
+    yield (
+        gr.update(value=_build_status), _iframe_html_update, _file_browser_update, # Build status, iframe, file browser
+        gr.update(value=owner_to_use), gr.update(value=space_to_use), # Update owner/space textboxes
+        _changeset_clear, gr.update(value=_changeset_summary_clear), _confirm_ui_hidden, _confirm_ui_hidden, _confirm_ui_hidden, # Changeset clear/hide
+        gr.update(value=_formatted_md), gr.update(value=_detected_preview), _download # Update markdown/preview/download
+    )
 
 
 def handle_load_file_for_editing(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, selected_file_path):
@@ -700,12 +759,15 @@ def handle_load_file_for_editing(hf_api_key_ui, ui_space_name_part, ui_owner_nam
     return content, f"Loaded `{selected_file_path}`", commit_msg, gr.update(language=lang)
 
 def handle_commit_file_changes(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, file_to_edit_path, edited_content, commit_message):
+    if not ui_owner_name_part or not ui_space_name_part:
+         return "Commit Error: Cannot commit changes. Please load a space first.", gr.update(), gr.update(), gr.update(), gr.update()
     if not file_to_edit_path:
          return "Commit Error: No file selected for commit.", gr.update(), gr.update(), gr.update(), gr.update()
 
     status_msg = update_space_file(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, file_to_edit_path, edited_content, commit_message)
 
     global parsed_code_blocks_state_cache
+    file_browser_update = gr.update()
     if "Successfully" in status_msg:
         found = False
         for block in parsed_code_blocks_state_cache:
@@ -728,14 +790,12 @@ def handle_commit_file_changes(hf_api_key_ui, ui_space_name_part, ui_owner_name_
         file_list, _ = list_space_files_for_browsing(hf_api_key_ui, ui_space_name_part, ui_owner_name_part)
         file_browser_update = gr.update(choices=sorted(file_list or []))
 
-    else:
-         file_browser_update = gr.update()
-
-
     _formatted, _detected, _download = _generate_ui_outputs_from_cache(ui_owner_name_part, ui_space_name_part)
-    return status_msg, file_browser_update, _formatted, _detected, _download
+    return status_msg, file_browser_update, gr.update(value=_formatted), gr.update(value=_detected), _download
 
 def handle_delete_file(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, file_to_delete_path):
+    if not ui_owner_name_part or not ui_space_name_part:
+         return "Delete Error: Cannot delete file. Please load a space first.", gr.update(), "", "", "plaintext", gr.update(), gr.update(), gr.update()
     if not file_to_delete_path:
         return "Delete Error: No file selected to delete.", gr.update(), "", "", "plaintext", gr.update(), gr.update(), gr.update()
 
@@ -762,8 +822,8 @@ def handle_delete_file(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, fi
         file_content_editor_update,
         commit_message_update,
         editor_lang_update,
-        _formatted,
-        _detected,
+        gr.update(value=_formatted),
+        gr.update(value=_detected),
         _download
     )
 
@@ -972,4 +1032,4 @@ with gr.Blocks(theme=custom_theme, css=custom_css) as demo:
     refresh_status_button.click(fn=handle_refresh_space_status, inputs=[hf_api_key_input, owner_name_input, space_name_input], outputs=[space_runtime_status_display])
 
 if __name__ == "__main__":
-    demo.launch(debug=False, mcp_server=True)
+    demo.launch(debug=False, mcp_server=True)