Update app.py

app.py

@@ -6,11 +6,11 @@ import io
 def fetch_github_file(github_url, ssh_private_key):
     try:
         # Parse the GitHub URL
-        parts = github_url.split('/')
-        owner = parts[3]
-        repo = parts[4]
-        branch = parts[6]
-        file_path = '/'.join(parts[7:])
+        parts = github_url.split(':')[1].split('/')
+        owner = parts[0]
+        repo = parts[1].split('.git')[0]
+        branch = 'main'  # You might want to make this configurable
+        file_path = ''  # You might want to make this configurable
 
         # Set up SSH client
         ssh = paramiko.SSHClient()
@@ -35,26 +35,24 @@ def fetch_github_file(github_url, ssh_private_key):
         ssh.close()
 
         if not file_content:
-            return {"error": "Error: File not found or empty."}
+            return "Error: File not found or empty."
 
-        return {file_path: file_content}
+        return file_content
     except paramiko.AuthenticationException:
-        return {"error": "Error: Authentication failed. Please check your SSH key."}
+        return "Error: Authentication failed. Please check your SSH key."
     except paramiko.SSHException as ssh_err:
-        return {"error": f"SSH Error: {str(ssh_err)}"}
+        return f"SSH Error: {str(ssh_err)}"
     except Exception as e:
-        return {"error": f"Error accessing GitHub: {str(e)}"}
+        return f"Error accessing GitHub: {str(e)}"
 
 def process_with_gemini(file_content, gemini_api_key):
     genai.configure(api_key=gemini_api_key)
     model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
     
-    combined_content = "\n\n".join([f"File: {name}\n{content}" for name, content in file_content.items()])
-    
     prompt = f"""
     Analyze the following file content for open-source license information:
 
-    {combined_content}
+    {file_content}
 
     Please provide:
     1. A list of dependencies and their versions
@@ -72,15 +70,15 @@ def process_input(file, github_url, ssh_private_key, gemini_api_key):
         return "Error: Please either upload a file OR provide a GitHub URL, not both."
     
     if file is not None:
-        file_content = {file.name: file.read().decode('utf-8')}
+        file_content = file.decode('utf-8')
     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_url.startswith("git@github.com:"):
+            return "Error: Invalid GitHub URL. Please use the format: git@github.com:username/repository.git"
         if not ssh_private_key.strip():
             return "Error: SSH Private Key is empty. Please provide a valid key."
         file_content = fetch_github_file(github_url, ssh_private_key)
-        if isinstance(file_content, dict) and "error" in file_content:
-            return file_content["error"]
+        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."