Update app.py

app.py

@@ -1,9 +1,10 @@
-import streamlit as st 
+import streamlit as st
 from langchain_google_genai import ChatGoogleGenerativeAI
+from datetime import datetime
 
 st.set_page_config(page_title="THE CHEF STORY", layout="wide")
 
-# Styling the page
+# Page styling without using HTML
 st.markdown(
     """
     <style>
@@ -46,6 +47,20 @@ st.markdown(
         flex: 1;
         margin-right: 10px;
     }
+    .quiz-button {
+        display: block;
+        margin: 0 auto;
+        padding: 10px 20px;
+        background-color: #007BFF;
+        color: #fff;
+        border: none;
+        border-radius: 5px;
+        cursor: pointer;
+        font-size: 16px;
+    }
+    .quiz-button:hover {
+        background-color: #0056b3;
+    }
     </style>
     """,
     unsafe_allow_html=True
@@ -59,24 +74,38 @@ def chef_bot(question, chat_history):
     history = "\n".join([f"{sender}: {message}" for sender, message in chat_history])
     prompt = (
         f"You are the chatbot for The Chef Story, a multi-cuisine restaurant chain known for its delicious food and vibrant folklore shows. "
-        f"The Chef Story is located in various cities, providing online ordering and home delivery services with minimal charges of $3 within 4km of radius and addition of $1 for each km. "
+        f"The Chef Story is located in various cities, providing online ordering and home delivery services with minimal charges. "
         f"We offer a variety of cuisines, including Indian, Chinese, Italian, low-calorie options, kids' specials, and desserts, all priced in dollars. "
         f"Here's the current interaction history:\n{history}\nCustomer: {question}\nYou: "
         f"Always respond positively, providing detailed information about our menu, services, and special events. "
         f"Include brief descriptions and prices of the menu items when asked, and make sure to mention our website (www.thechefstory.com) for more information. "
-        f"Welcome the customer warmly and offer a great experience at The Chef Story."
-        f"Ask a separate question from customer if it's the birthday or weddind anniversary of any of his companions. If yes then offer a free cake to the group. "
-        f"Ask a separate question from customer If he has rated the restaurant on the google, if yes then offer a coupon of 20% discount. "
+        f"Welcome the customer warmly and offer a great experience at The Chef Story. "
+        f"If it's the birthday of any of the customer's companions, offer a free cake to the group. "
+        f"If the customer has rated the restaurant on the website, offer a coupon of 20% discount. "
         f"If it is a weekend, offer a free drink to the customer. "
         f"If the customer is visiting on weekends between 3 to 6 p.m., offer a free snack item."
-        f"Highlight the above 4 offers in special bullet points"
-        f'Ask the customer if he wants to play the quiz. Quiz would be related to unique cuisines of different countries in which you ask three questions to the customer along with the options. Ask questions one by one. If he answers all the three correctly then gift him a hamper. Mention this under the "fun activities" category'
+        f"Highlight the above 4 offers in bullet points and bold"
     )
 
     llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"])
     answer = llm.invoke(prompt).content
     return answer
 
+def generate_quiz():
+    quiz_prompt = (
+        "You are the chatbot for The Chef Story, a multi-cuisine restaurant chain. Generate a quiz related to unique cuisine items. "
+        "The quiz should contain three questions. If the customer answers all three questions correctly, they will get a special gift hamper. "
+        "Here are the three quiz questions:\n"
+        "1. What is the main ingredient in the Indian dish 'Paneer Tikka'?\n"
+        "2. Which cuisine is 'Sushi' originally from?\n"
+        "3. What is the primary flavor of Italian 'Tiramisu' dessert?\n"
+        "Please answer the questions one by one."
+    )
+
+    llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"])
+    quiz = llm.invoke(quiz_prompt).content
+    return quiz
+
 if 'chat_history' not in st.session_state:
     st.session_state['chat_history'] = []
 
@@ -88,6 +117,10 @@ def handle_chat():
         st.session_state.chat_history.append(("Chef Story", bot_response))
         st.session_state['user_input'] = ""
 
+def handle_quiz():
+    quiz = generate_quiz()
+    st.session_state.chat_history.append(("Chef Story", quiz))
+
 # Welcome message
 st.markdown("Welcome to The Chef Story where culinary excellence meets folklore magic. We are thrilled to have you here. Hope you will enjoy your meal and have an amazing day.")
 
@@ -100,3 +133,6 @@ if st.session_state.chat_history:
 # Input field for user message
 st.text_input("How may I assist you today?", key="user_input", on_change=handle_chat, placeholder="Ask me anything...")
 
+# Quiz button
+if st.button("Start Quiz", key="quiz_button"):
+    handle_quiz()