feat: introduce gradio-based interface for courtroom trials

src/code_tribunal/ui/__init__.py

@@ -0,0 +1,3 @@
+"""CodeTribunal Gradio UI package."""
+
+from code_tribunal.ui.app import create_app, main

src/code_tribunal/ui/app.py

@@ -0,0 +1,68 @@
+"""Gradio application layout and launch."""
+
+import logging
+from pathlib import Path
+
+import gradio as gr
+
+from code_tribunal.config import TribunalConfig
+from code_tribunal.ui.exports import export_md, export_pdf
+from code_tribunal.ui.pipeline import handle_question, run_courtroom
+from code_tribunal.ui.styles import CSS
+
+logging.basicConfig(level=logging.DEBUG, format="[%(asctime)s][%(levelname)s] %(message)s", datefmt="%H:%M:%S")
+logging.getLogger("crewai").setLevel(logging.WARNING)
+
+
+def create_app() -> gr.Blocks:
+    logo = Path(__file__).resolve().parent.parent.parent.parent / "assets" / "logo.png"
+
+    with gr.Blocks(title="CodeTribunal") as app:
+        with gr.Column(visible=True) as hero:
+            if logo.exists():
+                gr.Image(value=str(logo), show_label=False, height=160, container=False, elem_classes=["hero-logo"])
+            gr.Markdown("# CodeTribunal\n### The AI Courtroom That Exposes Bad Freelance Code", elem_classes=["hero-title"])
+            gr.Markdown("Upload a .zip of code and watch a multi-agent forensic investigation unfold.", elem_classes=["hero-subtitle"])
+
+        with gr.Column(visible=True, elem_classes=["upload-area"]) as upload:
+            code_input = gr.File(label="Drop your .zip here", file_types=[".zip"], interactive=True)
+
+        with gr.Column(visible=False) as processing:
+            status = gr.Markdown("Initializing...", elem_classes=["status-phase"])
+            chatbot = gr.Chatbot(label="Courtroom Transcript", height=600, elem_classes=["chatbot"])
+            with gr.Row():
+                exp_md = gr.Button("📄 Export Markdown", visible=False)
+                exp_pdf = gr.Button("📑 Export PDF", visible=False)
+            exp_file = gr.File(label="Download", visible=False)
+            qa_input = gr.Textbox(label="Ask a follow-up", placeholder="e.g., 'Why was eval() critical?'", visible=False)
+            qa_btn = gr.Button("🕵️ Ask Expert Witness", variant="primary", visible=False)
+
+        state = gr.State(value={})
+
+        code_input.upload(
+            fn=run_courtroom, inputs=[code_input],
+            outputs=[hero, upload, processing, status, chatbot, exp_md, exp_pdf, exp_file, state],
+        ).then(
+            fn=lambda: (gr.update(visible=True), gr.update(visible=True)),
+            outputs=[qa_input, qa_btn],
+        )
+        exp_md.click(fn=export_md, inputs=[state], outputs=[exp_file])
+        exp_pdf.click(fn=export_pdf, inputs=[state], outputs=[exp_file])
+        qa_btn.click(fn=handle_question, inputs=[qa_input, chatbot, state], outputs=[chatbot, qa_input])
+        qa_input.submit(fn=handle_question, inputs=[qa_input, chatbot, state], outputs=[chatbot, qa_input])
+
+    return app
+
+
+def main() -> None:
+    app = create_app()
+    app.launch(
+        server_name=TribunalConfig().server_host,
+        server_port=TribunalConfig().server_port,
+        css=CSS,
+        theme=gr.themes.Base(primary_hue="amber", secondary_hue="slate", neutral_hue="slate"),
+    )
+
+
+if __name__ == "__main__":
+    main()

src/code_tribunal/ui/exports.py

@@ -0,0 +1,46 @@
+"""Report export: Markdown and PDF."""
+
+import tempfile
+import time
+from pathlib import Path
+
+from markdown_pdf import MarkdownPdf, Section
+
+from code_tribunal.ui.helpers import evidence_html
+
+
+def export_md(ctx: dict) -> str:
+    """Export trial context as a Markdown file. Returns the file path."""
+    md = [
+        "# CodeTribunal — Trial Report\n",
+        f"**Generated**: {time.strftime('%Y-%m-%d %H:%M:%S')}\n",
+        "---\n",
+    ]
+    for key in ["evidence", "verdict", "report"]:
+        value = ctx.get(key, "")
+        if value:
+            md.append(f"## {key.title()}\n\n{value}\n\n")
+    fp = tempfile.mktemp(suffix="_CodeTribunal_Report.md")
+    Path(fp).write_text("\n".join(md))
+    return fp
+
+
+def export_pdf(ctx: dict) -> str:
+    """Export trial context as a PDF file. Returns the file path."""
+    sections = []
+    for key in ["evidence", "verdict", "report"]:
+        value = ctx.get(key, "")
+        if value:
+            sections.append(f"## {key.title()}\n\n{value}")
+
+    full_md = (
+        f"# CodeTribunal - Trial Report\n\n"
+        f"*Generated: {time.strftime('%Y-%m-%d %H:%M:%S')}*\n\n---\n\n"
+        + "\n".join(sections)
+    )
+
+    pdf = MarkdownPdf()
+    pdf.add_section(Section(full_md))
+    fp = tempfile.mktemp(suffix="_CodeTribunal_Report.pdf")
+    pdf.save(fp)
+    return fp

src/code_tribunal/ui/helpers.py

@@ -0,0 +1,47 @@
+"""HTML rendering helpers for the Gradio UI."""
+
+from pathlib import Path
+
+from code_tribunal.ui.styles import SEVERITY_COLORS
+
+
+def escape_html(text: str) -> str:
+    return text.replace("&", "&amp;").replace("<", "&lt;").replace(">", "&gt;")
+
+
+def severity_badge(severity: str) -> str:
+    color = SEVERITY_COLORS.get(severity, "#6b7280")
+    return (
+        f'<span style="background:{color};color:white;padding:2px 8px;'
+        f'border-radius:4px;font-size:12px;font-weight:bold">{severity}</span>'
+    )
+
+
+def evidence_html(report) -> str:
+    """Render an EvidenceReport as an HTML table."""
+    lines = [
+        "<h3>Evidence Report</h3>",
+        f"<p>Files: <b>{report.file_count}</b> | Findings: <b>{len(report.findings)}</b></p>",
+    ]
+    for domain, findings in report.findings_by_domain.items():
+        lines.append(f"<h4>{domain.title()} ({len(findings)})</h4>")
+        lines.append('<table style="width:100%;border-collapse:collapse">')
+        lines.append(
+            '<tr style="border-bottom:1px solid #333">'
+            "<th style='text-align:left;padding:4px'>Sev</th>"
+            "<th style='text-align:left;padding:4px'>File</th>"
+            "<th style='text-align:left;padding:4px'>Line</th>"
+            "<th style='text-align:left;padding:4px'>Code</th></tr>"
+        )
+        for finding in findings:
+            lines.append(
+                f'<tr style="border-bottom:1px solid #222">'
+                f"<td style='padding:4px'>{severity_badge(finding.severity_hint)}</td>"
+                f"<td style='padding:4px;font-family:monospace;font-size:13px'>"
+                f"{Path(finding.file).name}</td>"
+                f"<td style='padding:4px;font-family:monospace'>{finding.line}</td>"
+                f"<td style='padding:4px;font-family:monospace;font-size:13px;color:#a0a0a0'>"
+                f"{escape_html(finding.code)}</td></tr>"
+            )
+        lines.append("</table>")
+    return "\n".join(lines)

src/code_tribunal/ui/pipeline.py

@@ -0,0 +1,145 @@
+"""Pipeline orchestration: runs the courtroom and streams UI updates."""
+
+import logging
+import tempfile
+
+import gradio as gr
+
+from code_tribunal.config import TribunalConfig
+from code_tribunal.courtroom import Courtroom
+from code_tribunal.evidence import safe_extract_zip
+from code_tribunal.pipeline import Phase
+from code_tribunal.ui.helpers import escape_html, evidence_html
+from code_tribunal.ui.styles import PHASE_LABELS
+
+log = logging.getLogger("code_tribunal")
+
+
+def run_courtroom(code_input, progress=gr.Progress()):
+    """Run the full pipeline, yielding updates to the UI."""
+    chat = []
+    ev_html = ""
+    verdict_text = ""
+    report_text = ""
+    config = TribunalConfig()
+
+    if code_input is None or not (hasattr(code_input, "name") and code_input.name.endswith(".zip")):
+        yield (
+            gr.update(visible=False), gr.update(visible=False), gr.update(visible=True),
+            "### Please upload a .zip file.", [],
+            gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), {},
+        )
+        return
+
+    yield (
+        gr.update(visible=False), gr.update(visible=False), gr.update(visible=True),
+        "### Extracting files...", [],
+        gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), {},
+    )
+
+    tmpdir = tempfile.mkdtemp()
+    try:
+        safe_extract_zip(code_input.name, tmpdir)
+    except ValueError as e:
+        yield (
+            gr.update(visible=False), gr.update(visible=False), gr.update(visible=True),
+            f"### Error: {escape_html(str(e))}", [],
+            gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), {},
+        )
+        return
+
+    if not config.is_configured:
+        yield (
+            gr.update(visible=False), gr.update(visible=False), gr.update(visible=True),
+            "### ZAI_API_KEY not set.", [],
+            gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), {},
+        )
+        return
+
+    courtroom = Courtroom(config)
+    courtroom.pipeline.create_run(code_input.name)
+    current_phase = Phase.IDLE
+
+    for event in courtroom.run(tmpdir):
+        if event.phase != current_phase:
+            current_phase = event.phase
+            log.debug("[UI] Phase -> %s", current_phase)
+
+        if event.data and "report" in event.data and event.data["report"] is not None:
+            robj = event.data["report"]
+            if hasattr(robj, "findings"):
+                ev_html = evidence_html(robj)
+                chat.append({
+                    "role": "user",
+                    "content": f"**Case Filed**: {robj.file_count} files, **{len(robj.findings)}** findings.",
+                })
+
+        if event.data and "reports" in event.data:
+            for domain, text in event.data["reports"].items():
+                icon = {"security": "🛡️", "quality": "📋", "architecture": "🏗️"}.get(domain, "📝")
+                chat.append({
+                    "role": "assistant",
+                    "content": f"**{icon} {domain.title()} Investigation**\n\n{text[:3000]}{'...' if len(text) > 3000 else ''}",
+                })
+
+        if event.data and "transcript" in event.data:
+            for section in event.data["transcript"].split("\n\n"):
+                for rnd in ["PROSECUTION", "DEFENSE", "REBUTTAL"]:
+                    if section.startswith(f"=== {rnd}"):
+                        content = section.replace(f"=== {rnd} ===", "").strip()
+                        icon = {"PROSECUTION": "⚖️", "DEFENSE": "🛡️", "REBUTTAL": "⚖️"}.get(rnd, "📝")
+                        chat.append({
+                            "role": "assistant",
+                            "content": f"**{icon} {rnd.title()}**\n\n{content[:3000]}{'...' if len(content) > 3000 else ''}",
+                        })
+                        break
+
+        if event.data and "verdict" in event.data:
+            verdict_text = event.data["verdict"]
+            chat.append({
+                "role": "assistant",
+                "content": f"**🔨 Judge's Verdict**\n\n{verdict_text[:3000]}{'...' if len(verdict_text) > 3000 else ''}",
+            })
+
+        if event.data and "report" in event.data and isinstance(event.data["report"], str):
+            report_text = event.data["report"]
+            chat.append({
+                "role": "assistant",
+                "content": f"**📄 Final Report**\n\n{report_text[:4000]}{'...' if len(report_text) > 4000 else ''}",
+            })
+
+        show_export = event.phase == Phase.COMPLETE
+        ctx = {}
+        if show_export:
+            log.debug("[UI] COMPLETE — showing export")
+            ctx = {"evidence": ev_html, "verdict": verdict_text, "report": report_text}
+
+        label = PHASE_LABELS.get(current_phase, current_phase.value)
+        if current_phase not in (Phase.COMPLETE, Phase.FAILED):
+            status = f'<h3 class="phase-active">{label}</h3>\n{event.status}'
+        else:
+            status = f"### {label}\n{event.status}"
+
+        log.debug("[UI] yield — chat msgs: %d, status: %s, export: %s", len(chat), label, show_export)
+
+        yield (
+            gr.update(visible=False), gr.update(visible=False), gr.update(visible=True),
+            status, chat,
+            gr.update(visible=show_export), gr.update(visible=show_export),
+            gr.update(visible=show_export), ctx,
+        )
+
+
+def handle_question(question: str, history: list, ctx: dict) -> tuple:
+    """Handle a follow-up question from the Q&A interface."""
+    if not question.strip():
+        return history, ""
+    config = TribunalConfig()
+    courtroom = Courtroom(config)
+    answer = courtroom.ask_question(question, ctx)
+    history.append({"role": "user", "content": question})
+    history.append({
+        "role": "assistant",
+        "content": f"**🕵️ Expert Witness**\n\n{answer[:3000]}{'...' if len(answer) > 3000 else ''}",
+    })
+    return history, ""

src/code_tribunal/ui/styles.py

@@ -0,0 +1,32 @@
+"""UI constants: CSS, severity colors, phase labels."""
+
+from code_tribunal.pipeline import Phase
+
+SEVERITY_COLORS = {"CRITICAL": "#dc2626", "HIGH": "#ea580c", "MEDIUM": "#ca8a04", "LOW": "#2563eb"}
+
+CSS = """
+.gradio-container{min-width:1280px!important;max-width:1310px!important;margin:0 auto!important}
+body{background:#0a0a14!important}
+.hero-logo{display:block!important;margin:0 auto 12px auto!important;border-radius:16px!important}
+.hero-title{text-align:center!important;color:#fbbf24!important;font-family:Georgia,serif!important;font-size:2.4em!important}
+.hero-subtitle{text-align:center!important;color:#94a3b8!important;font-size:1.1em!important}
+.upload-area .file-preview{min-height:220px!important;border:2px dashed #fbbf2440!important;border-radius:16px!important;background:#1a1a2e!important}
+.upload-area .file-preview:hover{border-color:#fbbf24!important}
+.chatbot{border:none!important;box-shadow:none!important}
+.contain{border:none!important;box-shadow:none!important;background:transparent!important}
+[data-testid="status-tracker"]{border:none!important;box-shadow:none!important}
+.bot.svelte-1nr59td.message{border:none!important;box-shadow:none!important;background:transparent!important}
+.phase-active{font-family:sans-serif;font-size:1.5rem;font-weight:bold;background:linear-gradient(to left,#333 20%,#888 40%,#eee 50%,#888 60%,#333 80%);background-size:200% auto;-webkit-background-clip:text;-webkit-text-fill-color:transparent;background-clip:text;color:transparent;animation:shine 3s linear infinite}
+@keyframes shine{to{background-position:200% center}}
+"""
+
+PHASE_LABELS = {
+    Phase.EVIDENCE: "Phase 1/7: Forensic Evidence",
+    Phase.GRAPH: "Phase 2/7: Code Graph",
+    Phase.INVESTIGATION: "Phase 3/7: Investigation",
+    Phase.TRIAL: "Phase 4/7: The Trial",
+    Phase.VERDICT: "Phase 5/7: Verdict",
+    Phase.REPORT: "Phase 6/7: Final Report",
+    Phase.COMPLETE: "Trial Complete",
+    Phase.FAILED: "Error",
+}