Update app.py

app.py

@@ -1,9 +1,6 @@
 import gradio as gr
-import requests
-import json
-import paramiko
-import os
 import google.generativeai as genai
+from github import Github, GithubException
 
 def fetch_github_file(github_url, ssh_private_key):
     try:
@@ -14,29 +11,32 @@ def fetch_github_file(github_url, ssh_private_key):
         branch = parts[6]
         file_path = '/'.join(parts[7:])
 
-        # Set up SSH client
-        ssh = paramiko.SSHClient()
-        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+        # Create a Github instance using SSH key or Personal Access Token
+        g = Github(login_or_token=ssh_private_key)
 
-        # Create a temporary file to store the SSH private key
-        with open('temp_key', 'w') as key_file:
-            key_file.write(ssh_private_key)
+        # Get the repository
+        repository = g.get_repo(f"{owner}/{repo}")
 
-        # Connect to GitHub using SSH
-        ssh.connect('github.com', username='git', key_filename='temp_key')
-
-        # Execute git command to fetch file content
-        stdin, stdout, stderr = ssh.exec_command(f"git archive [email protected]:{owner}/{repo}.git {branch} {file_path} | tar -xO")
-        file_content = stdout.read().decode('utf-8')
-
-        # Close SSH connection and remove temporary key file
-        ssh.close()
-        os.remove('temp_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
 
         if not file_content:
-            return "Error: File not found or empty."
+            return "Error: File is 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 Exception as e:
         return f"Error accessing GitHub: {str(e)}"
 
@@ -70,12 +70,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 is empty. Please provide a valid key."
+            return "Error: SSH Private Key or Personal Access Token is empty. Please provide a valid key or token."
         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."
+        return "Error: Please either upload a file OR provide both GitHub URL and SSH Private Key or Personal Access Token."
     
     try:
         # Process the file content with Gemini
@@ -89,7 +89,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 (required if using GitHub URL)", type="password"),
+        gr.Textbox(label="SSH Private Key or Personal Access Token (required if using GitHub URL)", type="password"),
         gr.Textbox(label="Gemini API Key", type="password"),
     ],
     outputs=gr.Textbox(label="License Information and Analysis"),