Update app.py

app.py

@@ -22,6 +22,7 @@ def configure_gemini(api_key):
     return genai.GenerativeModel('gemini-2.0-flash-thinking-exp')
 
 model = configure_gemini(os.environ['GOOGLE_API_KEY'])
+            
 # Initialize Gemini models
 llm_flash_exp = ChatGoogleGenerativeAI(
     model="gemini-2.0-flash-exp",
@@ -46,14 +47,16 @@ class SmartShoppingAssistant:
             cutoff=threshold
         )
         return matches if matches else []
-
+    
     def match_products_with_catalogue(self, extracted_items):
         """Match extracted items with catalogue products using Gemini"""
         product_list = self.df['ProductName'].tolist()
         product_string = ", ".join(product_list)
+        
         prompt = f"""
         Given these extracted items and quantities:
         {extracted_items}
+        
         And this product catalogue:
         {product_string}
 
@@ -68,16 +71,18 @@ class SmartShoppingAssistant:
 
         Only include products that have good matches in the catalogue.
         """
+        
         try:
             matches = llm_flash_exp.predict(prompt)
             return matches.strip()
         except Exception as e:
             return f"Error matching products: {str(e)}"
-
+    
     def search_products_fuzzy(self, product_names_with_quantities):
         """Improved fuzzy search with batch processing"""
         results = pd.DataFrame()
         matched_products = set()
+        
         for item in product_names_with_quantities:
             product_info = item.split('quantity:')
             clean_name = product_info[0].strip().upper().replace('PRODUCTNAME ==', '').strip(' "\'')
@@ -95,14 +100,16 @@ class SmartShoppingAssistant:
                     results = pd.concat([results, matched])
                     matched_products.add(clean_name)
                     break  # Take first good match
+        
         return results.drop_duplicates(subset=['CleanName'])
-
+        
     def setup_agent(self):
         """Set up the LangChain agent with necessary tools"""
         def search_products(query):
             try:
                 # Split into individual product entries
                 product_entries = [entry.strip() for entry in query.split('or')]
+                
                 results = self.search_products_fuzzy(product_entries)
                 if not results.empty:
                     # Format results with quantity
@@ -114,7 +121,7 @@ class SmartShoppingAssistant:
                 return "No products found matching your criteria."
             except Exception as e:
                 return f"Error executing query: {str(e)}"
-
+        
         tools = [
             Tool(
                 name="Product Search",
@@ -122,6 +129,7 @@ class SmartShoppingAssistant:
                 description="Search for products in the supermarket database using fuzzy matching"
             )
         ]
+        
         self.agent = initialize_agent(
             tools=tools,
             memory=self.memory,
@@ -130,7 +138,7 @@ class SmartShoppingAssistant:
             verbose=True,
             max_iterations=3
         )
-
+    
     def process_natural_language_query(self, query):
         """Process natural language query with two-step matching"""
         try:
@@ -138,18 +146,23 @@ class SmartShoppingAssistant:
             extraction_prompt = f"""
             Extract the products and their quantities from this shopping request.
             Include any specific requirements mentioned.
+            
             Shopping request: {query}
+            
             Format each item with its quantity (assume 1 if not specified).
             """
+            
             extracted_items = llm_flash_exp.predict(extraction_prompt)
+            
             # Second step: Match with catalogue
             matched_products = self.match_products_with_catalogue(extracted_items)
+            
             # Third step: Search and return results
             result = self.agent.run(f"Search for products matching the specified names: {matched_products}")
             return result
         except Exception as e:
             return f"Error processing query: {str(e)}"
-
+    
     def extract_text_from_image(self, image):
         """Extract text from uploaded image using Gemini"""
         prompt = """
@@ -162,7 +175,7 @@ class SmartShoppingAssistant:
             return response.text
         except Exception as e:
             return f"Error processing image: {str(e)}"
-
+    
     def extract_text_from_pdf(self, pdf_file):
         """Extract text from uploaded PDF"""
         try:
@@ -179,6 +192,7 @@ class SmartShoppingAssistant:
 def add_to_cart(product):
     if 'cart' not in st.session_state:
         st.session_state.cart = []
+        
     # Check if product exists in cart
     existing = next((item for item in st.session_state.cart if item['ProductName'] == product['ProductName']), None)
     if existing:
@@ -194,15 +208,18 @@ def generate_receipt():
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
+    
     pdf.cell(200, 10, txt="Bon Marche Receipt", ln=1, align='C')
     pdf.cell(200, 10, txt=f"Date: {pd.Timestamp.now().strftime('%Y-%m-%d %H:%M')}", ln=1)
+    
     total = 0
     for item in st.session_state.cart:
         price = item['RetailPrice'] * item['Quantity']
-        pdf.cell(200, 10,
+        pdf.cell(200, 10, 
                  txt=f"{item['ProductName']} x{item['Quantity']} - ${price:.2f}",
                  ln=1)
         total += price
+    
     pdf.cell(200, 10, txt=f"Total: ${total:.2f}", ln=1)
     return pdf.output(dest='S').encode('latin1')
 
@@ -210,20 +227,21 @@ def generate_receipt():
 def main():
     st.set_page_config(page_title="Smart Shopping Assistant", layout="wide")
     st.title("🛒 Smart Shopping Assistant")
-
+    
     @st.cache_data
     def load_product_data():
         return pd.read_csv('supermarket4i.csv')  # Ensure correct filename
-
+    
     df = load_product_data()
     assistant = SmartShoppingAssistant(df)
-
+    
     with st.sidebar:
         st.header("Upload Shopping List")
         uploaded_file = st.file_uploader(
             "Upload an image or PDF of your shopping list",
             type=['png', 'jpg', 'jpeg', 'pdf']
         )
+        
         if uploaded_file:
             try:
                 if uploaded_file.type.startswith('image'):
@@ -239,6 +257,7 @@ def main():
                 st.error(f"Error processing file: {str(e)}")
 
     col1, col2 = st.columns([2, 1])
+    
     with col1:
         st.header("Search Products")
         query = st.text_area(
@@ -246,29 +265,32 @@ def main():
             height=100,
             value=st.session_state.get('query', '')
         )
+        
         if st.button("Search"):
             if query:
                 with st.spinner("Searching..."):
                     results = assistant.process_natural_language_query(query)
                     st.session_state.last_results = results
+                    
                     # Display results with add to cart buttons
                     if isinstance(results, str):
                         st.write(results)
                     else:
                         for _, row in results.iterrows():
-                            cola, colb = st.columns([3, 1])
+                            cola, colb = st.columns([3,1])
                             with cola:
                                 st.write(f"**{row['ProductName']}**")
                                 st.write(f"Price: ${row['RetailPrice']} | Qty: {row['Quantity']}")
                             with colb:
                                 if st.button("Add", key=row['ProductName']):
                                     add_to_cart(row.to_dict())
+    
     with col2:
         st.header("Shopping Cart")
         if 'cart' in st.session_state and st.session_state.cart:
             total = 0
             for item in st.session_state.cart:
-                cols = st.columns([3, 1, 1])
+                cols = st.columns([3,1,1])
                 with cols[0]:
                     st.write(f"{item['ProductName']} x{item['Quantity']}")
                 with cols[1]:
@@ -280,6 +302,7 @@ def main():
                 total += item['RetailPrice'] * item['Quantity']
             st.divider()
             st.write(f"**Total: ${total:.2f}**")
+            
             if st.button("Checkout"):
                 receipt = generate_receipt()
                 st.download_button(