Update app.py

app.py

@@ -4,23 +4,15 @@ import gradio as gr
 from gtts import gTTS
 from datetime import datetime
 from openpyxl import Workbook, load_workbook
-import google.generativeai as genai
-
-# LangChain memory
-from langchain_core.prompts import ChatPromptTemplate
-from langchain_core.messages import HumanMessage, AIMessage
-from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.memory import ConversationBufferMemory
-from langchain.chains import ConversationChain
 
-# ========== SETUP ==========
-genai.configure(api_key="AIzaSyBJFmohAmhmqXQlM3fVxj8MLegVb26kyJk")
+# --- Gemini setup ---
+import google.generativeai as genai
+genai.configure(api_key="AIzaSyBJFmohAmhmqXQlM3fVxj8MLegVb26kyJk")  
 
-llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", convert_system_message_to_human=True)
-memory = ConversationBufferMemory(return_messages=True)
-chain = ConversationChain(llm=llm, memory=memory, verbose=False)
+model = genai.GenerativeModel("models/gemini-1.5-flash-latest")
 
-# Menu & State
+# --- Restaurant menu ---
 MENU = {
     "Cheeseburger": 5.99,
     "Fries": 2.99,
@@ -29,10 +21,14 @@ MENU = {
     "Chicken Wings": 7.99,
     "Salad": 6.99
 }
+
+# --- Memory with LangChain ---
+memory = ConversationBufferMemory(return_messages=True)
+chat_history = []
 order = []
 customer_name = ""
 
-# Excel Setup
+# --- Excel Setup ---
 EXCEL_FILE = "orders.xlsx"
 def setup_excel():
     if not os.path.exists(EXCEL_FILE):
@@ -54,50 +50,66 @@ def save_to_excel(name, items):
     wb.save(EXCEL_FILE)
     return order_id
 
-# Voice
-
+# --- TTS ---
 def clean_text(text):
-    text = re.sub(r"\*\*(.*?)\*\*", r"\1", text)
-    text = re.sub(r"Bot\s*:\s*", "", text, flags=re.IGNORECASE)
+    text = re.sub(r"\*\*(.*?)\*\*", r"\1", text)  # Remove bold
+    text = re.sub(r"Bot\s*:\s*", "", text, flags=re.IGNORECASE)  # Remove "Bot:"
     return text.strip()
 
 def speak(text, filename="response.mp3"):
-    cleaned = clean_text(text)
-    tts = gTTS(text=cleaned)
+    tts = gTTS(text=clean_text(text))
     tts.save(filename)
     return filename
 
-# Prompt Template
-system_prompt = """
-You are a helpful restaurant assistant at 'Systaurant'.
+# --- Generate Gemini response ---
+def generate_response(user_input):
+    global order, customer_name
+
+    menu_str = "\n".join([f"{item}: ${price}" for item, price in MENU.items()])
+    order_summary = ", ".join([f"{qty} x {item}" for item, qty in order]) if order else "No items yet"
 
-Your job is to:
-- Greet the user and ask their name if not known.
-- Show the menu if requested.
-- Extract food items and quantities from the customer's message.
-- If the user says 'done', summarize the order and ask for confirmation.
-- If confirmed, respond with a thank you and order ID.
-- Keep the tone friendly and human. Do not prefix with "Bot:".
+    prompt = f"""
+You are a friendly, intelligent restaurant assistant at 'Systaurant'.
 
 Menu:
-{menu}
-Customer name: {name}
-Order so far: {summary}
+{menu_str}
+
+Customer Name: {customer_name}
+Current Order: {order_summary}
+
+Instructions:
+- Ask for name if not known
+- Show menu if asked
+- Extract food items and quantity
+- Say "Order summary" and ask "Confirm?" when user is done
+- Respond naturally, no "Bot:" prefix
+
+Conversation so far:
 """
 
-prompt = ChatPromptTemplate.from_messages([
-    ("system", system_prompt),
-    ("human", "{input}")
-])
+    for m in memory.chat_memory.messages:
+        role = "Customer" if m.type == "human" else "Bot"
+        prompt += f"{role}: {m.content}\n"
 
-def generate_response(user_input):
+    prompt += f"Customer: {user_input}\nBot:"
+
+    try:
+        response = model.generate_content(prompt)
+        return response.text
+    except Exception as e:
+        return f"❌ Gemini Error: {e}"
+
+# --- Chat handler ---
+def handle_chat(user_input):
     global customer_name, order
 
-    # Track name
+    memory.chat_memory.add_user_message(user_input)
+    bot_reply = generate_response(user_input)
+    memory.chat_memory.add_ai_message(bot_reply)
+
     if "my name is" in user_input.lower():
         customer_name = user_input.split("my name is")[-1].strip().split()[0].title()
 
-    # Track items
     for item in MENU:
         if item.lower() in user_input.lower():
             qty = 1
@@ -107,40 +119,23 @@ def generate_response(user_input):
                     break
             order.append((item, qty))
 
-    # Confirmation
     if "confirm" in user_input.lower() or "yes" in user_input.lower():
         if customer_name and order:
             order_id = save_to_excel(customer_name, order)
-            memory.chat_memory.add_user_message(user_input)
-            memory.chat_memory.add_ai_message(f"✅ Your order ID is {order_id}. Thank you for ordering from Saad's Restaurant!")
-            return f"✅ Your order ID is {order_id}. Thank you for ordering from Saad's Restaurant!"
+            bot_reply += f"\n✅ Your order ID is {order_id}. Thank you for ordering from Systaurant!"
 
-    menu_text = "\n".join([f"{item}: ${price}" for item, price in MENU.items()])
-    summary = ", ".join([f"{qty} x {item}" for item, qty in order]) if order else "No items yet"
-
-    filled_prompt = prompt.invoke({
-        "input": user_input,
-        "menu": menu_text,
-        "name": customer_name or "Not provided",
-        "summary": summary
-    })
-
-    return chain.invoke(filled_prompt.to_string())
-
-def handle_chat(user_input):
-    bot_reply = generate_response(user_input)
     audio_file = speak(bot_reply)
     return bot_reply, audio_file
 
-# Gradio Interface
+# --- Gradio UI ---
 gr.Interface(
     fn=handle_chat,
-    inputs=gr.Textbox(label="👤 You", placeholder="Type your order..."),
+    inputs=gr.Textbox(label="👤 You", placeholder="Type your order here..."),
     outputs=[
         gr.Textbox(label="🤖 Bot Response"),
         gr.Audio(label="🔊 Speaking", autoplay=True)
     ],
-    title="🍔 SysTaurant Voice Bot with Memory",
-    description="Place your restaurant order through chat with memory.",
+    title="🍔 Systaurant Voice Bot",
+    description="A smart voice-enabled assistant to take food orders.",
     theme="soft"
 ).launch(share=True)