import gradio as gr
from model import rag

demo = gr.Interface(
    fn = rag,
    inputs = gr.Textbox(lines=5, label="Enter your medical question here"),
    outputs = gr.Textbox(lines=10, label="Generated Answer"),
    title = "Medical QA",
    description = "Type your query and get a model generated response.",
    theme = gr.themes.Glass()
)

demo.launch()