from gradio import Interface, Dropdown, CheckboxGroup

from data import get_cities
from llm import get_response

cities = get_cities()


def hotel_recommender(city, preferences) -> dict:
    return get_response(city, preferences)


demo = Interface(
    fn=hotel_recommender,
    inputs=[
        Dropdown(cities, label="City"),
        CheckboxGroup(["Kid friendly", "other option"], label="Preferences"),
            ],
    outputs="text"
)

if __name__ == "__main__":
    demo.launch()