Update app.py

app.py

@@ -1,3 +1,5 @@
+### File: app.py
+```python
 import gradio as gr
 import re
 import json
@@ -98,7 +100,7 @@ def escape_html_for_markdown(text):
     return text.replace("&", "&").replace("<", "<").replace(">", ">")
 
 def _infer_lang_from_filename(filename):
-    if not filename: return "text" # Changed from "plaintext" to "text" for gr.Code compatibility
+    if not filename: return "text"
     if '.' in filename:
         ext = filename.split('.')[-1].lower()
         mapping = {
@@ -110,17 +112,17 @@ def _infer_lang_from_filename(filename):
             'c': 'c', 'h': 'c', 'cpp': 'cpp', 'hpp': 'cpp', 'cs': 'csharp', 'java': 'java',
             'rb': 'ruby', 'php': 'php', 'go': 'go', 'rs': 'rust', 'swift': 'swift', 'kt': 'kotlin', 'kts': 'kotlin',
             'sql': 'sql', 'dockerfile': 'docker', 'tf': 'terraform', 'hcl': 'terraform',
-            'txt': 'text', 'log': 'text', 'ini': 'ini', 'conf': 'text', 'cfg': 'text', # Changed plaintext to text
-            'csv': 'text', 'tsv': 'text', 'err': 'text', # Changed plaintext to text
-            '.env': 'text', '.gitignore': 'text', '.npmrc': 'text', '.gitattributes': 'text', # Changed plaintext to text
+            'txt': 'text', 'log': 'text', 'ini': 'ini', 'conf': 'text', 'cfg': 'text',
+            'csv': 'text', 'tsv': 'text', 'err': 'text',
+            '.env': 'text', '.gitignore': 'text', '.npmrc': 'text', '.gitattributes': 'text',
             'makefile': 'makefile',
         }
-        return mapping.get(ext, "text") # Changed default to text
+        return mapping.get(ext, "text")
     base_filename = os.path.basename(filename)
     if base_filename == 'Dockerfile': return 'docker'
     if base_filename == 'Makefile': return 'makefile'
-    if base_filename.startswith('.'): return 'text' # Changed plaintext to text
-    return "text" # Changed default to text
+    if base_filename.startswith('.'): return 'text'
+    return "text"
 
 def _clean_filename(filename_line_content):
     text = filename_line_content.strip()
@@ -141,7 +143,7 @@ def _parse_and_update_state_cache(latest_bot_message_content, current_files_stat
 
     structure_match = structure_pattern.search(content)
     if structure_match:
-        structure_block_state = {"filename": "File Structure (from AI)", "language": structure_match.group("struct_lang") or "text", "code": structure_match.group("structure_code").strip(), "is_binary": False, "is_structure_block": True} # Changed plaintext to text
+        structure_block_state = {"filename": "File Structure (from AI)", "language": structure_match.group("struct_lang") or "text", "code": structure_match.group("structure_code").strip(), "is_binary": False, "is_structure_block": True}
 
     current_message_proposed_filenames = []
     for match in file_pattern.finditer(content):
@@ -200,7 +202,7 @@ def _export_selected_logic(selected_filenames, space_line_name_for_md, parsed_bl
         if block.get('is_binary') or content.startswith(("[Binary file", "[Error loading content:", "[Binary or Skipped file]")):
              output_lines.append(content)
         else:
-            lang = block.get('language', 'text') or 'text' # Changed plaintext to text
+            lang = block.get('language', 'text') or 'text'
             output_lines.extend([f"{bbb}{lang}", content, bbb])
         output_lines.append("")
         exported_content_count += 1
@@ -257,7 +259,7 @@ def _generate_ui_outputs_from_cache(owner, space_name):
                 if block.get('is_binary') or content.startswith(("[Binary file", "[Error loading content:", "[Binary or Skipped file]")):
                      preview_md_lines.append(f"\n`{escape_html_for_markdown(content.strip())}`\n")
                 else:
-                     lang = block.get('language', 'text') or 'text' # Changed plaintext to text
+                     lang = block.get('language', 'text') or 'text'
                      preview_md_lines.append(f"\n{bbb}{lang}\n{content.strip()}\n{bbb}\n")
         preview_md_val = "\n".join(preview_md_lines)
 
@@ -593,12 +595,21 @@ def handle_confirm_changes(hf_api_key, owner_name, space_name, changeset):
     global parsed_code_blocks_state_cache
 
     _status = "Applying changes..."
+    # Yield initial status, hide confirm UI, clear summary
     yield _status, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(value="*Applying changes...*")
-    # Explicitly yield None for the other 15 outputs initially
-    yield None, None, None, None, None, None, None, None, None, None, None, None, None, None, None # Ensure 16 None values
+    # Yield None for the other 15 outputs that are part of this generator's outputs list
+    yield None, None, None, None, None, None, None, None, None, None, None, None, None, None, None # 15 None values
+
 
     if not changeset:
-        return "No changes to apply.", gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(value="No changes were staged.")
+        # This case should be prevented by hiding the button, but safety first
+        yield (
+            "No changes to apply.", # status_output
+            gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(value="No changes were staged."), # Hide confirm UI, clear summary
+            None, None, None, None, None, None, None, None, None, None, None # 15 None values
+        )
+        return
+
 
     first_action = changeset[0] if changeset else None
     is_exclusive_duplicate = first_action and first_action.get('type') == 'DUPLICATE_SPACE'
@@ -615,8 +626,10 @@ def handle_confirm_changes(hf_api_key, owner_name, space_name, changeset):
                 private=change.get("private", False)
             )
             _status_reload = f"{status_message} | Attempting to load the new Space [{change['target_repo_id']}]..."
+            # Yield status update only
             yield gr.update(value=_status_reload), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
 
+
             target_repo_id = change['target_repo_id']
             new_owner, new_space_name = target_repo_id.split('/', 1) if '/' in target_repo_id else (None, target_repo_id)
 
@@ -664,18 +677,20 @@ def handle_confirm_changes(hf_api_key, owner_name, space_name, changeset):
 
                 runtime_status_md = handle_refresh_space_status(hf_api_key, new_owner, new_space_name)
 
+                # Final yield after loading the new space
                 yield (
-                     gr.update(value=final_overall_status), gr.update(value=_formatted), gr.update(value=_detected), _download,
-                     gr.update(visible=False), gr.update(visible=False), gr.update(visible=False),
-                     [], gr.update(value="*No changes proposed.*"),
-                     owner_update, space_update, file_browser_update, iframe_update,
-                     gr.update(value=runtime_status_md)
+                     gr.update(value=final_overall_status), gr.update(value=_formatted), gr.update(value=_detected), _download, # Status, markdown, preview, download
+                     gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), # Hide confirm UI
+                     [], gr.update(value="*No changes proposed.*"), # Clear changeset state and display
+                     owner_update, space_update, file_browser_update, iframe_update, # Update space/owner fields, file browser, iframe
+                     gr.update(value=runtime_status_md) # Update runtime status
                 )
 
             else:
                  reload_error = "Cannot load new Space state: Owner or Space Name missing after duplication."
                  _formatted, _detected, _download = _generate_ui_outputs_from_cache(None, None)
                  final_overall_status = status_message + f" | Reload Status: {reload_error}"
+                 # Keep old UI fields, clear file browser/iframe
                  yield (
                      gr.update(value=final_overall_status), gr.update(value=_formatted), gr.update(value=_detected), _download,
                      gr.update(visible=False), gr.update(visible=False), gr.update(visible=False),
@@ -705,10 +720,18 @@ def handle_confirm_changes(hf_api_key, owner_name, space_name, changeset):
          if not delete_repo_id_target or delete_repo_id_target != current_repo_id:
              final_overall_status = f"DELETE_SPACE Error: Action blocked. Cannot delete '{delete_repo_id_target}'. Only deletion of the currently loaded space ('{current_repo_id}') is permitted via AI action."
              print(f"Blocked DELETE_SPACE action via confirm: requested '{delete_repo_id_target}', current '{current_repo_id}'.")
+             _formatted, _detected, _download = _generate_ui_outputs_from_cache(owner_name, space_name)
+             owner_update = gr.update()
+             space_update = gr.update()
+             file_browser_update = gr.update()
+             iframe_update = gr.update()
+             runtime_status_update = handle_refresh_space_status(hf_api_key, owner_name, space_name) # Refresh status after failed delete attempt
+
          else:
              status_message = build_logic_delete_space(hf_api_key, delete_owner, delete_space)
              final_overall_status = status_message
              if "Successfully" in status_message:
+                  global parsed_code_blocks_state_cache
                   parsed_code_blocks_state_cache = []
                   _formatted, _detected, _download = _generate_ui_outputs_from_cache(None, None)
                   owner_update = gr.update(value="")
@@ -722,7 +745,7 @@ def handle_confirm_changes(hf_api_key, owner_name, space_name, changeset):
                   space_update = gr.update()
                   file_browser_update = gr.update()
                   iframe_update = gr.update()
-                  runtime_status_update = handle_refresh_space_status(hf_api_key, owner_name, space_name)
+                  runtime_status_update = handle_refresh_space_status(hf_api_key, owner_name, space_name) # Refresh status after failed delete attempt
 
 
          cleared_changeset = []
@@ -883,14 +906,15 @@ def update_models_dropdown(provider_select):
 def handle_detect_user_and_list_spaces(hf_api_key_ui):
     _status = "Detecting user from token..."
     # Yield initial state: update status, clear owner/space fields, clear list display, clear duplicate fields
-    yield (gr.update(value=_status), gr.update(value=""), gr.update(value=""), gr.update(value="*Listing spaces...*"), gr.update(visible=False, choices=[], value=None), gr.update(value=None, visible=False), gr.update(value=""), gr.update(value=""), gr.update(value="*Manual build status...*"), gr.update(value="*Select a file...*"), gr.update(value="*Runtime/Repo status will appear here.*"), [], gr.update(value="*No changes proposed.*"), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(value=""), gr.update(value=""), gr.update(value=False))
+    # outputs list: [status_output, owner_name_input, space_name_input, list_spaces_display, file_browser_dropdown, space_iframe_display, formatted_space_output_display, detected_files_display, build_status_display, edit_status_display, space_runtime_status_display, changeset_state, changeset_display, confirm_accordion, confirm_button, cancel_button, target_owner_input, target_space_name_input, target_private_checkbox] (19 items)
+    yield (gr.update(value=_status), gr.update(value=""), gr.update(value=""), gr.update(value="*Listing spaces...*"), gr.update(visible=False, choices=[], value=None), gr.update(value=None, visible=False), gr.update(value="*Load or create a space to see its definition.*"), gr.update(value="*A preview of the latest file versions will appear here.*"), gr.update(value="*Manual build status...*"), gr.update(value="*Select a file...*"), gr.update(value="*Runtime/Repo status will appear here.*"), [], gr.update(value="*No changes proposed.*"), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(value=""), gr.update(value=""), gr.update(value=False))
 
 
     token, token_err = build_logic_get_api_token(hf_api_key_ui)
     if token_err:
         _status = f"Detection Error: {token_err}"
         # Yield error status, keep fields cleared, update list display error
-        yield (gr.update(value=_status), gr.update(), gr.update(), gr.update(value="*Could not list spaces due to token error.*"), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
+        yield (gr.update(value=_status), gr.update(), gr.update(), gr.update(value=f"*Could not list spaces due to token error: {token_err}*"), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
         return
 
     try:
@@ -907,7 +931,8 @@ def handle_detect_user_and_list_spaces(hf_api_key_ui):
         else:
             list_spaces_update = gr.update(value=list_spaces_md)
 
-        yield (gr.update(value=_status), owner_update, gr.update(value=""), list_spaces_update, gr.update(visible=False, choices=[], value=None), gr.update(value=None, visible=False), gr.update(value=""), gr.update(value=""), gr.update(value="*Manual build status...*"), gr.update(value="*Select a file...*"), gr.update(value="*Runtime/Repo status will appear here.*"), [], gr.update(value="*No changes proposed.*"), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(value=""), gr.update(value=""), gr.update(value=False))
+        # Final yield: Update status, owner, list spaces, keep others cleared/defaulted
+        yield (gr.update(value=_status), owner_update, gr.update(value=""), list_spaces_update, gr.update(visible=False, choices=[], value=None), gr.update(value=None, visible=False), gr.update(value="*Load or create a space to see its definition.*"), gr.update(value="*A preview of the latest file versions will appear here.*"), gr.update(value="*Manual build status...*"), gr.update(value="*Select a file...*"), gr.update(value="*Runtime/Repo status will appear here.*"), [], gr.update(value="*No changes proposed.*"), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(value=""), gr.update(value=""), gr.update(value=False))
 
 
     except Exception as e:
@@ -915,7 +940,8 @@ def handle_detect_user_and_list_spaces(hf_api_key_ui):
         print(f"Error in handle_detect_user: {e}")
         import traceback
         traceback.print_exc()
-        yield (gr.update(value=_status), gr.update(), gr.update(), gr.update(value="*Could not list spaces due to user detection error.*"), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
+        # Final yield on error: Update status, keep others cleared/defaulted, list spaces error
+        yield (gr.update(value=_status), gr.update(), gr.update(), gr.update(value=f"*Could not list spaces due to user detection error: {e}*"), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
 
 
 def handle_load_existing_space(hf_api_key_ui, ui_owner_name, ui_space_name):
@@ -926,56 +952,64 @@ def handle_load_existing_space(hf_api_key_ui, ui_owner_name, ui_space_name):
     _changeset_clear = []
     _changeset_summary_clear = "*No changes proposed.*"
     _confirm_ui_hidden = gr.update(visible=False)
-    _list_spaces_display_placeholder = "*List of spaces will appear here.*"
+    _list_spaces_display_placeholder = gr.update(value="*List of spaces will appear here.*")
     _target_owner_clear, _target_space_clear, _target_private_clear = gr.update(value=""), gr.update(value=""), gr.update(value=False)
 
 
+    # outputs list: [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, list_spaces_display, target_owner_input, target_space_name_input, target_private_checkbox] (20 items)
     yield (
         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,
-        gr.update(value=_list_spaces_display_placeholder), _target_owner_clear, _target_space_clear, _target_private_clear
+        _list_spaces_display_placeholder, _target_owner_clear, _target_space_clear, _target_private_clear
     )
 
     owner_to_use = ui_owner_name
     token, token_err = build_logic_get_api_token(hf_api_key_ui)
     if token_err:
         _status_val = f"Load Error: {token_err}"
-        yield (gr.update(value=_status_val), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
+        # Yield only status and keep others as initial state
+        yield (gr.update(), gr.update(), gr.update(value=_status_val), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
         return
     if not owner_to_use:
         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.")
-            yield gr.update(value=owner_to_use), gr.update(value=f"Loading Space: {owner_to_use}/{ui_space_name} (Auto-detected owner)..."), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
+            # Yield updated owner and status, keep others as initial state
+            yield (gr.update(), gr.update(), gr.update(value=f"Loading Space: {owner_to_use}/{ui_space_name} (Auto-detected owner)..."), gr.update(), gr.update(value=owner_to_use), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
         except Exception as e:
             _status_val = f"Load Error: Error auto-detecting owner: {e}"
-            yield (gr.update(value=_status_val), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
+            # Yield only status and keep others as initial state
+            yield (gr.update(), gr.update(), gr.update(value=_status_val), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
             return
 
     if not owner_to_use or not ui_space_name:
         _status_val = "Load Error: Owner and Space Name are required."
-        yield (gr.update(value=_status_val), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
+        # Yield only status and keep others as initial state
+        yield (gr.update(), gr.update(), gr.update(value=_status_val), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
         return
 
     sdk, file_list, err = get_space_repository_info(hf_api_key_ui, ui_space_name, owner_to_use)
 
-    yield gr.update(value=owner_to_use), gr.update(value=ui_space_name), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
+    # Yield updated owner and space textboxes, keep others as initial state
+    yield (gr.update(), gr.update(), gr.update(), gr.update(), gr.update(value=owner_to_use), gr.update(value=ui_space_name), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
+
 
     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)
+        # Yield final state on error
         yield (
             gr.update(value=_formatted), gr.update(value=_detected), gr.update(value=_status_val),
-            gr.update(visible=False, choices=[], value=None),
-            gr.update(), gr.update(),
-            gr.update(value=None, visible=False),
-            _download, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
-            gr.update(value=_list_spaces_display_placeholder), gr.update(value=""), gr.update(value=""), gr.update(value=False)
+            gr.update(visible=False, choices=[], value=None), # Clear and hide file browser
+            gr.update(), gr.update(), # Keep owner/space textboxes
+            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(), # Build/Edit/Runtime status (from initial clear state), Changeset, Confirm UI
+            gr.update(value="*List of spaces will appear here.*"), gr.update(value=""), gr.update(value=""), gr.update(value=False) # List Spaces and Duplicate fields (from initial clear state)
         )
         return
 
@@ -1004,13 +1038,16 @@ def handle_load_existing_space(hf_api_key_ui, ui_owner_name, ui_space_name):
 
     runtime_status_md = handle_refresh_space_status(hf_api_key_ui, owner_to_use, ui_space_name)
 
+    # Final yield after success
     yield (
-        gr.update(value=_formatted), gr.update(value=_detected), gr.update(value=_status_val),
-        file_browser_update,
-        gr.update(), gr.update(),
-        iframe_update,
-        _download, gr.update(), gr.update(), gr.update(value=runtime_status_md), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
-        gr.update(value=_list_spaces_display_placeholder), gr.update(value=""), gr.update(value=""), gr.update(value=False)
+        gr.update(value=_formatted), gr.update(value=_detected), gr.update(value=_status_val), # Markdown, preview, status
+        file_browser_update, # File browser
+        gr.update(), gr.update(), # Keep owner/space textboxes
+        iframe_update, # Iframe
+        _download, # Download button
+        gr.update(), gr.update(), gr.update(value=runtime_status_md), # Build/Edit status (from initial clear state), Runtime status
+        gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), # Changeset, Confirm UI (from initial clear state)
+        gr.update(value="*List of spaces will appear here.*"), gr.update(value=""), gr.update(value=""), gr.update(value=False) # List Spaces and Duplicate fields (from initial clear state)
     )
 
 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):
@@ -1021,10 +1058,10 @@ 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)
 
+    # Outputs list: [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, space_runtime_status_display] (14 items)
     yield (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(), gr.update())
-
+           gr.update(), gr.update(), gr.update(), gr.update()) # Initial yield
 
     if not ui_space_name_part or "/" in ui_space_name_part:
         _build_status = f"Build Error: Invalid Space Name '{ui_space_name_part}'."
@@ -1129,11 +1166,11 @@ def handle_build_space_button(hf_api_key_ui, ui_space_name_part, ui_owner_name_p
 
 def handle_load_file_for_editing(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, selected_file_path):
     if not selected_file_path:
-        return "", "Select a file.", "", gr.update(language="text") # Changed plaintext to text
+        return "", "Select a file.", "", gr.update(language="text")
 
     content, err = get_space_file_content(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, selected_file_path)
     if err:
-        return "", f"Load Error: {err}", "", gr.update(language="text") # Changed plaintext to text
+        return "", f"Load Error: {err}", "", gr.update(language="text")
 
     lang = _infer_lang_from_filename(selected_file_path)
     commit_msg = f"Update {selected_file_path}"
@@ -1176,9 +1213,9 @@ 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 ui_owner_name_part or not ui_space_name_part:
-         return "Delete Error: Cannot delete file. Please load a space first.", gr.update(), "", "", "text", gr.update(), gr.update(), gr.update() # Changed plaintext to text
+         return "Delete Error: Cannot delete file. Please load a space first.", gr.update(), "", "", "text", gr.update(), gr.update(), gr.update()
     if not file_to_delete_path:
-        return "Delete Error: No file selected to delete.", gr.update(), "", "", "text", gr.update(), gr.update(), gr.update() # Changed plaintext to text
+        return "Delete Error: No file selected to delete.", gr.update(), "", "", "text", 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)
 
@@ -1192,7 +1229,7 @@ def handle_delete_file(hf_api_key_ui, ui_space_name_part, ui_owner_name_part, fi
         parsed_code_blocks_state_cache = [b for b in parsed_code_blocks_state_cache if b["filename"] != file_to_delete_path]
         file_content_editor_update = gr.update(value="")
         commit_message_update = gr.update(value="")
-        editor_lang_update = gr.update(language="text") # Changed plaintext to text
+        editor_lang_update = gr.update(language="text")
         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 []), value=None)
 
@@ -1290,7 +1327,7 @@ def handle_manual_duplicate_space(hf_api_key_ui, source_owner, source_space_name
         return "Duplicate Error: Target Owner and Target Space Name are required.", gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
     if "/" in target_space_name:
          return "Duplicate Error: Target Space Name should not contain '/'. Use Target Owner field for the owner part.", gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
-    global parsed_code_blocks_state_cache
+
 
     source_repo_id = f"{source_owner}/{source_space_name}"
     target_repo_id = f"{target_owner}/{target_space_name}"
@@ -1312,6 +1349,7 @@ def handle_manual_duplicate_space(hf_api_key_ui, source_owner, source_space_name
 
     if err_list:
         reload_error = f"Error reloading file list after duplication: {err_list}"
+        global parsed_code_blocks_state_cache
         parsed_code_blocks_state_cache = []
         _file_browser_update = gr.update(visible=False, choices=[], value=None)
         _iframe_html_update = gr.update(value=None, visible=False)
@@ -1323,6 +1361,7 @@ def handle_manual_duplicate_space(hf_api_key_ui, source_owner, source_space_name
             is_binary = lang == "binary" or (err_get is not None)
             code = f"[Error loading content: {err_get}]" if err_get else (content or "")
             loaded_files.append({"filename": file_path, "code": code, "language": lang, "is_binary": is_binary, "is_structure_block": False})
+        global parsed_code_blocks_state_cache
         parsed_code_blocks_state_cache = loaded_files
 
         _file_browser_update = gr.update(visible=True, choices=sorted([f["filename"] for f in parsed_code_blocks_state_cache if not f.get("is_structure_block")] or []), value=None)