Update app.py

app.py

@@ -1,11 +1,10 @@
 import gradio as gr
 import os
 import git
-from huggingface_hub import snapshot_download, login
 import shutil
 
-# Function to clone a GitHub repository
-def download_github_repo(repo_url):
+# Function to clone a GitHub or Hugging Face Space repository
+def clone_repo(repo_url):
     try:
         # Extract repo name from URL
         repo_name = repo_url.split('/')[-1]
@@ -13,40 +12,17 @@ def download_github_repo(repo_url):
         # Clone the repository
         if not os.path.exists(repo_name):
             git.Repo.clone_from(repo_url, repo_name)
-            return f"GitHub repository '{repo_name}' downloaded successfully."
+            return f"Repository '{repo_name}' cloned successfully from {repo_url}."
         else:
-            return f"GitHub repository '{repo_name}' already exists locally."
-    except Exception as e:
-        return f"An error occurred: {str(e)}"
-
-# Function to download a Hugging Face Space or model/dataset using an environment variable for the token
-def download_huggingface_repo(repo_id):
-    try:
-        # Get Hugging Face token from environment variable
-        hf_token = os.getenv("HF_TOKEN")
-        if hf_token is None:
-            return "Hugging Face token not found. Please set the HF_TOKEN environment variable."
-
-        # Log in using the Hugging Face token
-        login(token=hf_token)
-        
-        # Check if it's a Hugging Face Space
-        if repo_id.startswith("spaces/"):
-            repo_path = snapshot_download(repo_id, repo_type="space", token=hf_token)
-        else:
-            # Download the Hugging Face model or dataset snapshot
-            repo_path = snapshot_download(repo_id, token=hf_token)
-
-        return f"Hugging Face repository '{repo_id}' downloaded successfully."
+            return f"Repository '{repo_name}' already exists locally."
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
 # Gradio Interface
-def download_repo(repo_type, repo_url_or_id):
-    if repo_type == "GitHub":
-        return download_github_repo(repo_url_or_id)
-    elif repo_type == "Hugging Face":
-        return download_huggingface_repo(repo_url_or_id)
+def download_repo(repo_type, repo_url):
+    # Handle both GitHub and Hugging Face Space repositories with git clone
+    if repo_type in ["GitHub", "Hugging Face"]:
+        return clone_repo(repo_url)
     else:
         return "Invalid repository type selected."
 
@@ -55,11 +31,11 @@ interface = gr.Interface(
     fn=download_repo,
     inputs=[
         gr.Radio(choices=["GitHub", "Hugging Face"], label="Repository Type"),
-        gr.Textbox(label="Repository URL (GitHub) or Repo ID (Hugging Face)")
+        gr.Textbox(label="Repository URL (GitHub or Hugging Face)")
     ],
     outputs="text",
-    title="Repository Downloader",
-    description="Enter the GitHub repo URL or Hugging Face repo ID to download."
+    title="Repository Cloner",
+    description="Enter the GitHub repo URL or Hugging Face Space URL to clone."
 )
 
 # Launch the app