Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import google.generativeai as genai
-from github import Github, GithubException
+import paramiko
+import io
 
 def fetch_github_file(github_url, ssh_private_key):
     try:
@@ -11,32 +12,36 @@ def fetch_github_file(github_url, ssh_private_key):
         branch = parts[6]
         file_path = '/'.join(parts[7:])
 
-        # Create a Github instance using SSH key or Personal Access Token
-        g = Github(login_or_token=ssh_private_key)
+        # Set up SSH client
+        ssh = paramiko.SSHClient()
+        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
 
-        # Get the repository
-        repository = g.get_repo(f"{owner}/{repo}")
+        # Load the private key
+        private_key = paramiko.RSAKey.from_private_key(io.StringIO(ssh_private_key))
 
-        try:
-            # Get the file content
-            file_content = repository.get_contents(file_path, ref=branch).decoded_content.decode('utf-8')
-        except GithubException as e:
-            if e.status == 404:
-                return f"Error: File not found. Please check the path: {file_path}"
-            else:
-                raise
+        # Connect to GitHub using SSH
+        ssh.connect('github.com', username='git', pkey=private_key)
+
+        # Construct the git command to fetch the file content
+        git_command = f"git archive --remote=git@github.com:{owner}/{repo}.git {branch} {file_path} | tar -xO"
+
+        # Execute the command
+        stdin, stdout, stderr = ssh.exec_command(git_command)
+        
+        # Read the file content
+        file_content = stdout.read().decode('utf-8')
+
+        # Close the SSH connection
+        ssh.close()
 
         if not file_content:
-            return "Error: File is empty."
+            return "Error: File not found or empty."
 
         return file_content
-    except GithubException as e:
-        if e.status == 401:
-            return "Error: Authentication failed. Please check your SSH key or token."
-        elif e.status == 404:
-            return f"Error: Repository not found: {owner}/{repo}"
-        else:
-            return f"GitHub Error: {str(e)}"
+    except paramiko.AuthenticationException:
+        return "Error: Authentication failed. Please check your SSH key."
+    except paramiko.SSHException as ssh_err:
+        return f"SSH Error: {str(ssh_err)}"
     except Exception as e:
         return f"Error accessing GitHub: {str(e)}"
 
@@ -70,12 +75,12 @@ def process_input(file, github_url, ssh_private_key, gemini_api_key):
         if not github_url.startswith("https://github.com/"):
             return "Error: Invalid GitHub URL. Please use the format: https://github.com/username/repository/blob/branch/path/to/file"
         if not ssh_private_key.strip():
-            return "Error: SSH Private Key or Personal Access Token is empty. Please provide a valid key or token."
+            return "Error: SSH Private Key is empty. Please provide a valid key."
         file_content = fetch_github_file(github_url, ssh_private_key)
         if file_content.startswith("Error:"):
             return file_content
     else:
-        return "Error: Please either upload a file OR provide both GitHub URL and SSH Private Key or Personal Access Token."
+        return "Error: Please either upload a file OR provide both GitHub URL and SSH Private Key."
     
     try:
         # Process the file content with Gemini
@@ -89,7 +94,7 @@ iface = gr.Interface(
     inputs=[
         gr.File(label="Upload dependency file (e.g., requirements.txt, package.json, Gemfile)"),
         gr.Textbox(label="GitHub File URL (optional)"),
-        gr.Textbox(label="SSH Private Key or Personal Access Token (required if using GitHub URL)", type="password"),
+        gr.Textbox(label="SSH Private Key (required if using GitHub URL)", type="password"),
         gr.Textbox(label="Gemini API Key", type="password"),
     ],
     outputs=gr.Textbox(label="License Information and Analysis"),