Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import os
 from PIL import Image
-import numpy as np
 from chatbot import Chatbot  # Assuming you have a chatbot module
 
 # Function to save uploaded file
@@ -43,11 +42,19 @@ def show_dashboard():
             _, recommended_products = chatbot.generate_response(caption)
 
             st.write("### Recommended Products")
-            col1, col2, col3, col4, col5 = st.columns(5)
-            for i, idx in enumerate(recommended_products[:5]):
-                with col1 if i == 0 else col2 if i == 1 else col3 if i == 2 else col4 if i == 3 else col5:
+            if 'num_recommendations' not in st.session_state:
+                st.session_state.num_recommendations = 5
+
+            for i, idx in enumerate(recommended_products[:st.session_state.num_recommendations]):
+                col = st.columns(5)[i % 5]
+                with col:
                     product_image = chatbot.images[idx['corpus_id']]
                     st.image(product_image, caption=f"Product {i+1}", width=150)
+
+            if st.session_state.num_recommendations < len(recommended_products):
+                if st.button("Suggest More"):
+                    st.session_state.num_recommendations += 5
+                    st.experimental_rerun()
         else:
             st.error("Error in uploading the file.")