36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
import gradio as gr
|
|
from test_generator import generate_tests
|
|
|
|
def create_interface():
|
|
with gr.Blocks(title="Unit Test Generator") as ui:
|
|
gr.Markdown("# Unit Test Generator")
|
|
gr.Markdown("Paste your Python code and get AI-generated unit tests")
|
|
|
|
with gr.Row():
|
|
with gr.Column(scale=1):
|
|
code_input = gr.Code(
|
|
label="Your Code",
|
|
language="python",
|
|
lines=15
|
|
)
|
|
generate_btn = gr.Button("Generate Tests", variant="primary")
|
|
|
|
with gr.Column(scale=1):
|
|
tests_output = gr.Textbox(
|
|
label="Generated Tests",
|
|
lines=15,
|
|
interactive=False
|
|
)
|
|
|
|
generate_btn.click(
|
|
fn=generate_tests,
|
|
inputs=[code_input],
|
|
outputs=[tests_output]
|
|
)
|
|
|
|
return ui
|
|
|
|
def launch():
|
|
ui = create_interface()
|
|
ui.launch(server_name="localhost", server_port=7860)
|