Update app.py

app.py

@@ -3,19 +3,53 @@ import os
 import time
 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 = [
-    ("🕵️‍♂️", "Investigating the GitHub realm..."),
-    ("🧬", "Decoding repository DNA..."),
-    ("🐛", "Hunting for bugs and features..."),
-    ("🔍", "Examining pull request tea leaves..."),
-    ("🧠", "Activating AI brain cells..."),
-    ("📝", "Crafting the legendary report..."),
-]
+# Mock functions for testing - replace these with your actual functions
+def get_repo_info(repo_input):
+    return repo_input.split('/')[-2:]
+
+def clone_repo(owner, repo_name, temp_dir):
+    return os.path.join(temp_dir, f"{owner}_{repo_name}")
+
+def analyze_code(repo_path):
+    return {"files": 10, "lines": 1000}
+
+def analyze_issues(github_repo, max_issues):
+    return [{"title": "Test Issue", "number": 1}]
+
+def analyze_pull_requests(github_repo, max_prs):
+    return [{"title": "Test PR", "number": 1}]
+
+def llm_analyze_code(client, code_analysis):
+    return "Code analysis summary"
+
+def llm_analyze_issues(client, issues_data, repo_url):
+    return {"summary": "Issues analysis summary"}
+
+def llm_analyze_prs(client, prs_data, repo_url):
+    return {"summary": "PRs analysis summary"}
+
+def llm_synthesize_findings(client, code_analysis, issues_analysis, pr_analysis):
+    return "Synthesized findings"
+
+def generate_report(repo_info, code_analysis, issues_analysis, pr_analysis, final_analysis):
+    return f"""
+# Repository Analysis Report for {repo_info['owner']}/{repo_info['repo_name']}
+
+## Code Analysis
+{code_analysis['llm_analysis']}
+
+## Issues Analysis
+{issues_analysis['summary']}
+
+## Pull Requests Analysis
+{pr_analysis['summary']}
+
+## Final Analysis
+{final_analysis}
+"""
 
 def analyze_github_repo(repo_input, github_token=None):
     if github_token:
@@ -23,23 +57,20 @@ def analyze_github_repo(repo_input, github_token=None):
     
     github_token = os.environ.get("GITHUB_TOKEN")
     if not github_token:
+        print("GitHub token not found")
         return "❌ Error: GITHUB_TOKEN environment variable not set.", gr.update(visible=True), gr.update(visible=False)
 
     openrouter_api_key = os.environ.get("OPENROUTER_API_KEY")
     if not openrouter_api_key:
+        print("OpenRouter API key not found")
         return "❌ Error: OPENROUTER_API_KEY environment variable not set.", gr.update(visible=True), gr.update(visible=False)
 
     progress_md = ""
-    yield progress_md, gr.update(visible=True), gr.update(visible=False)  # Initial empty output
-
-    for emoji, message in PROGRESS_STEPS:
-        progress_md += f"{emoji} {message}  \n"
-        yield progress_md, gr.update(visible=True), gr.update(visible=False)
-        time.sleep(4)
-
+    
     try:
         owner, repo_name = get_repo_info(repo_input)
         repo_url = f"https://github.com/{owner}/{repo_name}"
+        print(f"Analyzing repository: {repo_url}")
         
         g = Github(github_token)
         github_repo = g.get_repo(f"{owner}/{repo_name}")
@@ -51,21 +82,25 @@ def analyze_github_repo(repo_input, github_token=None):
 
         with tempfile.TemporaryDirectory() as temp_dir:
             repo_path = clone_repo(owner, repo_name, temp_dir)
+            print(f"Cloned repository to: {repo_path}")
 
             progress_md += "🔬 Analyzing code structure...  \n"
             yield progress_md, gr.update(visible=True), gr.update(visible=False)
             code_analysis = analyze_code(repo_path)
             code_analysis['llm_analysis'] = llm_analyze_code(client, code_analysis)
+            print("Code analysis complete")
 
             progress_md += f"📊 Analyzing issues (max {max_issues})...  \n"
             yield progress_md, gr.update(visible=True), gr.update(visible=False)
             issues_data = analyze_issues(github_repo, max_issues)
             issues_analysis = llm_analyze_issues(client, issues_data, repo_url)
+            print("Issues analysis complete")
 
             progress_md += f"🔀 Analyzing pull requests (max {max_prs})...  \n"
             yield progress_md, gr.update(visible=True), gr.update(visible=False)
             prs_data = analyze_pull_requests(github_repo, max_prs)
             pr_analysis = llm_analyze_prs(client, prs_data, repo_url)
+            print("Pull requests analysis complete")
 
             progress_md += "🧠 Synthesizing findings...  \n"
             yield progress_md, gr.update(visible=True), gr.update(visible=False)
@@ -75,6 +110,7 @@ def analyze_github_repo(repo_input, github_token=None):
                 issues_analysis.get('summary', ''),
                 pr_analysis.get('summary', '')
             )
+            print("Findings synthesized")
 
             repo_info = {
                 "owner": owner,
@@ -84,11 +120,13 @@ def analyze_github_repo(repo_input, github_token=None):
             progress_md += "📝 Generating report...  \n"
             yield progress_md, gr.update(visible=True), gr.update(visible=False)
             report = generate_report(repo_info, code_analysis, issues_analysis, pr_analysis, final_analysis)
+            print("Report generated")
 
-        # Return the final Markdown report
+        print("Analysis complete, returning results")
         return progress_md + "✅ Analysis complete!", gr.update(visible=False), gr.update(visible=True, value=report)
     except Exception as e:
         error_message = f"❌ An error occurred: {str(e)}"
+        print(f"Error occurred: {error_message}")
         traceback.print_exc()
         return progress_md + error_message, gr.update(visible=True), gr.update(visible=False)