Update tools/space_builder.py

tools/space_builder.py

@@ -21,11 +21,9 @@ def _initialize_client():
 def create_huggingface_space(owner: str, space_name: str, sdk: str, markdown_content: str):
     if not _initialize_client():
         return {"error": "Space builder client could not be initialized."}
-
     hf_token = os.getenv("HF_TOKEN")
     if not hf_token:
         return {"error": "Hugging Face token (HF_TOKEN) is not set in environment variables."}
-
     logger.info(f"SPACE_BUILDER: Calling /create_space for {owner}/{space_name}")
     try:
         result = client.predict(
@@ -36,7 +34,6 @@ def create_huggingface_space(owner: str, space_name: str, sdk: str, markdown_con
             markdown_input=markdown_content,
             api_name="/create_space"
         )
-        logger.info(f"SPACE_BUILDER: API response for creation: {result}")
         return {"result": result}
     except Exception as e:
         logger.error(f"SPACE_BUILDER: An error occurred during space creation: {e}", exc_info=True)
@@ -45,11 +42,9 @@ def create_huggingface_space(owner: str, space_name: str, sdk: str, markdown_con
 def update_huggingface_space_file(owner: str, space_name: str, file_path: str, new_content: str, commit_message: str):
     if not _initialize_client():
         return {"error": "Space builder client could not be initialized."}
-
     hf_token = os.getenv("HF_TOKEN")
     if not hf_token:
         return {"error": "Hugging Face token (HF_TOKEN) is not set in environment variables."}
-
     logger.info(f"SPACE_BUILDER: Calling /update_space_file for {owner}/{space_name}/{file_path}")
     try:
         result = client.predict(
@@ -61,11 +56,47 @@ def update_huggingface_space_file(owner: str, space_name: str, file_path: str, n
             commit_message_ui=commit_message or f"Update {file_path}",
             api_name="/update_space_file"
         )
-        logger.info(f"SPACE_BUILDER: API response for file update: {result}")
         return {"result": result}
     except Exception as e:
         if "is not in the list of choices" in str(e):
-             logger.error(f"SPACE_BUILDER: The file '{file_path}' may not exist or is not selectable in the target space.")
              return {"error": f"The file '{file_path}' could not be selected for update. It may not exist in the space."}
         logger.error(f"SPACE_BUILDER: An error occurred during file update: {e}", exc_info=True)
+        return {"error": f"An API or client error occurred: {e}"}
+
+def list_space_files(owner: str, space_name: str):
+    if not _initialize_client():
+        return {"error": "Space builder client could not be initialized."}
+    hf_token = os.getenv("HF_TOKEN")
+    if not hf_token:
+        return {"error": "Hugging Face token (HF_TOKEN) is not set in environment variables."}
+    logger.info(f"SPACE_BUILDER: Calling /handle_load_space_files_list for {owner}/{space_name}")
+    try:
+        status_msg, file_list, _ = client.predict(
+            token_from_ui=hf_token,
+            space_name=space_name,
+            owner_name=owner,
+            api_name="/handle_load_space_files_list"
+        )
+        return {"status": status_msg, "files": file_list}
+    except Exception as e:
+        logger.error(f"SPACE_BUILDER: An error occurred while listing files: {e}", exc_info=True)
+        return {"error": f"An API or client error occurred: {e}"}
+
+def get_space_file_content(owner: str, space_name: str, file_path: str):
+    if not _initialize_client():
+        return {"error": "Space builder client could not be initialized."}
+    hf_token = os.getenv("HF_TOKEN")
+    if not hf_token:
+        return {"error": "Hugging Face token (HF_TOKEN) is not set in environment variables."}
+    logger.info(f"SPACE_BUILDER: Calling /handle_file_selected_for_editing for {owner}/{space_name}/{file_path}")
+    try:
+        content, status_msg = client.predict(
+            token_from_ui=hf_token,
+            space_name=space_name,
+            owner_name=owner,
+            api_name="/handle_file_selected_for_editing"
+        )
+        return {"status": status_msg, "content": content}
+    except Exception as e:
+        logger.error(f"SPACE_BUILDER: An error occurred while getting file content: {e}", exc_info=True)
         return {"error": f"An API or client error occurred: {e}"}