import gradio as gr
import test2 as test2
import test3 as test3

theme = gr.themes.Soft(
    font_mono=[gr.themes.GoogleFont('Fira Code'), 'ui-monospace', 'Consolas', 'monospace'],
).set(
    embed_radius='*radius_md'
)

with gr.Blocks(theme=theme, title="AI Research Assistant") as demo:
    output = gr.Textbox(label="Output")
    button = gr.Button("Start")
    button.click(fn=test2.generator_, outputs=output)
    
demo.queue()
demo.launch()