Update app.py

app.py

@@ -1,97 +1,11 @@
 import gradio as gr
 import requests
 import json
-import re
-from github import Github, GithubException
-import google.generativeai as genai
 import paramiko
-import base64
 import os
+import google.generativeai as genai
 
-def extract_licenses(file_content, github_url, ssh_private_key, gemini_api_key):
-    # Parse the dependency file
-    dependencies = parse_dependency_file(file_content)
-    
-    # Fetch license information
-    licenses = fetch_license_info(dependencies)
-    
-    # Determine probable packages
-    probable_packages = determine_probable_packages(file_content, dependencies)
-    
-    # Enrich information using Gemini
-    enriched_info = enrich_with_gemini(licenses, probable_packages, gemini_api_key)
-    
-    return enriched_info
-
-def parse_dependency_file(file_content):
-    dependencies = []
-    lines = file_content.split('\n')
-    for line in lines:
-        if '=' in line or '@' in line or ':' in line:
-            parts = re.split(r'[=@:]', line)
-            package = parts[0].strip()
-            version = parts[1].strip() if len(parts) > 1 else "latest"
-            dependencies.append((package, version))
-    return dependencies
-
-def fetch_license_info(dependencies):
-    licenses = []
-    for package, version in dependencies:
-        try:
-            response = requests.get(f"https://pypi.org/pypi/{package}/{version}/json")
-            data = response.json()
-            license = data['info'].get('license', 'Unknown')
-            description = data['info'].get('summary', 'No description available')
-            licenses.append(f"Package: {package}\nVersion: {version}\nLicense: {license}\nDescription: {description}\n")
-        except:
-            licenses.append(f"Package: {package}\nVersion: {version}\nLicense: Unknown\nDescription: Unable to fetch information\n")
-    return "\n".join(licenses)
-
-def determine_probable_packages(file_content, dependencies):
-    probable_packages = []
-    if "package.json" in file_content.lower():
-        probable_packages.append("npm (Node Package Manager)")
-    elif "gemfile" in file_content.lower():
-        probable_packages.append("Bundler (Ruby)")
-    elif "requirements.txt" in file_content.lower():
-        probable_packages.append("pip (Python Package Installer)")
-    
-    # Add more probable packages based on common dependencies
-    common_packages = {
-        "react": "React (JavaScript library)",
-        "django": "Django (Python web framework)",
-        "rails": "Ruby on Rails (Web application framework)",
-    }
-    
-    for package, _ in dependencies:
-        if package.lower() in common_packages:
-            probable_packages.append(common_packages[package.lower()])
-    
-    return "\n".join(probable_packages)
-
-def enrich_with_gemini(licenses, probable_packages, api_key):
-    genai.configure(api_key=api_key)
-    model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
-    
-    prompt = f"""
-    Analyze the following open-source license information and probable packages:
-
-    License Information:
-    {licenses}
-
-    Probable Packages:
-    {probable_packages}
-
-    Please provide a summary of the project based on these dependencies, including:
-    1. The likely type of project (e.g., web application, data science, etc.)
-    2. Any potential license conflicts or considerations
-    3. Suggestions for best practices in open-source license management for this project
-    """
-
-    response = model.generate_content(prompt)
-    return response.text
-
-def fetch_github_info(github_url, ssh_private_key):
+def fetch_github_file(github_url, ssh_private_key):
     try:
         # Parse the GitHub URL
         parts = github_url.split('/')
@@ -126,6 +40,26 @@ def fetch_github_info(github_url, ssh_private_key):
     except Exception as 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')
+    
+    prompt = f"""
+    Analyze the following file content for open-source license information:
+
+    {file_content}
+
+    Please provide:
+    1. A list of dependencies and their versions
+    2. The licenses associated with each dependency (if available)
+    3. A summary of the project based on these dependencies
+    4. Any potential license conflicts or considerations
+    5. Suggestions for best practices in open-source license management for this project
+    """
+
+    response = model.generate_content(prompt)
+    return response.text
+
 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."
@@ -137,14 +71,16 @@ def process_input(file, github_url, ssh_private_key, gemini_api_key):
             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."
-        file_content = fetch_github_info(github_url, ssh_private_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."
     
     try:
-        return extract_licenses(file_content, github_url, ssh_private_key, gemini_api_key)
+        # Process the file content with Gemini
+        analysis = process_with_gemini(file_content, gemini_api_key)
+        return analysis
     except Exception as e:
         return f"Error processing the file: {str(e)}"