import gradio as gr
from pipeline import run_with_chain

def ask_dailywellness(query: str) -> str:
    return run_with_chain(query)

interface = gr.Interface(
    fn=ask_dailywellness,
    inputs=gr.Textbox(lines=2, label="Ask DailyWellnessAI"),
    outputs=gr.Textbox(label="DailyWellnessAI Answer"),
    title="DailyWellnessAI",
    description="Ask about wellness or DailyWellnessAI brand. Out-of-scope"
)

if __name__ == "__main__":
    interface.launch(server_name="0.0.0.0", server_port=7860)