Update app.py

app.py

@@ -3,76 +3,43 @@ 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
 
-# Mock functions for testing - replace these with your actual functions
-def get_repo_info(repo_input):
-    return repo_input.split('/')[-2:]
+# 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..."),
+]
 
-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']}
+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 "❌ Error: GITHUB_TOKEN environment variable not set.", gr.update(visible=True), gr.update(visible=False)
 
-## Pull Requests Analysis
-{pr_analysis['summary']}
+    openrouter_api_key = os.environ.get("OPENROUTER_API_KEY")
+    if not openrouter_api_key:
+        return "❌ Error: OPENROUTER_API_KEY environment variable not set.", gr.update(visible=True), gr.update(visible=False)
 
-## Final Analysis
-{final_analysis}
-"""
+    progress_md = ""
+    yield progress_md, gr.update(visible=True), gr.update(visible=False)  # Initial empty output
 
-def analyze_github_repo(repo_input, github_token=None):
-    def update_output(message):
-        print(message)  # Console log
-        return message  # Update Gradio interface
+    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:
-        if github_token:
-            os.environ["GITHUB_TOKEN"] = github_token
-        
-        github_token = os.environ.get("GITHUB_TOKEN")
-        if not github_token:
-            return update_output("❌ Error: GITHUB_TOKEN environment variable not set.")
-
-        openrouter_api_key = os.environ.get("OPENROUTER_API_KEY")
-        if not openrouter_api_key:
-            return update_output("❌ Error: OPENROUTER_API_KEY environment variable not set.")
-
-        yield update_output("🚀 Starting analysis...")
-        
         owner, repo_name = get_repo_info(repo_input)
         repo_url = f"https://github.com/{owner}/{repo_name}"
-        yield update_output(f"📊 Analyzing repository: {repo_url}")
         
         g = Github(github_token)
         github_repo = g.get_repo(f"{owner}/{repo_name}")
@@ -84,46 +51,46 @@ def analyze_github_repo(repo_input, github_token=None):
 
         with tempfile.TemporaryDirectory() as temp_dir:
             repo_path = clone_repo(owner, repo_name, temp_dir)
-            yield update_output(f"📁 Cloned repository to temporary directory")
 
-            yield update_output("🔬 Analyzing code structure...")
+            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)
-            yield update_output("✅ Code analysis complete")
 
-            yield update_output(f"📊 Analyzing issues (max {max_issues})...")
+            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)
-            yield update_output("✅ Issues analysis complete")
 
-            yield update_output(f"🔀 Analyzing pull requests (max {max_prs})...")
+            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)
-            yield update_output("✅ Pull requests analysis complete")
 
-            yield update_output("🧠 Synthesizing findings...")
+            progress_md += "🧠 Synthesizing findings...  \n"
+            yield progress_md, gr.update(visible=True), gr.update(visible=False)
             final_analysis = llm_synthesize_findings(
                 client, 
                 code_analysis.get('llm_analysis', ''),
                 issues_analysis.get('summary', ''),
                 pr_analysis.get('summary', '')
             )
-            yield update_output("✅ Findings synthesized")
 
             repo_info = {
                 "owner": owner,
                 "repo_name": repo_name,
             }
 
-            yield update_output("📝 Generating report...")
+            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)
-            yield update_output("✅ Report generated")
 
-        yield update_output("🎉 Analysis complete! Here's the report:\n\n" + report)
+        # Return the final Markdown report
+        yield 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)}"
         traceback.print_exc()
-        yield update_output(error_message)
+        yield progress_md + error_message, gr.update(visible=True), gr.update(visible=False)
 
 # Define the Gradio interface
 with gr.Blocks() as app:
@@ -136,12 +103,13 @@ with gr.Blocks() as app:
     
     analyze_button = gr.Button("Analyze Repository")
     
-    output = gr.Markdown(label="Analysis Output")
+    progress_output = gr.Markdown(label="Progress")
+    report_output = gr.Markdown(label="Analysis Report", visible=False)
     
     analyze_button.click(
         analyze_github_repo,
         inputs=[repo_input, github_token],
-        outputs=output,
+        outputs=[progress_output, progress_output, report_output],
     )
 
 # Launch the app