Upload 2_upload_single.py with huggingface_hub

2_upload_single.py

@@ -0,0 +1,116 @@
+# from huggingface_hub import HfApi
+# import os
+# import sys
+
+# def verify_authentication():
+#     api = HfApi()
+#     try:
+#         user_info = api.whoami()
+#         print(f"Authenticated as: {user_info['name']}")
+#         return True
+#     except Exception as e:
+#         print("Authentication failed. Please ensure you're logged in to Hugging Face.")
+#         print(f"Error: {e}")
+#         return False
+
+# def upload_to_hf_space(api, local_path, repo_id, path_in_repo):
+#     try:
+#         url = api.upload_file(
+#             path_or_fileobj=local_path,
+#             path_in_repo=path_in_repo,
+#             repo_id=repo_id,
+#             token=os.environ.get("HF_TOKEN"),
+#             repo_type="space",
+#         )
+#         print(f"File uploaded successfully: {local_path} -> {url}")
+#     except Exception as e:
+#         print(f"An error occurred while uploading {local_path}: {e}")
+
+# def upload_folder(api, repo_id, local_dir, folder_name):
+#     folder_path = os.path.join(local_dir, folder_name)
+#     for root, dirs, files in os.walk(folder_path):
+#         for file in files:
+#             local_path = os.path.join(root, file)
+#             relative_path = os.path.relpath(local_path, local_dir)
+#             upload_to_hf_space(api, local_path, repo_id, relative_path)
+
+# def upload_items(repo_id, local_dir, items):
+#     if not verify_authentication():
+#         sys.exit(1)
+
+#     api = HfApi()
+
+#     for item in items:
+#         full_path = os.path.join(local_dir, item)
+#         if os.path.isfile(full_path):
+#             upload_to_hf_space(api, full_path, repo_id, item)
+#         elif os.path.isdir(full_path):
+#             upload_folder(api, repo_id, local_dir, item)
+#         else:
+#             print(f"The item {item} does not exist in {local_dir}.")
+
+# if __name__ == "__main__":
+#     repo_id = "liuganghuggingface/polymer-design"
+#     local_dir = "./"  # Use the current directory
+#     items_to_upload = ["app.py", "requirements.txt", "model_labeled", "model_all"]  # List of files and folders to upload
+
+#     # Ensure the HF_TOKEN is set
+#     if "HF_TOKEN" not in os.environ:
+#         print("Please set the HF_TOKEN environment variable with your Hugging Face token.")
+#         sys.exit(1)
+
+#     upload_items(repo_id, local_dir, items_to_upload)
+
+
+
+from huggingface_hub import HfApi
+import os
+import sys
+
+def upload_single_file(api, repo_id, local_path, remote_path):
+    print(f"Attempting to upload: {local_path} to {remote_path}")
+    try:
+        if not os.path.exists(local_path):
+            print(f"Error: Local file does not exist: {local_path}")
+            return
+
+        url = api.upload_file(
+            path_or_fileobj=local_path,
+            path_in_repo=remote_path,
+            repo_id=repo_id,
+            token=os.environ.get("HF_TOKEN"),
+            repo_type="space",
+        )
+        print(f"File uploaded successfully: {local_path} -> {url}")
+        
+        # Verify the upload
+        files = api.list_repo_files(repo_id=repo_id, repo_type="space")
+        if remote_path in files:
+            print(f"Verified: {remote_path} is present in the repository.")
+        else:
+            print(f"Warning: {remote_path} was uploaded but not found in the repository listing.")
+    except Exception as e:
+        print(f"An error occurred while uploading {local_path}: {e}")
+
+def main():
+    repo_id = "liuganghuggingface/polymer-design"
+    local_dir = "./"  # Use the current directory
+    
+    files_to_upload = [
+        ("model_labeled/data.meta.json", "model_labeled/data.meta.json"),
+        ("model_all/config.yaml", "model_all/config.yaml"),
+        ("model_all/model.pt", "model_all/model.pt")
+    ]
+
+    api = HfApi()
+
+    for local_path, remote_path in files_to_upload:
+        full_local_path = os.path.join(local_dir, local_path)
+        upload_single_file(api, repo_id, full_local_path, remote_path)
+
+if __name__ == "__main__":
+    if "HF_TOKEN" not in os.environ:
+        print("Please set the HF_TOKEN environment variable with your Hugging Face token.")
+        sys.exit(1)
+
+    main()