| import os |
|
|
| import gradio as gr |
|
|
| from app.agents.graph import AuditGraph |
| from app.schemas import AuditReport |
| from app.services.report_formatter import format_report_markdown |
|
|
|
|
| EXAMPLE_REPOS = { |
| "Requests": "https://github.com/psf/requests", |
| "ItsDangerous": "https://github.com/pallets/itsdangerous", |
| "Flask": "https://github.com/pallets/flask", |
| } |
|
|
|
|
| async def analyze_repo(repo_url: str): |
| if not repo_url.strip(): |
| yield "Paste a public GitHub repository URL to start.", "" |
| return |
|
|
| progress: list[str] = [] |
| report_markdown = "" |
| try: |
| async for event in AuditGraph().run_with_progress(repo_url.strip()): |
| if isinstance(event, AuditReport): |
| report_markdown = format_report_markdown(event) |
| else: |
| progress.append(event) |
| yield "\n".join(progress), report_markdown |
| except Exception as exc: |
| progress.append(f"Audit failed: {exc}") |
| yield "\n".join(progress), report_markdown |
|
|
|
|
| def choose_example(example_name: str) -> str: |
| return EXAMPLE_REPOS.get(example_name, "") |
|
|
|
|
| def build_app() -> gr.Blocks: |
| with gr.Blocks(title="SwarmAudit") as demo: |
| gr.Markdown( |
| "# SwarmAudit\n" |
| "Paste a public GitHub URL and get a structured multi-agent audit report." |
| ) |
|
|
| with gr.Row(): |
| repo_url = gr.Textbox( |
| label="GitHub Repository URL", |
| placeholder="https://github.com/owner/repo", |
| scale=4, |
| ) |
| analyze = gr.Button("Analyze", variant="primary", scale=1) |
|
|
| example = gr.Dropdown( |
| label="Example repos", |
| choices=list(EXAMPLE_REPOS.keys()), |
| value=None, |
| interactive=True, |
| ) |
| example.change(choose_example, inputs=example, outputs=repo_url) |
|
|
| with gr.Row(): |
| progress_output = gr.Textbox( |
| label="Agent Progress", |
| lines=10, |
| interactive=False, |
| ) |
| report_output = gr.Markdown(label="Audit Report") |
|
|
| analyze.click(analyze_repo, inputs=repo_url, outputs=[progress_output, report_output]) |
| return demo |
|
|
|
|
| def launch_app() -> None: |
| server_name = os.getenv("GRADIO_SERVER_NAME", "0.0.0.0") |
| server_port = int(os.getenv("PORT", os.getenv("GRADIO_SERVER_PORT", "7860"))) |
| build_app().queue().launch(server_name=server_name, server_port=server_port) |
|
|
|
|
| if __name__ == "__main__": |
| launch_app() |
|
|
