Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import pandas as pd
 import json
 
 # Function to validate zipcode
@@ -46,26 +47,14 @@ if 'form_submitted' not in st.session_state:
 if 'shelter_index' not in st.session_state:
     st.session_state.shelter_index = 0
 
-shelters = [
-    {
-        "title": "Shelter 1",
-        "description": "This is the 1st shelter. API Call will generate this outcome.",
-        "header": "A cat",
-        "image_url": "https://static.streamlit.io/examples/cat.jpg"
-    },
-    {
-        "title": "Shelter 2",
-        "description": "This is the 2nd shelter. API Call will generate this outcome....",
-        "header": "A dog",
-        "image_url": "https://static.streamlit.io/examples/dog.jpg"
-    },
-    {
-        "title": "Shelter 3",
-        "description": "This is the 3rd shelter. API Call will generate this outcome....",
-        "header": "An owl",
-        "image_url": "https://static.streamlit.io/examples/owl.jpg"
-    }
-]
+# Load the master database
+master_database = pd.DataFrame({
+    "gender": ["Male", "Female", "Other", "Male", "Female", "Other"],
+    "city": ["Oakland", "Berkeley", "SF", "SF", "Oakland", "Berkeley"],
+    "zipcode": ["94612", "94704", "94103", "94103", "94612", "94704"],
+    "duration": ["Long Term", "Transitional", "Temporary", "Long Term", "Transitional", "Temporary"],
+    "needs": ["Food and shelter", "Medical assistance", "Job placement", "Food and job", "Medical and shelter", "Shelter and education"]
+})
 
 # Page config
 st.set_page_config(
@@ -95,26 +84,53 @@ else:
     # Display
     st.json(data)
 
-    # Display the current shelter information
-    shelter = shelters[st.session_state.shelter_index]
-    st.write(shelter["description"])
-    st.header(shelter["header"])
-    st.image(shelter["image_url"])
-
-    # Navigation buttons
-    col1, col2, col3 = st.columns([1, 1, 1])
-
-    with col1:
-        if st.button("Previous"):
-            if st.session_state.shelter_index > 0:
-                st.session_state.shelter_index -= 1
-                st.experimental_rerun()
-
-    with col2:
-        st.write("")
-
-    with col3:
-        if st.button("Next"):
-            if st.session_state.shelter_index < len(shelters) - 1:
-                st.session_state.shelter_index += 1
-                st.experimental_rerun()
+    # Filter the master database
+    filtered_data = master_database[
+        (master_database['gender'] == data['Gender']) & 
+        (master_database['city'] == data['City']) &
+        (master_database['duration'] == data['Duration']) & 
+        (master_database['needs'].str.contains(data['Needs'], case=False))
+    ]
+
+    # Create the client database
+    client_database = filtered_data.to_dict(orient='records')
+
+    # Convert client_database to shelters format
+    shelters = [
+        {
+            "title": f"Shelter {i+1}",
+            "description": f"This is shelter {i+1}.",
+            "header": "Details",
+            "image_url": "https://static.streamlit.io/examples/cat.jpg" if i % 2 == 0 else "https://static.streamlit.io/examples/dog.jpg",
+            "info": row
+        }
+        for i, row in enumerate(client_database)
+    ]
+
+    if len(shelters) == 0:
+        st.warning("No shelters found matching your criteria.")
+    else:
+        # Display the current shelter information
+        shelter = shelters[st.session_state.shelter_index]
+        st.write(shelter["description"])
+        st.header(shelter["header"])
+        st.image(shelter["image_url"])
+        st.json(shelter["info"])
+
+        # Navigation buttons
+        col1, col2, col3 = st.columns([1, 1, 1])
+
+        with col1:
+            if st.button("Previous"):
+                if st.session_state.shelter_index > 0:
+                    st.session_state.shelter_index -= 1
+                    st.experimental_rerun()
+
+        with col2:
+            st.write("")
+
+        with col3:
+            if st.button("Next"):
+                if st.session_state.shelter_index < len(shelters) - 1:
+                    st.session_state.shelter_index += 1
+                    st.experimental_rerun()