Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import sqlite3
 import os
 import ast
+import re
+import base64
 from langchain.agents import create_sql_agent, initialize_agent, Tool
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain.agents.agent_types import AgentType
@@ -14,7 +16,7 @@ warnings.filterwarnings("ignore", category=DeprecationWarning)
 
 # Remove any Colab-specific code
 # Streamlit uses its own secrets management
-
+'''
 # Page Configuration
 st.set_page_config(
     page_title="Food Delivery Chatbot",
@@ -78,6 +80,138 @@ if 'db_agent' not in st.session_state:
     st.session_state.db_agent = None
 if 'llm' not in st.session_state:
     st.session_state.llm = None
+'''
+# --- App Configuration ---
+st.set_page_config(page_title="FoodHub Chatbot", page_icon="🍽️", layout="wide")
+
+# --- Load Local JPG Background Image ---
+def get_base64_image(image_path):
+    with open(image_path, "rb") as img_file:
+        encoded = base64.b64encode(img_file.read()).decode()
+    return encoded
+
+image_base64 = get_base64_image("foodhub_background_jpg.jpg")  # Make sure this file exists
+
+# --- Session State Initialization ---
+if "authenticated" not in st.session_state:
+    st.session_state.authenticated = False
+if "customer_id" not in st.session_state:
+    st.session_state.customer_id = None
+if "clear_input" not in st.session_state:
+    st.session_state.clear_input = False
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
+
+if not st.session_state.authenticated:
+    # --- Inject CSS for Background ---
+    st.markdown(
+        f"""
+        <style>
+        .stApp {{
+            background-image: url("data:image/jpeg;base64,{image_base64}");
+            background-size: cover;
+            background-repeat: no-repeat;
+            background-attachment: fixed;
+        }}
+        </style>
+       """,
+       unsafe_allow_html=True
+    )
+else:
+    # Clear background after login
+    st.markdown(f"""
+        <style>
+        /* Remove background from stApp so blurred layer shows through */
+        .stApp {{
+            background-image: none !important;
+            background-color: transparent !important;
+        }}
+        .blurred-bg {{
+            position: fixed;
+            top: 0;
+            left: 0;
+            width: 100vw;
+            height: 100vh;
+            background-image: url("data:image/jpeg;base64,{image_base64}");
+            background-size: cover;
+            background-repeat: no-repeat;
+            background-attachment: fixed;
+            z-index: -1;
+            filter: blur(6px);
+        }}
+        </style>
+        <div class="blurred-bg"></div>
+        """, 
+        unsafe_allow_html=True
+    )
+
+# --- Simple validator ---
+def extract_cust_id(text: str):
+    """Return cust_id in format C#### or None"""
+    m = re.search(r"\b(C\d{4})\b", text, flags=re.I)
+    return m.group(1).upper() if m else None
+
+# --- Validate customer ID using direct SQL ---
+def is_valid_customer(customer_id: str) -> bool:
+    cust_id = extract_cust_id(customer_id)
+    if not cust_id:
+        return True
+
+    try:
+        # Connect to your database
+        conn = sqlite3.connect("customer_orders.db")  # Replace with your actual DB connection
+        cursor = conn.cursor()
+
+        # Run a simple query to check existence
+        cursor.execute("SELECT 1 FROM orders WHERE cust_id = ?", (cust_id,))
+        result = cursor.fetchone()
+
+        conn.close()
+        return result is not None
+
+    except Exception as e:
+        print(f"Database error: {e}")
+        return True
+        
+    if not st.session_state.authenticated:
+    # Login form
+    col1, col2 = st.columns([2, 2]) # Adjust ratios for desired spacing
+
+    with col2:
+        col1, col2 = st.columns([1, 4])
+        with col1:
+            st.image("foodhub_logo.png", width=500)
+        with col2:
+            st.markdown("<h1 style='color: #ff4b4b; padding-top: 10px;'>Welcome to FoodHub Chatbot</h1>", unsafe_allow_html=True)
+        # Instructional message in black
+        st.markdown("<p style='color: black; font-size: 16px;'>Please enter customer ID and password to continue</p>", unsafe_allow_html=True)
+        with st.form("login_form"):
+
+            # Labels in black using label_visibility workaround
+            #st.markdown("<label style='color: black;'>Customer ID</label>", unsafe_allow_html=True)
+            customer_id = st.text_input("Customer ID", placeholder="eg: C1018")
+
+            #st.markdown("<label style='color: black;'>Password</label>", unsafe_allow_html=True)
+            password = st.text_input("Password", type="password")
+            
+            submitted = st.form_submit_button("Login")
+
+            print('password submitted by user : ',password, flush=True)
+            sys.stdout.flush()
+            
+            if submitted:
+                # Add your login logic here
+                if is_valid_customer(customer_id) and password == "foodhub123":
+                    st.session_state.authenticated = True
+                    st.session_state.customer_id = customer_id
+                    
+                    print('user {customer_id} successfully authenticated!!', flush=True)
+                    sys.stdout.flush()
+                    
+                    st.rerun()
+                else:
+                    st.error("Invalid credentials. Please try again.")
+
 
 # Database setup
 @st.cache_resource
@@ -487,6 +621,110 @@ def chatbot_response(cust_id: str, user_query: str) -> str:
 
     return final_llm_response
 
+# --- Chatbot Interface ---
+if st.session_state.authenticated:
+    customer_id = st.session_state.get("customer_id")
+    # Ensure chat history
+    if not st.session_state.chat_history:
+        st.session_state["chat_history"] = [
+            {"role": "assistant", "content": f"Hi! How can I help you today?"}
+        ]
+    spacer_left, chat_col = st.columns([2, 2])
+    
+    with chat_col:
+        col1, col2 = st.columns([2, 2])
+        with col1:
+            st.image("foodhub_logo.png", width=100)
+        with col2:
+            st.markdown(
+                f"""
+                <div style='display: flex; align-items: center; justify-content: flex-start; height: 100%;'>
+                    <h1 style='color: #ff4b4b; margin: 0;'>Hey {customer_id}, Welcome!</h1>
+                </div>
+                """,
+                unsafe_allow_html=True
+            )
+                
+        st.markdown("---")
+        
+        # Inject custom CSS for chat bubbles
+        st.markdown("""
+        <style>
+        .chat-bubble-user {
+            background-color: #dfe6e9;  /* soft gray-blue, neutral on light/dark */
+            color: #000000;             /* black text */
+            padding: 10px 14px;
+            border-radius: 12px;
+            margin-bottom: 6px;
+            display: inline-block;
+            max-width: 80%;
+            text-align: right;
+        }
+        .chat-bubble-bot {
+            background-color: #f1f0f0;  /* light gray, works on both themes */
+            color: #000000;             /* black text */
+            padding: 10px 14px;
+            border-radius: 12px;
+            margin-bottom: 6px;
+            display: inline-block;
+            max-width: 80%;
+            text-align: left;
+        }
+        </style>
+        """, unsafe_allow_html=True)
+
+        # Chat rendering
+        for m in st.session_state.chat_history:
+            left_col, right_col = st.columns([1, 1])
+
+            if m["role"] == "user":
+                with right_col:
+                    st.markdown(
+                        f"""
+                        <div style='display: flex; justify-content: flex-end; align-items: center; margin-bottom: 8px;'>
+                            <div class='chat-bubble-user'>{m['content']}</div>
+                            <div style='font-size: 20px; margin-left: 8px;'>🙋</div>
+                        </div>
+                        """,
+                        unsafe_allow_html=True
+                    )
+
+            else:
+                with left_col:
+                    st.markdown(
+                        f"""
+                        <div style='display: flex; justify-content: flex-start; align-items: center; margin-bottom: 8px;'>
+                            <div style='font-size: 20px; margin-right: 8px;'>🤖</div>
+                            <div class='chat-bubble-bot'>{m['content']}</div>
+                        </div>
+                        """,
+                        unsafe_allow_html=True
+                    )
+
+        # 2) input → append → bot → append → rerun
+        user_input = st.chat_input("Ask about your order or menu...")
+        if user_input:
+            st.session_state.chat_history.append({"role":"user","content":user_input})
+            with st.spinner("Let me check that for you..."):
+                       
+              # Step 2: Refine response using Answer Tool
+              final_response = chatbot_response(customer_id, user_input)
+
+              print('Output of chatbot_response:',final_response, flush=True)
+              sys.stdout.flush()
+                
+            st.session_state.chat_history.append({"role":"assistant","content":final_response})
+            st.rerun()
+            
+        st.markdown("<div style='text-align: right; padding-top: 10px;'>", unsafe_allow_html=True)
+        if st.button("Logout"):
+            st.session_state.authenticated = False
+            st.session_state.customer_id = None
+            st.session_state.chat_history = []
+            st.rerun()
+        st.markdown("</div>", unsafe_allow_html=True)
+
+'''
 # Streamlit UI
 st.markdown(
     '<h1 class="main-header" style="font-weight: normal; font-family: Century, sans-serif;">🛵 FoodHub - AI Chatbot 🤖</h1>', 
@@ -611,3 +849,4 @@ st.markdown(
     '<div style="text-align: center; color: #888;">🙏 Thank you for choosing FoodHub! 🍽️</div>',
     unsafe_allow_html=True
 )
+'''