Update app.py

app.py

@@ -2,10 +2,13 @@ import gradio as gr
 import requests
 import json
 import re
-from github import Github, GithubException, Auth
+from github import Github, GithubException
 import google.generativeai as genai
+import paramiko
+import base64
+import os
 
-def extract_licenses(file_content, github_url, github_token, gemini_api_key):
+def extract_licenses(file_content, github_url, ssh_private_key, gemini_api_key):
     # Parse the dependency file
     dependencies = parse_dependency_file(file_content)
     
@@ -88,57 +91,60 @@ def enrich_with_gemini(licenses, probable_packages, api_key):
     response = model.generate_content(prompt)
     return response.text
 
-def fetch_github_info(github_url, github_token):
+def fetch_github_info(github_url, ssh_private_key):
     try:
-        # Create an authentication object
-        auth = Auth.Token(github_token)
-        
-        # Create a Github instance with authentication
-        g = Github(auth=auth)
-        
-        # Extract the repository name and file path from the URL
-        _, _, _, owner, repo, _, *path_parts = github_url.split('/')
-        repo_name = f"{owner}/{repo}"
-        file_path = '/'.join(path_parts)
-        
-        print(f"Attempting to access file: {file_path} in repository: {repo_name}")
-        
-        # Get the repository
-        repo = g.get_repo(repo_name)
-        
-        # Get the file contents
-        contents = repo.get_contents(file_path)
-        file_content = contents.decoded_content.decode('utf-8')
-        print(f"Successfully retrieved {file_path}")
-        
+        # Parse the GitHub URL
+        parts = github_url.split('/')
+        owner = parts[3]
+        repo = parts[4]
+        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 temporary file to store the SSH private key
+        with open('temp_key', 'w') as key_file:
+            key_file.write(ssh_private_key)
+
+        # 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')
+
+        if not file_content:
+            return "Error: File not found or empty."
+
         return file_content
-    except GithubException as e:
-        if e.status == 404:
-            return f"Error: File or repository not found. Please check the URL and ensure you have the correct access permissions. Details: {str(e)}"
-        else:
-            return f"Error accessing GitHub: {str(e)}"
     except Exception as e:
-        return f"Unexpected error: {str(e)}"
+        return f"Error accessing GitHub: {str(e)}"
 
-def process_input(file, github_url, github_token, gemini_api_key):
+def process_input(file, github_url, ssh_private_key, gemini_api_key):
     if file is not None and github_url:
         return "Error: Please either upload a file OR provide a GitHub URL, not both."
     
     if file is not None:
         file_content = file.decode('utf-8')
-    elif github_url and github_token:
+    elif github_url and ssh_private_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 github_token.strip():
-            return "Error: GitHub Personal Access Token is empty. Please provide a valid token."
-        file_content = fetch_github_info(github_url, github_token)
+        if not ssh_private_key.strip():
+            return "Error: SSH Private Key is empty. Please provide a valid key."
+        file_content = fetch_github_info(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 access token."
+        return "Error: Please either upload a file OR provide both GitHub URL and SSH Private Key."
     
     try:
-        return extract_licenses(file_content, github_url, github_token, gemini_api_key)
+        return extract_licenses(file_content, github_url, ssh_private_key, gemini_api_key)
     except Exception as e:
         return f"Error processing the file: {str(e)}"
 
@@ -147,7 +153,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="GitHub 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"),