import os, sys, json, tempfile
from pathlib import Path
sys.path.insert(0, str(Path(__file__).parent))
import multi_agent as ma

import gradio as gr

MODES = ["binary", "web", "apk", "ipa"]

def run_analysis(file, url, mode):
    target = url.strip() if (mode == "web" and url.strip()) else (file.name if file else None)
    if not target:
        return "Please upload a file or enter a URL.", None, None, ""
    try:
        orch = ma.Orchestrator(mode=mode, rebuild=False)
        fr = orch.run(target)
    except Exception as e:
        import traceback; return f"Error: {e}\n{traceback.format_exc()}", None, None, ""
    tmp = Path(tempfile.mkdtemp())
    (tmp / "report.md").write_text(fr.report)
    (tmp / "report.json").write_text(json.dumps(orch.to_dict(fr), indent=2, default=str))
    agents = ", ".join(fr.agents_run)
    return fr.report, str(tmp / "report.md"), str(tmp / "report.json"), f"Agents run: {agents}"

def rebuild(file, url, mode):
    target = url.strip() if (mode == "web" and url.strip()) else (file.name if file else None)
    if not target:
        return None, "No target."
    try:
        orch = ma.Orchestrator(mode=mode, rebuild=True)
        fr = orch.run(target)
    except Exception as e:
        import traceback; return None, f"Error: {e}\n{traceback.format_exc()}"
    return fr.rebuild_path, f"Rebuild scaffold: {fr.rebuild_path}"

with gr.Blocks(title="Multi-Agent Reverse Engineering") as demo:
    gr.Markdown("# Multi-Agent Reverse Engineering System\nSpecialist agents (Binary, Web, Mobile, Firmware, Malware, Reconstructor, Reporter) collaborate.")
    with gr.Row():
        mode = gr.Radio(MODES, value="binary", label="Mode")
    with gr.Row():
        file = gr.File(label="Upload file (binary/apk/ipa)")
        url = gr.Textbox(label="Or URL (web mode)", placeholder="https://example.com")
    with gr.Row():
        analyze_btn = gr.Button("Run Analysis", variant="primary")
        rebuild_btn = gr.Button("Rebuild from scratch")
    agents_info = gr.Textbox(label="Agents", lines=1)
    report = gr.Textbox(label="Report", lines=30, max_lines=60)
    with gr.Row():
        md_file = gr.File(label="report.md")
        json_file = gr.File(label="report.json")
    rebuild_file = gr.File(label="rebuilt.zip")
    rebuild_status = gr.Textbox(label="Rebuild status")
    analyze_btn.click(run_analysis, [file, url, mode], [report, md_file, json_file, agents_info])
    rebuild_btn.click(rebuild, [file, url, mode], [rebuild_file, rebuild_status])

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False)