Update app.py

app.py

@@ -214,4 +214,107 @@ def generate_receipt():
     
     total = 0
     for item in st.session_state.cart:
-        price = item['RetailPrice']
+        price = item['RetailPrice']
+        * item['Quantity']
+        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')
+
+# Update main function
+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'):
+                    with st.spinner("Extracting items from image..."):
+                        image = Image.open(uploaded_file)
+                        extracted_text = assistant.extract_text_from_image(image)
+                        st.session_state.query = extracted_text
+                elif uploaded_file.type == 'application/pdf':
+                    with st.spinner("Extracting items from PDF..."):
+                        extracted_text = assistant.extract_text_from_pdf(uploaded_file)
+                        st.session_state.query = extracted_text
+            except Exception as e:
+                st.error(f"Error processing file: {str(e)}")
+
+    col1, col2 = st.columns([2, 1])
+    
+    with col1:
+        st.header("Search Products")
+        query = st.text_area(
+            "Describe what you're looking for (include quantities if needed):",
+            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
+
+                    if isinstance(results, str):
+                        st.write(results)  # Display error messages directly
+                    elif isinstance(results, pd.DataFrame) and not results.empty:  # Check if results is a DataFrame
+                        total_price = 0
+                        for _, row in results.iterrows():
+                            st.write(f"{row['ProductName']} | ${row['RetailPrice']:.2f} x {row['Quantity']} = ${row['RetailPrice'] * row['Quantity']:.2f}")
+                            total_price += row['RetailPrice'] * row['Quantity']
+                            if st.button("Add to Cart", key=f"add_{row['ProductName']}"):  # Unique key for each button
+                                add_to_cart(row.to_dict())
+                        st.write(f"**Total: ${total_price:.2f}**")  # Display total price for search results
+                    else:
+                        st.write("No products found matching your criteria.")  # Handle empty DataFrame case
+
+
+    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])
+                with cols[0]:
+                    st.write(f"{item['ProductName']} x{item['Quantity']}")
+                with cols[1]:
+                    st.write(f"${item['RetailPrice'] * item['Quantity']:.2f}")
+                with cols[2]:
+                    if st.button("❌", key=f"del_{item['ProductName']}"):
+                        remove_from_cart(item['ProductName'])
+                        st.rerun()
+                total += item['RetailPrice'] * item['Quantity']
+            st.divider()
+            st.write(f"**Total: ${total:.2f}**")
+            
+            if st.button("Checkout"):
+                receipt = generate_receipt()
+                st.download_button(
+                    label="Download Receipt",
+                    data=receipt,
+                    file_name="bon_marche_receipt.pdf",
+                    mime="application/pdf"
+                )
+        else:
+            st.write("Your cart is empty")
+
+if __name__ == "__main__":
+    main()