Update app.py

app.py

@@ -52,25 +52,27 @@ def generate_report(repo_info, code_analysis, issues_analysis, pr_analysis, fina
 """
 
 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:
-        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)
+    def update_output(message):
+        print(message)  # Console log
+        return message  # Update Gradio interface
 
-    progress_md = ""
-    
     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}"
-        print(f"Analyzing repository: {repo_url}")
+        yield update_output(f"📊 Analyzing repository: {repo_url}")
         
         g = Github(github_token)
         github_repo = g.get_repo(f"{owner}/{repo_name}")
@@ -82,53 +84,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)
-            print(f"Cloned repository to: {repo_path}")
+            yield update_output(f"📁 Cloned repository to temporary directory")
 
-            progress_md += "🔬 Analyzing code structure...  \n"
-            yield progress_md, gr.update(visible=True), gr.update(visible=False)
+            yield update_output("🔬 Analyzing code structure...")
             code_analysis = analyze_code(repo_path)
             code_analysis['llm_analysis'] = llm_analyze_code(client, code_analysis)
-            print("Code analysis complete")
+            yield update_output("✅ Code analysis complete")
 
-            progress_md += f"📊 Analyzing issues (max {max_issues})...  \n"
-            yield progress_md, gr.update(visible=True), gr.update(visible=False)
+            yield update_output(f"📊 Analyzing issues (max {max_issues})...")
             issues_data = analyze_issues(github_repo, max_issues)
             issues_analysis = llm_analyze_issues(client, issues_data, repo_url)
-            print("Issues analysis complete")
+            yield update_output("✅ Issues analysis complete")
 
-            progress_md += f"🔀 Analyzing pull requests (max {max_prs})...  \n"
-            yield progress_md, gr.update(visible=True), gr.update(visible=False)
+            yield update_output(f"🔀 Analyzing pull requests (max {max_prs})...")
             prs_data = analyze_pull_requests(github_repo, max_prs)
             pr_analysis = llm_analyze_prs(client, prs_data, repo_url)
-            print("Pull requests analysis complete")
+            yield update_output("✅ Pull requests analysis complete")
 
-            progress_md += "🧠 Synthesizing findings...  \n"
-            yield progress_md, gr.update(visible=True), gr.update(visible=False)
+            yield update_output("🧠 Synthesizing findings...")
             final_analysis = llm_synthesize_findings(
                 client, 
                 code_analysis.get('llm_analysis', ''),
                 issues_analysis.get('summary', ''),
                 pr_analysis.get('summary', '')
             )
-            print("Findings synthesized")
+            yield update_output("✅ Findings synthesized")
 
             repo_info = {
                 "owner": owner,
                 "repo_name": repo_name,
             }
 
-            progress_md += "📝 Generating report...  \n"
-            yield progress_md, gr.update(visible=True), gr.update(visible=False)
+            yield update_output("📝 Generating report...")
             report = generate_report(repo_info, code_analysis, issues_analysis, pr_analysis, final_analysis)
-            print("Report generated")
+            yield update_output("✅ Report generated")
 
-        print("Analysis complete, returning results")
-        return progress_md + "✅ Analysis complete!", gr.update(visible=False), gr.update(visible=True, value=report)
+        yield update_output("🎉 Analysis complete! Here's the report:\n\n" + 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)
+        yield update_output(error_message)
 
 # Define the Gradio interface
 with gr.Blocks() as app:
@@ -141,13 +136,12 @@ with gr.Blocks() as app:
     
     analyze_button = gr.Button("Analyze Repository")
     
-    progress_output = gr.Markdown(label="Progress")
-    report_output = gr.Markdown(label="Analysis Report", visible=False)
+    output = gr.Markdown(label="Analysis Output")
     
     analyze_button.click(
         analyze_github_repo,
         inputs=[repo_input, github_token],
-        outputs=[progress_output, progress_output, report_output],
+        outputs=output,
     )
 
 # Launch the app