Update app.py

app.py

@@ -6,7 +6,59 @@ import json
 from tenacity import retry, stop_after_attempt, wait_fixed
 
 def fetch_github_files(github_url, personal_access_token):
-    # ... (keep this function as is) ...
+    try:
+        # Parse the GitHub URL
+        parts = github_url.split('/')
+        owner = parts[3]
+        repo = parts[4].split('.git')[0]
+        branch = 'main'  # You might want to make this configurable
+
+        # List of common dependency files to look for
+        dependency_files = [
+            'requirements.txt',
+            'package.json',
+            'Gemfile',
+            'pom.xml',
+            'build.gradle',
+            'composer.json',
+            'Cargo.toml',
+            'go.mod',
+            'Pipfile'
+        ]
+
+        all_content = ""
+
+        # Set up headers with the personal access token
+        headers = {
+            "Authorization": f"token {personal_access_token}",
+            "Accept": "application/vnd.github.v3+json"
+        }
+
+        for file_path in dependency_files:
+            # Construct the API URL
+            api_url = f"https://api.github.com/repos/{owner}/{repo}/contents/{file_path}?ref={branch}"
+
+            # Make the API request
+            response = requests.get(api_url, headers=headers)
+
+            if response.status_code == 200:
+                content = response.json()
+                if isinstance(content, dict) and 'content' in content:
+                    # This is a file
+                    file_content = base64.b64decode(content['content']).decode('utf-8')
+                    all_content += f"\n\n--- {file_path} ---\n{file_content}"
+                else:
+                    # This is a directory or something else, skip it
+                    continue
+
+        if not all_content:
+            return "Error: No dependency files found in the repository."
+
+        return all_content
+    except requests.exceptions.RequestException as e:
+        return f"Error accessing GitHub: {str(e)}"
+    except json.JSONDecodeError:
+        return f"Error: Unable to parse GitHub API response for {file_path}"
 
 def process_chunk_with_gemini(chunk, gemini_api_key):
     genai.configure(api_key=gemini_api_key)