Update build_logic.py

build_logic.py

@@ -14,9 +14,9 @@ from huggingface_hub import (
     delete_file as hf_delete_file,
     HfApi
 )
-# More direct import for CommitOperationDelete in case of re-export issues
 from huggingface_hub.hf_api import CommitOperationDelete
-from huggingface_hub.utils import SpaceNotFoundError, GatedRepoError
+# Import the general HTTP error from huggingface_hub.utils
+from huggingface_hub.utils import HfHubHTTPError # For catching specific HF HTTP errors
 
 # Setup basic logging
 logging.basicConfig(
@@ -139,18 +139,41 @@ def get_space_repository_info(ui_api_token_from_textbox, space_name_ui, owner_ui
         files = [sibling.rfilename for sibling in repo_info_obj.siblings if sibling.rfilename]
         if not files and repo_info_obj.siblings:
             logger.warning(f"Repo {repo_id} has siblings but no rfilenames extracted.")
-    except Exception as e:
+    except HfHubHTTPError as e_http: # Catch specific HF HTTP errors first
+        logger.error(f"HTTP error getting repo info for {repo_id_for_error_logging}: {e_http}")
+        error_message = str(e_http)
+        if e_http.response is not None: # Check if response object exists
+            if e_http.response.status_code == 404:
+                error = f"Space '{repo_id_for_error_logging}' not found (404)."
+            elif e_http.response.status_code in:
+                error = f"Access denied for '{repo_id_for_error_logging}' (401/403). Check token permissions."
+            else:
+                error = f"HTTP Error {e_http.response.status_code} for '{repo_id_for_error_logging}': {error_message}"
+        else: # No response object, use generic message
+            error = f"Network or HTTP Error for '{repo_id_for_error_logging}': {error_message}"
+    except Exception as e: # Catch other general exceptions
         logger.warning(f"Could not get full repo_info for {repo_id_for_error_logging}, falling back to list_repo_files: {e}")
         try:
             resolved_api_token_fb, token_err_fb = _get_api_token(ui_api_token_from_textbox)
-            if token_err_fb: return None, None, token_err_fb
+            if token_err_fb: return None, None, token_err_fb # Propagate error
             repo_id_fb, err_repo_id_fb = _determine_repo_id(ui_api_token_from_textbox, space_name_ui, owner_ui)
-            if err_repo_id_fb: return None, None, err_repo_id_fb
+            if err_repo_id_fb: return None, None, err_repo_id_fb # Propagate error
             files = list_repo_files(repo_id=repo_id_fb, token=resolved_api_token_fb, repo_type="space")
+        except HfHubHTTPError as e2_http:
+            logger.error(f"HTTP error during fallback list_repo_files for {repo_id_for_error_logging}: {e2_http}")
+            error_message_fb = str(e2_http)
+            if e2_http.response is not None:
+                 if e2_http.response.status_code == 404:
+                    error = f"Space '{repo_id_for_error_logging}' not found during fallback (404)."
+                 else:
+                    error = f"HTTP Error {e2_http.response.status_code} for '{repo_id_for_error_logging}' during fallback: {error_message_fb}"
+            else:
+                 error = f"Network or HTTP Error for '{repo_id_for_error_logging}' during fallback: {error_message_fb}"
         except Exception as e2:
             logger.exception(f"Error listing files for {repo_id_for_error_logging} during fallback: {e2}")
             error = f"Error listing files for `{repo_id_for_error_logging}`: {str(e2)}"
 
+
     if not files and not error:
         error = f"No files found in Space `{repo_id_for_error_logging}` (or an issue fetching them)."
     return sdk, files, error
@@ -175,9 +198,13 @@ def get_space_file_content(ui_api_token_from_textbox, space_name_ui, owner_ui, f
         downloaded_file_path = hf_hub_download(repo_id=repo_id, filename=file_path_in_repo, repo_type="space", token=resolved_api_token)
         content = Path(downloaded_file_path).read_text(encoding="utf-8")
         return content, None
+    except HfHubHTTPError as e_http:
+        logger.error(f"HTTP error fetching file {file_path_in_repo} from {repo_id_for_error_logging}: {e_http}")
+        error_message = str(e_http)
+        if e_http.response is not None and e_http.response.status_code == 404:
+            return None, f"Error: File '{file_path_in_repo}' not found in Space '{repo_id_for_error_logging}' (404)."
+        return None, f"HTTP Error fetching file '{file_path_in_repo}': {error_message}"
     except Exception as e:
-        if "404" in str(e) or "not found" in str(e).lower() or "does not exist" in str(e).lower():
-             return None, f"Error: File '{file_path_in_repo}' not found in Space '{repo_id_for_error_logging}'."
         logger.exception(f"Error fetching file content for {file_path_in_repo} from {repo_id_for_error_logging}:")
         return None, f"Error fetching file content: {str(e)}"
 
@@ -217,6 +244,7 @@ def create_space(ui_api_token_from_textbox, space_name_ui, owner_ui, sdk_ui, mar
                 files_to_delete_on_hub = list(current_hub_files - markdown_filenames)
                 files_to_delete_on_hub = [f for f in files_to_delete_on_hub if not (f.startswith('.git') or (f == "README.md" and "README.md" not in markdown_filenames))]
 
+
                 if files_to_delete_on_hub:
                     logger.info(f"Deleting {len(files_to_delete_on_hub)} files from {repo_id} not in new markdown: {files_to_delete_on_hub}")
                     delete_operations = [CommitOperationDelete(path_in_repo=f) for f in files_to_delete_on_hub]
@@ -275,9 +303,13 @@ def delete_space_file(ui_api_token_from_textbox, space_name_ui, owner_ui, file_p
         effective_commit_message = commit_message_ui or f"Deleted file: {file_path_in_repo} via AI Space Editor"
         hf_delete_file(path_in_repo=file_path_in_repo, repo_id=repo_id, repo_type="space", token=resolved_api_token, commit_message=effective_commit_message)
         return f"Successfully deleted file: {file_path_in_repo}"
+    except HfHubHTTPError as e_http: # Catch specific HF HTTP errors
+        logger.error(f"HTTP error deleting file {file_path_in_repo} from {repo_id_for_error_logging}: {e_http}")
+        error_message = str(e_http)
+        if e_http.response is not None and e_http.response.status_code == 404:
+             return f"Error: File '{file_path_in_repo}' not found in Space '{repo_id_for_error_logging}' for deletion (404)."
+        return f"HTTP Error deleting file '{file_path_in_repo}': {error_message}"
     except Exception as e:
-        if "404" in str(e) or "not found" in str(e).lower() or "does not exist" in str(e).lower():
-             return f"Error: File '{file_path_in_repo}' not found in Space '{repo_id_for_error_logging}' for deletion."
         logger.exception(f"Error deleting file {file_path_in_repo} from {repo_id_for_error_logging}:")
         return f"Error deleting file '{file_path_in_repo}': {str(e)}"
 
@@ -293,7 +325,7 @@ def get_space_runtime_status(ui_api_token_from_textbox, space_name_ui, owner_ui)
         api = HfApi(token=resolved_api_token)
         logger.info(f"Fetching runtime status for Space: {repo_id}")
         
-        runtime_info = api.get_space_runtime(repo_id=repo_id) # CORRECTED: get_space_runtime
+        runtime_info = api.get_space_runtime(repo_id=repo_id)
         
         status_details = {
             "stage": runtime_info.stage,
@@ -318,13 +350,15 @@ def get_space_runtime_status(ui_api_token_from_textbox, space_name_ui, owner_ui)
 
         logger.info(f"Runtime status for {repo_id}: {status_details['stage']}")
         return status_details, None
+    except HfHubHTTPError as e_http: # Catch specific HF HTTP errors
+        logger.error(f"HTTP error fetching runtime status for {repo_id_for_error_logging}: {e_http}")
+        error_message = str(e_http)
+        if e_http.response is not None:
+            if e_http.response.status_code == 404:
+                return None, f"Error: Space '{repo_id_for_error_logging}' not found or has no active runtime (404)."
+            elif e_http.response.status_code in:
+                 return None, f"Error: Access denied or authentication required for '{repo_id_for_error_logging}' ({e_http.response.status_code})."
+        return None, f"HTTP Error fetching runtime status for '{repo_id_for_error_logging}': {error_message}"
     except Exception as e:
-        # Check for specific Hugging Face Hub exceptions if available, otherwise generic check
-        if "SpaceNotFoundError" in str(type(e)) or "404" in str(e):
-            logger.warning(f"Space not found or no runtime info: {repo_id_for_error_logging} - {e}")
-            return None, f"Error: Space '{repo_id_for_error_logging}' not found or has no active runtime."
-        elif "GatedRepoError" in str(type(e)) or "must be authenticated" in str(e).lower():
-            logger.warning(f"Access denied for Space runtime: {repo_id_for_error_logging} - {e}")
-            return None, f"Error: Access denied or authentication required for '{repo_id_for_error_logging}'."
         logger.exception(f"Error fetching runtime status for {repo_id_for_error_logging}:")
         return None, f"Error fetching runtime status: {str(e)}"