Update app.py

app.py

@@ -24,18 +24,18 @@ def analyze_github_repo(repo_input, github_token=None):
     
     github_token = os.environ.get("GITHUB_TOKEN")
     if not github_token:
-        return "❌ Error: GITHUB_TOKEN environment variable not set."
+        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:
-        return "❌ Error: OPENROUTER_API_KEY environment variable not set."
+        return "❌ Error: OPENROUTER_API_KEY environment variable not set.", gr.update(visible=True), gr.update(visible=False)
 
     progress_html = ""
-    yield progress_html  # Initial empty output
+    yield progress_html, gr.update(visible=True), gr.update(visible=False)  # Initial empty output
 
     for emoji, message in PROGRESS_STEPS:
         progress_html += f"{emoji} {message}<br>"
-        yield progress_html
+        yield progress_html, gr.update(visible=True), gr.update(visible=False)
         time.sleep(1)  # Reduced sleep time for faster feedback
 
     try:
@@ -54,22 +54,22 @@ def analyze_github_repo(repo_input, github_token=None):
             repo_path = clone_repo(owner, repo_name, temp_dir)
 
             progress_html += "🔬 Analyzing code structure...<br>"
-            yield progress_html
+            yield progress_html, gr.update(visible=True), gr.update(visible=False)
             code_analysis = analyze_code(repo_path)
             code_analysis['llm_analysis'] = llm_analyze_code(client, code_analysis)
 
             progress_html += f"📊 Analyzing issues (max {max_issues})...<br>"
-            yield progress_html
+            yield progress_html, 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)
 
             progress_html += f"🔀 Analyzing pull requests (max {max_prs})...<br>"
-            yield progress_html
+            yield progress_html, 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)
 
             progress_html += "🧠 Synthesizing findings...<br>"
-            yield progress_html
+            yield progress_html, gr.update(visible=True), gr.update(visible=False)
             final_analysis = llm_synthesize_findings(
                 client, 
                 code_analysis.get('llm_analysis', ''),
@@ -83,18 +83,18 @@ def analyze_github_repo(repo_input, github_token=None):
             }
 
             progress_html += "📝 Generating report...<br>"
-            yield progress_html
+            yield progress_html, gr.update(visible=True), gr.update(visible=False)
             report = generate_report(repo_info, code_analysis, issues_analysis, pr_analysis, final_analysis)
 
         # Convert markdown to HTML
         html_report = markdown.markdown(report)
         
         # Return the final HTML report
-        return f"{progress_html}<br><h2>Analysis Report</h2>{html_report}"
+        return progress_html + "✅ Analysis complete!", gr.update(visible=False), gr.update(visible=True, value=html_report)
     except Exception as e:
         error_message = f"❌ An error occurred: {str(e)}"
         traceback.print_exc()
-        return f"{progress_html}<br>{error_message}"
+        return progress_html + error_message, gr.update(visible=True), gr.update(visible=False)
 
 # Define the Gradio interface
 with gr.Blocks() as app:
@@ -107,12 +107,13 @@ with gr.Blocks() as app:
     
     analyze_button = gr.Button("Analyze Repository")
     
-    output = gr.HTML(label="Analysis Output")
+    progress_output = gr.HTML(label="Progress")
+    report_output = gr.HTML(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