Fix formatting

app.py

@@ -4,31 +4,35 @@ 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"]
+    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
+    return f"# {restaurant_name}", menu_items_formatted
 
 
 with gr.Blocks() as gradio_app:
-    gr.Markdown("# Restaurant Name Generator")
-    inp_cuisine = gr.Dropdown(
-        ["Indian", "Italian", "Mexican", "Arabic"],
-        label="Pick a cuisine",
-    )
-
-    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],
-    )
+    gr.Markdown("## Restaurant Name Generator")
+
+    with gr.Row():
+        with gr.Column() as left:
+            inp_cuisine = gr.Dropdown(
+                ["Indian", "Italian", "Mexican", "Arabic"],
+                label="Pick a cuisine",
+            )
+
+        with gr.Column(variant="panel") 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__":

langchain_helper.py

@@ -22,7 +22,7 @@ name_chain = LLMChain(
 
 prompt_template_items = PromptTemplate(
     input_variables=["restaurant_name"],
-    template="Suggest some menu items for {restaurant_name}. Return the menu items as a comma separated string with no additional preamble.",
+    template="Suggest some menu items for {restaurant_name}. Return the menu items as a single, comma separated string with no additional preamble.",
 )
 
 food_items_chain = LLMChain(
@@ -39,7 +39,10 @@ chain = SequentialChain(
 
 
 def generate_restaurant_name_and_items(cuisine: str) -> dict[str, str]:
-    return chain({"cuisine": cuisine})
+    result = chain({"cuisine": cuisine})
+    print(result)
+    return result
+
     # return {
     #     "restaurant_name": "Curry Delight",
     #     "menu_items": "Butter Chicken, Naan, Paneer Tikka, Chole Bhature, Lassi, Gulab Jamun",