import gradio as gr
import traceback
from llm_router import LLMRouter, AllProvidersFailed

router = LLMRouter()

def execute_task(username, task):
    if not task.strip():
        return "", "No task provided.", ""

    try:
        code = router.generate_code(task)

        local_scope = {}
        exec(code, {}, local_scope)

        result = local_scope.get("result", "Execution complete.")

        return code, str(result), "Task executed successfully."

    except AllProvidersFailed as e:
        return "", f"All LLM providers failed: {str(e)}", "Failover exhausted."

    except Exception as e:
        tb = traceback.format_exc()
        return "", f"Execution error: {str(e)}", tb


with gr.Blocks() as app:
    gr.Markdown("## 🤖 Codex Reflex AGI vΩΞ-R (Resilient Mode)")

    username = gr.Textbox(label="👤 Username")
    task = gr.Textbox(label="🎯 Task")

    execute_btn = gr.Button("⚡ Execute")

    generated_code = gr.Code(label="🧠 Generated Code", language="python")
    result = gr.Textbox(label="📦 Result")
    log = gr.Textbox(label="🧩 Task Log")

    execute_btn.click(
        execute_task,
        inputs=[username, task],
        outputs=[generated_code, result, log]
    )

app.launch()