Update app.py

app.py

@@ -1,59 +1,43 @@
-import gradio as gr
 import os
+import gradio as gr
 from langchain.chains import LLMChain, SequentialChain
-from langchain.prompts import PromptTemplate
+from langchain.prompts import PromptTemplate 
 from langchain.llms import OpenAI
 
+os.environ["OPENAI_API_KEY"]= os.getenv["OPENAI_API_KEY"]
+llm = OpenAI(temperature= 0.5)
 
-llm = OpenAI(temperature=0.9, openai_api_key=os.getenv("openai"))
+prompt_name = PromptTemplate(
+    input_variables= ['cuisine'],
+    template = "I want to open a chic restuarant for {cuisine} food. Suggest an eye-catching fancy name for this"
+)
 
-# Set OpenAI API key
-openai = os.getenv("openai")
-if not openai:
-    raise ValueError("Please set the in your Hugging space")
+name_chain = LLMChain(llm=llm, prompt =prompt_name, output_key = "restaurant_name" )
 
-os.environ["openai"] = openai
+promp_items= PromptTemplate(
+    input_variables=['restaurant_name'],
+    template = ["Create a menu for {restaurant_name}. Divide it into sections like starters, brekfast, lunch and dinner finishing it off with desserts. "
+    "a small description underneath each dish. price across. The format of a five michelin star restaurant."]
+)
 
-# Define the LLM
-llm = OpenAI(temperature=0.9)
+menu_chain = LLMChain(llm=llm, prompt= promp_items, output_key = "men_items")
 
-# Chain 1: Fancy restaurant name
-prompt_template_name = PromptTemplate(
-    input_variables=["cuisine"],
-    template="I want to open a fancy restaurant for {cuisine} cuisine. Suggest a creative and appealing name for this restaurant."
+main_chain = SequentialChain(
+    chains= [name_chain, menu_chain],
+    input_variables = ['cuisine'],
+    output_variables = ['restaurant_name', 'men_items']
 )
-name_chain = LLMChain(llm=llm, prompt=prompt_template_name, output_key="restaurant_name")
-
-# Chain 2: Menu items with sections
-prompt_template_items = PromptTemplate(
-    input_variables=["restaurant_name"],
-    template=(
-        "Create a detailed menu for a restaurant named '{restaurant_name}'. "
-        "The menu should include unique and appealing dishes under the following sections:\n"
-        "1. Breakfast\n"
-        "2. Lunch\n"
-        "3. Dinner\n"
-        "Each section should have at least 3 items with short descriptions. The menu should reflect the theme and cuisine of the restaurant."
-    )
-)
-food_items_chain = LLMChain(llm=llm, prompt=prompt_template_items, output_key="menu_items")
 
-# Combine chains
-chain = SequentialChain(
-    chains=[name_chain, food_items_chain],
-    input_variables=["cuisine"],
-    output_variables=["restaurant_name", "menu_items"]
-)
+def hotel_function(cuisine):
+    result = main_chain({'cuisine': cuisine} )
+    return result['restaurant_name'], result['menu_items']
+
+
+iface = gr.Interface(
+    fn = hotel_function,
+    inputs = gr.Textbox(label = "Enter cuisine:") ,
+    output = [gr.TextBox(label= "Restaurant_name"), gr.TextBox(label = "Menu Items")],
+    title = "Restaurant Menu and Items",
+    description="Powered by Langchain and OpenAI")
 
-# Gradio interface
-def generate_restaurant_idea(cuisine):
-    result = chain({'cuisine': cuisine})
-    return f"🍽️ **Restaurant Name:** {result['restaurant_name']}\n\n📜 **Menu:**\n{result['menu_items']}"
-
-gr.Interface(
-    fn=generate_restaurant_idea,
-    inputs=gr.Textbox(lines=1, placeholder="Enter a cuisine type, e.g., Italian, Japanese, etc."),
-    outputs=gr.Markdown(),
-    title="🍴 Restaurant Business Idea Generator",
-    description="Enter a cuisine type and get a fancy restaurant name along with a structured menu for breakfast, lunch, and dinner."
-).launch()
+iface.launch()