Update SwarmAudit final demo

README.md

@@ -1,10 +1,14 @@
  ---
 title: SwarmAudit
+emoji: 🚀
+colorFrom: blue
+colorTo: cyan
 sdk: gradio
 sdk_version: 6.14.0
 app_file: app.py
 pinned: false
 license: mit
+short_description: Multi-agent production-readiness scanner for AI-generated code
 ---
 
 # SwarmAudit
@@ -301,4 +305,3 @@ For the hackathon submission, highlight:
 - Hugging Face Space deployment
 - practical business value: production readiness for AI-generated code
 - originality: combining security, operations, dependency, and CUDA-to-ROCm portability checks in one audit workflow
-

app/ui/gradio_app.py

@@ -570,10 +570,10 @@ button.primary:hover,
     color: var(--sa-text) !important;
 }
 
-.severity-filter-radio label:has(input[value^="Critical"]) span { color: var(--sa-red) !important; }
-.severity-filter-radio label:has(input[value^="High"]) span { color: var(--sa-orange) !important; }
-.severity-filter-radio label:has(input[value^="Medium"]) span { color: var(--sa-yellow) !important; }
-.severity-filter-radio label:has(input[value^="Low"]) span { color: var(--sa-blue) !important; }
+.severity-filter-radio label:has(input[value="critical"]) span { color: var(--sa-red) !important; }
+.severity-filter-radio label:has(input[value="high"]) span { color: var(--sa-orange) !important; }
+.severity-filter-radio label:has(input[value="medium"]) span { color: var(--sa-yellow) !important; }
+.severity-filter-radio label:has(input[value="low"]) span { color: var(--sa-blue) !important; }
 
 .severity-filter-radio span {
     font: 700 10px/14px JetBrains Mono, monospace !important;
@@ -1053,15 +1053,15 @@ def render_report_toolbar(report: AuditReport | None) -> str:
     """
 
 
-def build_severity_filter_choices(report: AuditReport | None) -> list[str]:
+def build_severity_filter_choices(report: AuditReport | None) -> list[tuple[str, str]]:
     if report is None:
-        return ["All 0"]
+        return [("All 0", "all")]
 
     displayed_counts = {severity: 0 for severity in Severity}
     for finding in report.findings:
         displayed_counts[finding.severity] += 1
 
-    choices = [f"All {len(report.findings)}"]
+    choices: list[tuple[str, str]] = [(f"All {len(report.findings)}", "all")]
     for severity, label in [
         (Severity.critical, "Critical"),
         (Severity.high, "High"),
@@ -1070,7 +1070,7 @@ def build_severity_filter_choices(report: AuditReport | None) -> list[str]:
     ]:
         count = displayed_counts.get(severity, 0)
         if count > 0:
-            choices.append(f"{label} {count}")
+            choices.append((f"{label} {count}", severity.value.lower()))
     return choices
 
 
@@ -1110,7 +1110,7 @@ async def analyze_repo(repo_url: str):
             render_agent_swarm(),
             render_empty_summary(),
             render_report_toolbar(None),
-            gr.update(choices=["All 0"], value="All 0"),
+            gr.update(choices=[("All 0", "all")], value="all"),
             format_report_overview_html(None),
             gr.update(choices=[], value=None),
             format_empty_finding_detail_html(),
@@ -1124,7 +1124,7 @@ async def analyze_repo(repo_url: str):
     agent_html = render_agent_swarm(progress)
     summary_html = render_empty_summary()
     report_toolbar_html = render_report_toolbar(None)
-    severity_filter_update = gr.update(choices=["All 0"], value="All 0")
+    severity_filter_update = gr.update(choices=[("All 0", "all")], value="all")
     report_overview_html = format_report_overview_html(None)
     finding_choice_update = gr.update(choices=[], value=None)
     finding_detail_html = format_empty_finding_detail_html()
@@ -1136,7 +1136,7 @@ async def analyze_repo(repo_url: str):
             if isinstance(event, AuditReport):
                 report_state = event
                 filter_choices = build_severity_filter_choices(event)
-                selected_filter = filter_choices[0]
+                selected_filter = "all"
                 severity_filter_update = gr.update(choices=filter_choices, value=selected_filter)
                 finding_choices = build_finding_choices(event, selected_filter)
                 finding_choice_update = gr.update(
@@ -1171,7 +1171,7 @@ async def analyze_repo(repo_url: str):
             render_agent_swarm(progress),
             render_empty_summary(),
             render_report_toolbar(None),
-            gr.update(choices=["All 0"], value="All 0"),
+            gr.update(choices=[("All 0", "all")], value="all"),
             format_report_overview_html(None),
             gr.update(choices=[], value=None),
             format_empty_finding_detail_html(),
@@ -1374,8 +1374,8 @@ def build_app() -> gr.Blocks:
                         with gr.Row(elem_classes=["report-header-row"]):
                             report_toolbar = gr.HTML(render_report_toolbar(None), scale=1)
                             severity_filter = gr.Radio(
-                                choices=["All 0"],
-                                value="All 0",
+                                choices=[("All 0", "all")],
+                                value="all",
                                 interactive=True,
                                 show_label=False,
                                 scale=0,

tests/test_gradio_app.py

@@ -80,7 +80,41 @@ def test_render_report_summary_uses_report_counts():
         repo_url="https://github.com/example/project",
         scanned_file_count=4,
         skipped_file_count=1,
-        findings=[],
+        findings=[
+            Finding(
+                title="Critical issue",
+                severity=Severity.critical,
+                file_path="app.py",
+                line_start=1,
+                line_end=1,
+                description="Critical.",
+                why_it_matters="Important.",
+                suggested_fix="Fix.",
+                agent_source="Security Agent",
+            ),
+            Finding(
+                title="High issue one",
+                severity=Severity.high,
+                file_path="app.py",
+                line_start=2,
+                line_end=2,
+                description="High.",
+                why_it_matters="Important.",
+                suggested_fix="Fix.",
+                agent_source="Security Agent",
+            ),
+            Finding(
+                title="High issue two",
+                severity=Severity.high,
+                file_path="app.py",
+                line_start=3,
+                line_end=3,
+                description="High.",
+                why_it_matters="Important.",
+                suggested_fix="Fix.",
+                agent_source="Security Agent",
+            ),
+        ],
         severity_summary={
             Severity.critical: 1,
             Severity.high: 2,
@@ -133,7 +167,41 @@ def test_build_severity_filter_choices_uses_actual_counts():
         repo_url="https://github.com/example/project",
         scanned_file_count=4,
         skipped_file_count=1,
-        findings=[],
+        findings=[
+            Finding(
+                title="Critical issue",
+                severity=Severity.critical,
+                file_path="app.py",
+                line_start=1,
+                line_end=1,
+                description="Critical.",
+                why_it_matters="Important.",
+                suggested_fix="Fix.",
+                agent_source="Security Agent",
+            ),
+            Finding(
+                title="High issue one",
+                severity=Severity.high,
+                file_path="app.py",
+                line_start=2,
+                line_end=2,
+                description="High.",
+                why_it_matters="Important.",
+                suggested_fix="Fix.",
+                agent_source="Security Agent",
+            ),
+            Finding(
+                title="High issue two",
+                severity=Severity.high,
+                file_path="app.py",
+                line_start=3,
+                line_end=3,
+                description="High.",
+                why_it_matters="Important.",
+                suggested_fix="Fix.",
+                agent_source="Security Agent",
+            ),
+        ],
         severity_summary={
             Severity.critical: 1,
             Severity.high: 2,
@@ -144,7 +212,11 @@ def test_build_severity_filter_choices_uses_actual_counts():
         agents_run=["Synthesizer Agent"],
     )
 
-    assert build_severity_filter_choices(report) == ["All 3", "Critical 1", "High 2"]
+    assert build_severity_filter_choices(report) == [
+        ("All 3", "all"),
+        ("Critical 1", "critical"),
+        ("High 2", "high"),
+    ]
 
 
 def make_report_with_findings() -> AuditReport:
@@ -205,7 +277,7 @@ def test_filter_findings_returns_only_selected_severity():
     report = make_report_with_findings()
     report.findings.append(high)
 
-    update, html = filter_findings("High 1", report)
+    update, html = filter_findings("high", report)
 
     assert update["choices"] == ["HIGH  High risk\napp.py:20  |  Security Agent"]
     assert "High risk" in html
@@ -309,7 +381,7 @@ async def test_analyze_repo_empty_input_clears_report_exports():
     assert "Agent swarm" in result[1]
     assert "Files scanned" in result[2]
     assert "Audit report" in result[3]
-    assert result[4]["choices"] == ["All 0"]
+    assert result[4]["choices"] == [("All 0", "all")]
     assert "Security Score" in result[5]
     assert result[6]["choices"] == []
     assert result[6]["value"] is None
@@ -334,7 +406,7 @@ async def test_analyze_repo_failure_clears_report_exports(monkeypatch):
     assert "Agent swarm" in updates[-1][1]
     assert "Files scanned" in updates[-1][2]
     assert "Audit report" in updates[-1][3]
-    assert updates[-1][4]["choices"] == ["All 0"]
+    assert updates[-1][4]["choices"] == [("All 0", "all")]
     assert "Security Score" in updates[-1][5]
     assert updates[-1][6]["choices"] == []
     assert updates[-1][6]["value"] is None