ShelterSearch / app.py
KeshavRa's picture
Update app.py
4c8b4d5 verified
raw
history blame
3.48 kB
import streamlit as st
import json
import pandas as pd
def validate_zipcode(zipcode):
try:
if len(zipcode) == 5 and zipcode.isdigit():
return True
else:
return False
except:
return False
# Function to handle form submission
@st.experimental_dialog("Fill out the form")
def form_dialog():
sex = st.radio("Sex", ("Male", "Female", "Other"))
city = st.radio("City", ("Oakland", "Berkeley", "SF"))
lgbtq = st.radio("Do you identify as LGBTQ+ (some shelters serve specifically this community", ("Yes", "No"))
domestic_violence = st.radio("Are you experiencing domestic violence (some shelters serve these individuals specifically", ("Yes", "No"))
zipcode = st.text_input("Zipcode (Optional)", value="")
urgency = st.radio("How quickly do you need help?", ("Today", "In the next few days", "In a week or more"))
duration = st.radio("How long do you need a place to stay?", ("A year or more", "A couple of months", "A month or less"))
needs = st.text_area("Needs (Optional - tell us what you need and how we can help)", value="")
if st.button("Submit"):
if zipcode and not validate_zipcode(zipcode):
st.error("Please enter a valid 5-digit zipcode.")
else:
data = {
"Sex": sex,
"City": city,
"LGBTQ": lgbtq,
"Domestic Violence": domestic_violence
"Zipcode": zipcode,
"Urgency": urgency,
"Duration": duration,
"Needs": needs
}
with open('data.json', 'w') as f:
json.dump(data, f)
st.session_state.form_submitted = True
st.session_state.data = data
st.rerun()
shelters = pd.read_csv("Database.csv")
filtered_shelters = shelters[(shelters['Sex'] == data['Sex']) & (shelters['City'] == data['City']) & (shelters['LGBTQ'] == data['LGBTQ']) & (shelters['Domestic Violence'] == data['Domestic Violence'])]
st.table(filtered_shelters)
# Initialize session state
if 'form_submitted' not in st.session_state:
st.session_state.form_submitted = False
if 'shelter_index' not in st.session_state:
st.session_state.shelter_index = 0
shelters = [
{"title": "Shelter 1", "description": "This is the 1st shelter",},
{"title": "Shelter 2", "description": "This is the 2nd shelter.",},
{"title": "Shelter 3", "description": "This is the 3rd shelter.",}
]
# Page config
st.set_page_config(
page_title="ShelterSearch",
layout="wide",
)
st.title("ShelterSearch")
if not st.session_state.form_submitted:
if st.button("Open Form"):
form_dialog()
else:
with open('data.json', 'r') as f:
data = json.load(f)
# Display
st.json(data)
# Display the current shelter information
shelter = shelters[st.session_state.shelter_index]
st.write(shelter["description"])
# Create two columns
col1, col2 = st.columns([1,1])
# Add buttons to each column
with col1:
if st.button("Previous"):
if st.session_state.shelter_index > 0:
st.session_state.shelter_index -= 1
st.experimental_rerun()
with col2:
if st.button("Next"):
if st.session_state.shelter_index < len(shelters) - 1:
st.session_state.shelter_index += 1
st.experimental_rerun()