Update tools/space_builder.py

tools/space_builder.py

@@ -1,49 +1,77 @@
-import requests
-import logging
 import os
+import logging
+from gradio_client import Client, GradioClientError
 
 logger = logging.getLogger(__name__)
 
-SPACE_BUILDER_API_URL = os.getenv("SPACE_BUILDER_API_URL", "https://broadfield-dev-build-space.hf.space/run/predict")
+client = None
+
+def _initialize_client():
+    global client
+    if client is None:
+        try:
+            logger.info("Initializing Gradio client for broadfield-dev/build-space...")
+            client = Client("broadfield-dev/build-space")
+            logger.info("Gradio client initialized successfully.")
+        except Exception as e:
+            logger.error(f"Failed to initialize Gradio client: {e}")
+            client = None
+    return 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(
+            ui_api_token_from_textbox=hf_token,
+            space_name_ui=space_name,
+            owner_ui=owner,
+            sdk_ui=sdk,
+            markdown_input=markdown_content,
+            api_name="/create_space"
+        )
+        logger.info(f"SPACE_BUILDER: API response for creation: {result}")
+        return {"result": result}
+    except GradioClientError as e:
+        logger.error(f"SPACE_BUILDER: API Client Error during space creation: {e}")
+        return {"error": f"API Client Error: {e}"}
+    except Exception as e:
+        logger.error(f"SPACE_BUILDER: An unexpected error occurred during space creation: {e}", exc_info=True)
+        return {"error": f"An unexpected error occurred: {e}"}
 
-def build_huggingface_space(build_prompt: str):
-    """
-    Calls the build-space API to generate a Hugging Face Space.
+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."}
 
-    Args:
-        build_prompt: A string describing the space to be built.
+    hf_token = os.getenv("HF_TOKEN")
+    if not hf_token:
+        return {"error": "Hugging Face token (HF_TOKEN) is not set in environment variables."}
 
-    Returns:
-        A dictionary containing the result or an error message.
-    """
-    logger.info(f"SPACE_BUILDER: Calling API with prompt: '{build_prompt[:100]}...'")
-    payload = {"data": [build_prompt]}
-    
+    logger.info(f"SPACE_BUILDER: Calling /update_space_file for {owner}/{space_name}/{file_path}")
     try:
-        # A long timeout is set as space building can take time.
-        response = requests.post(SPACE_BUILDER_API_URL, json=payload, timeout=300) 
-        response.raise_for_status()
-        
-        response_data = response.json()
-        
-        if "data" in response_data and isinstance(response_data["data"], list) and len(response_data["data"]) > 0:
-            result_text = response_data["data"][0]
-            logger.info(f"SPACE_BUILDER: Successfully received response: {result_text}")
-            return {"prompt": build_prompt, "result": result_text}
-        else:
-            logger.error(f"SPACE_BUILDER: API response in unexpected format: {response_data}")
-            return {"prompt": build_prompt, "error": "API response format is invalid."}
-            
-    except requests.exceptions.HTTPError as e:
-        error_message = f"HTTP error {e.response.status_code}: {e.response.text[:200]}"
-        logger.error(f"SPACE_BUILDER: {error_message}")
-        return {"prompt": build_prompt, "error": error_message}
-    except requests.exceptions.Timeout:
-        logger.error("SPACE_BUILDER: API call timed out.")
-        return {"prompt": build_prompt, "error": "The request timed out."}
-    except requests.exceptions.RequestException as e:
-        logger.error(f"SPACE_BUILDER: Request failed: {e}")
-        return {"prompt": build_prompt, "error": f"Request failed: {e}"}
+        result = client.predict(
+            ui_api_token_from_textbox=hf_token,
+            space_name_ui=space_name,
+            owner_ui=owner,
+            file_path_in_repo=file_path,
+            file_content=new_content,
+            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 GradioClientError 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: API Client Error during file update: {e}")
+        return {"error": f"API Client Error: {e}"}
     except Exception as e:
-        logger.error(f"SPACE_BUILDER: An unexpected error occurred: {e}", exc_info=True)
-        return {"prompt": build_prompt, "error": f"An internal error occurred: {e}"}
+        logger.error(f"SPACE_BUILDER: An unexpected error occurred during file update: {e}", exc_info=True)
+        return {"error": f"An unexpected error occurred: {e}"}