Update app.py

app.py

@@ -7,6 +7,13 @@ def load_furniture_data():
         data = json.load(file)
     return data
 
+def load_images():
+    with open('images.json', 'r') as file:
+        images_data = json.load(file)
+    return images_data
+
+
+
 # Custom CSS to style the buttons
 st.markdown("""
 <style>
@@ -14,6 +21,26 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
 
+def filter_images(images_data, selected_options):
+    filtered_images = []
+
+    for image_name, image_info in images_data.items():
+        matches_all = True  # A flag to check if the image matches all selected attributes
+
+        for attribute, selected_values in selected_options.items():
+            if selected_values:  # Check if any value is selected for this attribute
+                image_attribute_value = image_info.get('attributes', {}).get(attribute)
+                if image_attribute_value not in selected_values:
+                    matches_all = False
+                    break  # Break the loop if any attribute does not match
+
+        if matches_all:
+            filtered_images.append(image_name)
+
+    return filtered_images
+
+
+    
 # Function to display the main page
 def main_page(furniture_data):
     st.title("Social Media - What's Trending")
@@ -22,6 +49,9 @@ def main_page(furniture_data):
     brands = ['Arhaus', 'Crate&Barrel', 'Williams Sonoma', 'Pottery Barn', 'West Elm', 'Z Gallerie']
     brand = st.selectbox('Select Brand', brands)
 
+    # Dictionary to store selected options for each attribute
+    selected_options = {}
+
     # Dropdown for selecting furniture category
     categories = list(furniture_data['Furniture'].keys())
     selected_category = st.selectbox('Select Category', categories)
@@ -30,10 +60,17 @@ def main_page(furniture_data):
     attributes = furniture_data['Furniture'][selected_category]['attributes']
     for attribute, options in attributes.items():
         st.multiselect(f'Choose {attribute}', options)
+    # Create a multi-select widget for each attribute
+    for attribute, options in attributes.items():
+        selected_options[attribute] = st.multiselect(f'Choose {attribute}', options)
+
+    filtered_images = filter_images(images_data, selected_options)
 
-    if st.button('Show Trending TikToks'):
-        st.write("Display trending TikTok images here...")
-        # Display TikTok images and links (implementation depends on data source)
+    if filtered_images:
+        for image in filtered_images:
+            st.image(image)  # Display each filtered image
+    else:
+        st.write("Not Trending")
 
 # Function to display the TikTok video page
 def tiktok_video_page():
@@ -46,6 +83,7 @@ def tiktok_video_page():
 
 # Main app logic
 furniture_data = load_furniture_data()
+images_data = load_images()
 if 'page' not in st.session_state:
     st.session_state.page = 'main'