Update tools/space_builder.py

tools/space_builder.py

@@ -1,6 +1,6 @@
 import os
 import logging
-from gradio_client import Client, GradioClientError
+from gradio_client import Client
 
 logger = logging.getLogger(__name__)
 
@@ -38,12 +38,9 @@ def create_huggingface_space(owner: str, space_name: str, sdk: str, markdown_con
         )
         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}"}
+        logger.error(f"SPACE_BUILDER: An error occurred during space creation: {e}", exc_info=True)
+        return {"error": f"An API or client error occurred: {e}"}
 
 def update_huggingface_space_file(owner: str, space_name: str, file_path: str, new_content: str, commit_message: str):
     if not _initialize_client():
@@ -66,12 +63,9 @@ def update_huggingface_space_file(owner: str, space_name: str, file_path: str, n
         )
         logger.info(f"SPACE_BUILDER: API response for file update: {result}")
         return {"result": result}
-    except GradioClientError as e:
+    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: 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 during file update: {e}", exc_info=True)
-        return {"error": f"An unexpected error occurred: {e}"}
+        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}"}