25:45

main.py → app.py

@@ -1,11 +1,5 @@
 import gradio as gr
-
-
-def generate_restaurant_name_and_items(cuisine: str) -> dict[str, str]:
-    return {
-        "restaurant_name": "Curry Delight",
-        "menu_items": "Butter Chicken, Naan, Paneer Tikka, Chole Bhature, Lassi, Gulab Jamun",
-    }
+from langchain_helper import generate_restaurant_name_and_items
 
 
 def update_outputs(cuisine: str) -> tuple[str, str]:
@@ -20,7 +14,7 @@ def update_outputs(cuisine: str) -> tuple[str, str]:
     return f"## {restaurant_name}", menu_items_formatted
 
 
-with gr.Blocks() as app:
+with gr.Blocks() as gradio_app:
     gr.Markdown("# Restaurant Name Generator")
     inp_cuisine = gr.Dropdown(
         ["Indian", "Italian", "Mexican", "Arabic"],
@@ -38,4 +32,4 @@ with gr.Blocks() as app:
 
 
 if __name__ == "__main__":
-    app.launch()
+    gradio_app.launch()

langchain_helper.py

@@ -0,0 +1,44 @@
+import os
+from keys import openapi_key
+from langchain.llms import OpenAI
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain, SequentialChain
+
+os.environ["OPENAI_API_KEY"] = openapi_key
+llm = OpenAI(temperature=0.7)
+
+prompt_template_name = PromptTemplate(
+    input_variables=["cuisine"],
+    template="I want to open a restaurant for {cuisine} food. Suggest a fancy name for this.",
+)
+
+name_chain = LLMChain(
+    llm=llm, prompt=prompt_template_name, output_key="restaurant_name"
+)
+
+prompt_template_items = PromptTemplate(
+    input_variables=["restaurant_name"],
+    template="Suggest some menu items for {restaurant_name}. Return it as a comma separated string.",
+)
+
+food_items_chain = LLMChain(
+    llm=llm, prompt=prompt_template_items, output_key="menu_items"
+)
+
+chain = SequentialChain(
+    chains=[name_chain, food_items_chain],
+    input_variables=["cuisine"],
+    output_variables=["restaurant_name", "menu_items"],
+)
+
+
+def generate_restaurant_name_and_items(cuisine: str) -> dict[str, str]:
+    return chain({"cuisine": cuisine})
+    # return {
+    #     "restaurant_name": "Curry Delight",
+    #     "menu_items": "Butter Chicken, Naan, Paneer Tikka, Chole Bhature, Lassi, Gulab Jamun",
+    # }
+
+
+if __name__ == "__main__":
+    print(generate_restaurant_name_and_items("Singaporean"))