Update app.py

app.py

@@ -1,20 +1,20 @@
 import os
+import gradio as gr
 from langchain.chains import LLMChain, SequentialChain
 from langchain.prompts import PromptTemplate
-from langchain.llms import OpenAI
+from langchain_community.llms import OpenAI  # <-- UPDATED IMPORT
 
-# Get OpenAI API key from environment variable (configured in Hugging Face Secrets)
+# Get OpenAI API key from environment variable
 openai_api_key = os.getenv("OPENAI_API_KEY")
 llm = OpenAI(openai_api_key=openai_api_key, temperature=0.9)
 
 # Define prompt templates
 prompt_template_name = PromptTemplate(
     input_variables=['cuisine'],
-    template="i want to open a restaurant for {cuisine} food. Suggest a fancy name for this"
+    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="restaraunt_name")
 
-llm = OpenAI(openai_api_key=openai_api_key, temperature=0.9)
 prompt_template_items = PromptTemplate(
     input_variables=['restaraunt_name', 'cuisine'],
     template="Suggest a comprehensive menu for {restaraunt_name} which is {cuisine} cuisine. For each item, provide a brief, one-sentence description. Do not use any numbering or bullet points. Only mention items names and descriptions, break down according to category. Format the output as 'Category:\nItem Name - Description\n...'"
@@ -27,7 +27,6 @@ chain = SequentialChain(
     output_variables=['restaraunt_name', 'menu_items']
 )
 
-# Dropdown options
 TOP_CUISINES = [
     "Italian", "Chinese", "Japanese", "Mexican", "Indian",
     "French", "Thai", "Mediterranean", "American", "Spanish"
@@ -114,7 +113,6 @@ def on_generate(selected, custom):
     cuisine = custom.strip() if custom and custom.strip() else selected
     return generate_restaurant(cuisine)
 
-# Build Gradio interface
 with gr.Blocks(title="Restaurant Business Generator", theme="soft", css="""
     @import url('https://fonts.googleapis.com/css2?family=Comic+Neue:wght@400;700&display=swap');
     body {