Update app.py

app.py

@@ -1,4 +1,11 @@
 import streamlit as st
+import json
+
+# Load furniture categories and attributes from a JSON file
+def load_furniture_data():
+    with open('./config/furniture_taxonomy.json', 'r') as file:
+        data = json.load(file)
+    return data
 
 # Custom CSS to style the buttons
 st.markdown("""
@@ -8,23 +15,25 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # Function to display the main page
-def main_page():
+def main_page(furniture_data):
     st.title("Trending TikTok Furniture Showcase")
-    
-    brand = st.selectbox('Select Brand', ['IKEA', 'Wayfair', 'Ashley Furniture'])
-    color = st.multiselect('Select Color', ['Red', 'Blue', 'Green'])
-    pattern = st.multiselect('Select Pattern', ['Floral', 'Striped', 'Solid'])
-    
+
+    # Dropdown for selecting furniture category
+    categories = list(furniture_data['Furniture'].keys())
+    selected_category = st.selectbox('Select Category', categories)
+
+    # Load attributes based on the selected category
+    attributes = furniture_data['Furniture'][selected_category]['attributes']
+    for attribute, options in attributes.items():
+        st.multiselect(f'Choose {attribute}', options)
+
     if st.button('Show Trending TikToks'):
         st.write("Display trending TikTok images here...")
-        # Here you would add the code to display TikTok images
-        # and links to their videos (considering API availability and legal compliance)
-    
+        # Display TikTok images and links (implementation depends on data source)
+
 # Function to display the TikTok video page
 def tiktok_video_page():
     st.title("TikTok Video and Metadata")
-    
-    # Placeholder for TikTok video and metadata
     st.video("URL_of_the_TikTok_video")
     st.write("Metadata of the TikTok video...")
     
@@ -32,10 +41,11 @@ def tiktok_video_page():
         main_page()
 
 # Main app logic
+furniture_data = load_furniture_data()
 if 'page' not in st.session_state:
     st.session_state.page = 'main'
 
 if st.session_state.page == 'main':
-    main_page()
+    main_page(furniture_data)
 elif st.session_state.page == 'tiktok_video':
     tiktok_video_page()