| import gradio as gr | |
| from langchain_helper import generate_restaurant_name_and_items | |
| def update_outputs(cuisine: str) -> tuple[str, str]: | |
| response = generate_restaurant_name_and_items(cuisine) | |
| restaurant_name = response["restaurant_name"].strip() | |
| menu_items = response["menu_items"].split(",") | |
| menu_items_formatted = "" | |
| for item in menu_items: | |
| menu_items_formatted += f"- {item.strip()}\n" | |
| return f"# {restaurant_name}", menu_items_formatted | |
| with gr.Blocks() as gradio_app: | |
| gr.Markdown("## Restaurant Name Generator") | |
| with gr.Row(): | |
| with gr.Column(scale=1) as left: | |
| inp_cuisine = gr.Dropdown( | |
| ["Arabic", "French", "Indian", "Italian", "Mexican"], | |
| label="Pick a cuisine", | |
| ) | |
| with gr.Column(variant="panel", scale=4) as right: | |
| out_restaurant_name = gr.Markdown() | |
| out_menu_items = gr.Markdown() | |
| inp_cuisine.change( | |
| fn=update_outputs, | |
| inputs=inp_cuisine, | |
| outputs=[out_restaurant_name, out_menu_items], | |
| ) | |
| if __name__ == "__main__": | |
| gradio_app.launch() | |