Update app.py

app.py

@@ -38,25 +38,34 @@ def filter_images(images_data, selected_options, category):
 # Function to display the main page
 def main_page(furniture_data):
     st.title("Social Media - What's Trending")
-    
-    category = st.selectbox('Select Category', ['Apparel', 'Furnitures', 'Beauty'])
-    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)
+    brands = ['Arhaus', 'Crate&Barrel', 'Williams Sonoma', 'Pottery Barn', 'West Elm', 'Z Gallerie']
+    
+    col1, col2, col3 = st.beta_columns(3)
+
+    with col1:
+        category = st.selectbox('Select Category', ['Apparel', 'Furnitures', 'Beauty'])
+
+    with col2:
+        brand = st.selectbox('Select Brand', brands)
+
+    with col3:
+        selected_category = st.selectbox('Select Category', categories)
+    
 
     # Load attributes based on the selected category
     attributes = furniture_data['Furniture'][selected_category]['attributes']
+    columns = st.beta_columns(len(attributes))
         
     # Create a multi-select widget for each attribute with a unique key
-    for attribute, options in attributes.items():
-        key = f"{selected_category}_{attribute}"  # Unique key combining category and attribute
-        selected_options[attribute] = st.multiselect(f'Choose {attribute}', options, key=key)
+    for i, (attribute, options) in enumerate(attributes.items()):
+        with columns[i]:
+            key = f"{selected_category}_{attribute}"  # Unique key combining category and attribute
+            selected_options[attribute] = st.multiselect(f'Choose {attribute}', options, key=key)
 
     if st.button('Show Filtered Images'):
         # Assuming images_data is loaded from your JSON file