Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import sqlite3
 import streamlit as st
 from werkzeug.security import generate_password_hash, check_password_hash
-from datetime import datetime
 
 from langchain_groq import ChatGroq
 from langchain_huggingface import HuggingFaceEmbeddings
@@ -47,13 +46,11 @@ def init_db():
                   features TEXT NOT NULL,
                   stock INTEGER NOT NULL DEFAULT 0)''')
     
-    # Inventory log table
-    c.execute('''CREATE TABLE IF NOT EXISTS inventory_log
-                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
-                  timestamp TEXT NOT NULL,
-                  product_name TEXT NOT NULL,
-                  found INTEGER NOT NULL,
-                  username TEXT)''')
+    # Check and update schema if needed
+    c.execute("PRAGMA table_info(products)")
+    columns = [column[1] for column in c.fetchall()]
+    if 'stock' not in columns:
+        c.execute('ALTER TABLE products ADD COLUMN stock INTEGER NOT NULL DEFAULT 0')
     
     # Insert default company settings if empty
     c.execute('SELECT COUNT(*) FROM company_settings')
@@ -71,40 +68,8 @@ def init_db():
 
 conn = init_db()
 
-# --- Inventory Management Functions ---
-def log_inventory_access(product_name: str, found: bool, username: str = None):
-    """Log all inventory access attempts"""
-    c = conn.cursor()
-    c.execute('''INSERT INTO inventory_log 
-                 (timestamp, product_name, found, username)
-                 VALUES (?, ?, ?, ?)''',
-              (datetime.now().isoformat(), product_name, int(found), username))
-    conn.commit()
-
-def validate_product_in_inventory(product_name: str) -> bool:
-    """Check if product exists and is in stock"""
-    c = conn.cursor()
-    c.execute('''SELECT id FROM products 
-                 WHERE LOWER(name) LIKE LOWER(?) AND stock > 0''',
-              (f'%{product_name}%',))
-    return c.fetchone() is not None
-
-def get_inventory_report():
-    """Generate inventory report for admin"""
-    c = conn.cursor()
-    c.execute('''SELECT name, category, price, stock 
-                 FROM products ORDER BY category, name''')
-    return c.fetchall()
-
-def get_inventory_alerts():
-    """Get low stock alerts"""
-    c = conn.cursor()
-    c.execute('''SELECT name, stock FROM products 
-                 WHERE stock <= 3 AND stock > 0
-                 ORDER BY stock ASC''')
-    return c.fetchall()
-
 # --- Admin Classes ---
+
 class Company:
     @staticmethod
     def get_settings():
@@ -128,15 +93,6 @@ class Product:
         c.execute('SELECT * FROM products')
         return c.fetchall()
 
-    @staticmethod
-    def search(query):
-        c = conn.cursor()
-        c.execute('''SELECT * FROM products 
-                     WHERE LOWER(name) LIKE LOWER(?) 
-                     OR LOWER(category) LIKE LOWER(?)''',
-                  (f'%{query}%', f'%{query}%'))
-        return c.fetchall()
-
     @staticmethod
     def get_by_id(product_id):
         c = conn.cursor()
@@ -176,17 +132,21 @@ class User:
     @classmethod
     def create(cls, username, password):
         hashed_pw = generate_password_hash(password)
+        conn = sqlite3.connect('users.db')
         c = conn.cursor()
         c.execute('INSERT INTO users (username, password) VALUES (?, ?)', (username, hashed_pw))
         user_id = c.lastrowid
         conn.commit()
+        conn.close()
         return cls(user_id, username, hashed_pw)
 
     @classmethod
     def get_by_username(cls, username):
+        conn = sqlite3.connect('users.db')
         c = conn.cursor()
         c.execute('SELECT * FROM users WHERE username = ?', (username,))
         user = c.fetchone()
+        conn.close()
         if user:
             return cls(user[0], user[1], user[2], 
                       eval(user[3]) if user[3] else [],
@@ -195,19 +155,23 @@ class User:
 
     def update_chat_history(self, new_messages):
         updated_history = self.chat_history + new_messages
+        conn = sqlite3.connect('users.db')
         c = conn.cursor()
         c.execute('UPDATE users SET previous_chat_history = ? WHERE id = ?',
                  (str(updated_history), self.id))
         conn.commit()
-        self.chat_history = updated_history
+        conn.close()
+        self.chat_history = updated_history  # Update in-memory
 
     def update_products_bought(self, new_products):
         updated_products = self.products_bought + new_products
+        conn = sqlite3.connect('users.db')
         c = conn.cursor()
         c.execute('UPDATE users SET previous_products_bought = ? WHERE id = ?',
                  (str(updated_products), self.id))
         conn.commit()
-        self.products_bought = updated_products
+        conn.close()
+        self.products_bought = updated_products  # Update in-memory
 
 # --- AI Setup ---
 os.environ["GROQ_API_KEY"] = st.secrets["GROQ_API_KEY"]
@@ -236,23 +200,14 @@ def load_data():
 retriever = load_data()
 
 def retrieve_query(query: str):
-    """Enhanced product retrieval with strict inventory checks"""
     docs = retriever.get_relevant_documents(query)
-    valid_docs = [doc for doc in docs if doc.metadata.get('stock', 0) > 0]
-    
-    if not valid_docs:
-        return [Document(
-            page_content=f"No matching in-stock products found for '{query}'",
-            metadata={"id": -1, "stock": 0}
-        )]
-    
-    st.session_state.last_retrieved_docs = valid_docs
-    return valid_docs
+    st.session_state.last_retrieved_docs = docs
+    return docs
 
 tool = Tool(
     name="product_retriever",
     func=retrieve_query,
-    description="Useful for retrieving product information from CURRENT INVENTORY only"
+    description="Useful for retrieving product information including current stock levels"
 )
 
 # --- Admin Dashboard ---
@@ -311,39 +266,13 @@ def admin_dashboard():
                     st.rerun()
         else:
             st.info("No products found in database")
-    
-    with st.expander("Inventory Reports"):
-        st.subheader("Current Inventory Status")
-        inventory = get_inventory_report()
-        if inventory:
-            st.table(inventory)
-        else:
-            st.warning("No inventory data available")
-        
-        st.subheader("Low Stock Alerts")
-        alerts = get_inventory_alerts()
-        if alerts:
-            for item in alerts:
-                st.warning(f"{item[0]} - Only {item[1]} left!")
-        else:
-            st.info("No low stock alerts")
-        
-        st.subheader("Inventory Access Logs")
-        c = conn.cursor()
-        c.execute('''SELECT timestamp, product_name, found, username 
-                     FROM inventory_log ORDER BY timestamp DESC LIMIT 100''')
-        logs = c.fetchall()
-        if logs:
-            st.table(logs)
-        else:
-            st.info("No inventory access logs yet")
 
 # --- Main App ---
 def main():
     company_settings = Company.get_settings()
     company_name = company_settings[1]
 
-    st.title(f"{company_name} AI Sales Assistant 🤖")
+    st.title("AI Sales Assistant 🤖")
     
     if 'user' not in st.session_state:
         st.session_state.user = None
@@ -353,7 +282,7 @@ def main():
     
     # Authentication
     if not st.session_state.user and not st.session_state.admin_mode:
-        st.header("Login/Register")
+        st.header("Login/Register Admin")
         tab1, tab2, tab3 = st.tabs(["Login", "Register", "Admin"])
         
         with tab1:
@@ -400,8 +329,8 @@ def main():
 
     else:
         # Chat Interface
-        st.header(f"Welcome, {st.session_state.user.username}!")
-        st.caption(f"You're chatting with {company_settings[3]}, your AI sales assistant")
+        st.header(f"Welcome to {company_name}, {st.session_state.user.username}!")
+        st.subheader("Chat with our AI Sales Assistant")
 
         # Display Chat History
         for msg in st.session_state.chat_history:
@@ -425,19 +354,6 @@ def main():
         - Key Features: {company_settings[4]}
         - Agent Name: {company_settings[3]}
 
-        CURRENT INVENTORY:
-        {current_products}
-        
-        STRICT INVENTORY POLICY:
-        - You MUST ONLY recommend products that exist in our CURRENT INVENTORY
-        - NEVER suggest, mention, or describe products not in our database
-        - If asked for unavailable items, respond: "I apologize, we don't currently carry that product."
-
-        When recommending products:
-        1. FIRST verify the product exists in the above inventory
-        2. ONLY discuss products that appear in the inventory list
-        3. For unavailable items, DO NOT suggest alternatives unless they exist in inventory 
-        
         ## Inventory Management Policy
         1. **Stock Verification**:
            - ALWAYS check current stock before recommending any product
@@ -482,6 +398,9 @@ def main():
         - Emojis: Use sparingly (1-2 per message max)
         - Branding: Consistently reference {company_settings[1]} when appropriate
 
+        ## Current Inventory Status
+        {current_products}
+
         ## Important Reminders
         1. NEVER recommend products not in our inventory
         2. ALWAYS verify stock before discussing any product
@@ -490,7 +409,6 @@ def main():
         5. For complex queries, offer to connect with human support
         """
 
-
         prompt = ChatPromptTemplate.from_messages([
             ("system", system_prompt),
             MessagesPlaceholder(variable_name="chat_history"),
@@ -503,35 +421,20 @@ def main():
         agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)
 
         if prompt_input := st.chat_input("Type your message here..."):
-            # Log inventory access attempt
-            log_inventory_access(
-                product_name=prompt_input,
-                found=validate_product_in_inventory(prompt_input),
-                username=st.session_state.user.username
-            )
-
             with st.chat_message("user"):
                 st.write(prompt_input)
 
             with st.chat_message("assistant"):
                 response = agent_executor.invoke({
                     "input": prompt_input,
-                    "chat_history": [HumanMessage(content=msg["content"]) if msg["type"] == "human" 
-                                   else AIMessage(content=msg["content"]) 
+                    "chat_history": [HumanMessage(content=msg["content"]) if msg["type"] == "human" else AIMessage(content=msg["content"]) 
                                    for msg in st.session_state.chat_history]
                 })["output"]
-                
-                # Post-process response to ensure inventory compliance
-                if "last_retrieved_docs" in st.session_state:
-                    if st.session_state.last_retrieved_docs[0].metadata.get("id") == -1:
-                        response = f"I apologize, we don't currently carry that product in our inventory."
-                    elif not st.session_state.last_retrieved_docs:
-                        response = f"I couldn't find that product in our current inventory."
-                
                 st.write(response)
                 
                 # Handle inventory update on purchase
                 if "https://www.example.com/payment" in response:
+                    # Extract product name from the agent's response
                     product_name = None
                     if "proceed with the purchase of" in response:
                         start = response.find("proceed with the purchase of") + len("proceed with the purchase of")
@@ -539,16 +442,19 @@ def main():
                         product_name = response[start:end].strip()
                     
                     if product_name:
+                        # Retrieve current product details
                         docs = retrieve_query(product_name)
-                        if docs and docs[0].metadata.get("id") != -1:
+                        if docs:
                             product_doc = docs[0]
                             product_id = product_doc.metadata.get("id")
                             current_stock = product_doc.metadata.get("stock")
                             
                             if product_id and current_stock > 0:
                                 try:
-                                    # Verify current stock
+                                    conn = sqlite3.connect('users.db')
                                     c = conn.cursor()
+                                    
+                                    # Verify current stock
                                     c.execute('SELECT stock FROM products WHERE id = ?', (product_id,))
                                     actual_stock = c.fetchone()[0]
                                     
@@ -558,23 +464,38 @@ def main():
                                                 (actual_stock - 1, product_id))
                                         
                                         # Update user's purchase history
-                                        st.session_state.user.products_bought.append(product_name)
-                                        st.session_state.user.update_products_bought([product_name])
+                                        c.execute('SELECT previous_products_bought FROM users WHERE id = ?', 
+                                                (st.session_state.user.id,))
+                                        current_purchases = c.fetchone()[0]
+                                        
+                                        updated_purchases = eval(current_purchases) + [product_name] if current_purchases else [product_name]
+                                        
+                                        c.execute('UPDATE users SET previous_products_bought = ? WHERE id = ?',
+                                                (str(updated_purchases), st.session_state.user.id))
                                         
                                         conn.commit()
+                                        st.session_state.user.products_bought.append(product_name)
                                         st.success(f"Successfully purchased {product_name}! Stock updated.")
                                         load_data.clear()
                                         
                                         # Check for low stock
-                                        if (actual_stock - 1) <= 3:
-                                            st.warning(f"Low stock alert: Only {actual_stock - 1} units remaining!")
+                                        if (actual_stock - 1) <= 3 and (actual_stock - 1) > 0:
+                                            st.warning(f"Low stock alert: Only {actual_stock - 1} units of {product_name} remaining!")
                                     else:
                                         st.error(f"Sorry, {product_name} is now out of stock!")
                                 except Exception as e:
                                     st.error(f"Error processing purchase: {str(e)}")
-                                    conn.rollback()
+                                    if conn:
+                                        conn.rollback()
                                 finally:
-                                    c.close()
+                                    if conn:
+                                        conn.close()
+                            else:
+                                st.error("Product is out of stock or not found in the inventory.")
+                        else:
+                            st.error("Could not retrieve product details. Please try again.")
+                    else:
+                        st.error("Please specify which product you'd like to purchase.")
 
             new_messages = [
                 {"type": "human", "content": prompt_input},