import gradio as gr
from multi_agent import multi_agent_cycle
title = "Thought CODER"
description = "Made a Multi-Agent CODER LLM using Qwen2.5-72B API which not only Converts Thought to code but tests them out as well \n\n CONVERT YOUR THOUGHT TO CODE"
article = "Created by [Eternal Bliassard](https://github.com/EternalBlissard)."

# Create the Gradio demo
demo = gr.Interface(fn=multi_agent_cycle, 
                    inputs=[gr.Textbox(label="Code Requirements")], 
                    outputs=[gr.Textbox(label="Code")],
                    additional_inputs = [
        gr.Textbox(label="Input TestCase"),
        gr.Textbox(label="Output TestCase"),
        gr.Slider(minimum=0.1, maximum=2.0, value=0.5, step=0.1, label="Temperature"),
        gr.Slider(minimum=128, maximum=4096, value=1024, step=1, label="Max New Tokens"),
        gr.Slider(
            minimum=0.1,
            maximum=1.0,
            value=0.95,
            step=0.05,
            label="Top-P",
        ),
        gr.Slider(minimum=1, maximum=5, value=3, step=1, label="Reasoner Iterations"),
        gr.Slider(minimum=3, maximum=11, value=5, step=1, label="Test Cases To be Generated"),
],
                    title=title,
                    description=description,
                    article=article)

# Launch the demo!
demo.launch(share=True)