Update app.py

app.py

@@ -2,7 +2,11 @@ import gradio as gr
 import os
 import time
 import markdown
-from github_repo_analyzer import main as analyze_repo, get_repo_info
+import tempfile
+import traceback
+from github_repo_analyzer import get_repo_info, clone_repo, analyze_code, analyze_issues, analyze_pull_requests, llm_analyze_code, llm_analyze_issues, llm_analyze_prs, llm_synthesize_findings, generate_report
+from github import Github
+from openai import OpenAI
 
 # Emojis and fun statements for progress updates
 PROGRESS_STEPS = [
@@ -18,6 +22,14 @@ def analyze_github_repo(repo_input, github_token=None):
     if github_token:
         os.environ["GITHUB_TOKEN"] = github_token
     
+    github_token = os.environ.get("GITHUB_TOKEN")
+    if not github_token:
+        return "<p>❌ Error: GITHUB_TOKEN environment variable not set.</p>", ""
+
+    openrouter_api_key = os.environ.get("OPENROUTER_API_KEY")
+    if not openrouter_api_key:
+        return "<p>❌ Error: OPENROUTER_API_KEY environment variable not set.</p>", ""
+
     progress_html = ""
     yield progress_html, ""  # Initial empty output
 
@@ -28,17 +40,59 @@ def analyze_github_repo(repo_input, github_token=None):
 
     try:
         owner, repo_name = get_repo_info(repo_input)
-        max_issues = 10
-        max_prs = 10
+        repo_url = f"https://github.com/{owner}/{repo_name}"
         
-        report = analyze_repo(repo_input, max_issues, max_prs)
+        g = Github(github_token)
+        github_repo = g.get_repo(f"{owner}/{repo_name}")
         
+        client = OpenAI(api_key=openrouter_api_key, base_url="https://openrouter.ai/api/v1")
+
+        max_issues = 10
+        max_prs = 10
+
+        with tempfile.TemporaryDirectory() as temp_dir:
+            repo_path = clone_repo(owner, repo_name, temp_dir)
+
+            progress_html += "<p>🔬 Analyzing code structure...</p>"
+            yield progress_html, ""
+            code_analysis = analyze_code(repo_path)
+            code_analysis['llm_analysis'] = llm_analyze_code(client, code_analysis)
+
+            progress_html += f"<p>📊 Analyzing issues (max {max_issues})...</p>"
+            yield progress_html, ""
+            issues_data = analyze_issues(github_repo, max_issues)
+            issues_analysis = llm_analyze_issues(client, issues_data, repo_url)
+
+            progress_html += f"<p>🔀 Analyzing pull requests (max {max_prs})...</p>"
+            yield progress_html, ""
+            prs_data = analyze_pull_requests(github_repo, max_prs)
+            pr_analysis = llm_analyze_prs(client, prs_data, repo_url)
+
+            progress_html += "<p>🧠 Synthesizing findings...</p>"
+            yield progress_html, ""
+            final_analysis = llm_synthesize_findings(
+                client, 
+                code_analysis.get('llm_analysis', ''),
+                issues_analysis.get('summary', ''),
+                pr_analysis.get('summary', '')
+            )
+
+            repo_info = {
+                "owner": owner,
+                "repo_name": repo_name,
+            }
+
+            progress_html += "<p>📝 Generating report...</p>"
+            yield progress_html, ""
+            report = generate_report(repo_info, code_analysis, issues_analysis, pr_analysis, final_analysis)
+
         # Convert markdown to HTML
         html_report = markdown.markdown(report)
         
         return progress_html + "<p>✅ Analysis complete!</p>", html_report
     except Exception as e:
         error_message = f"<p>❌ An error occurred: {str(e)}</p>"
+        traceback.print_exc()
         return progress_html + error_message, ""
 
 # Define the Gradio interface
@@ -63,4 +117,4 @@ with gr.Blocks() as app:
 
 # Launch the app
 if __name__ == "__main__":
-    app.launch()
+    app.launch()