Update app.py

app.py

@@ -37,95 +37,29 @@ def gen_load() -> pd.DataFrame:
 
 #AI Emotional State Score: Anxiety, Confusion, Trepidation, Fear, Guilt
 
+import streamlit as st
+import pandas as pd
+import pydeck as pdk
 
-#queried_ctss = st.multiselect("Ctss",options=gen_load()["Ctss"].value_counts().reset_index())
-def load_data(state = None,zip_code = None,city = None,age = None,male  = None,female = None,selected_cohorts = None) -> pd.DataFrame:
-    df = gen_load()
-    try:
-        if (male or female) and not (male and female):
-            df = df.loc[df['Gender'] == ("Male" if male else ("Female" if female else ""))]
-        elif not (male or female):
-            df = df.loc[df['Gender'] == "(null)"]
-    except:
-        pass
-    try:
-        df = df[(df['Age'].gt(age[0]) & df['Age'].lt(age[1]))]
-    except:
-        pass
-    try:
-        if zip_code or state or city:
-            val = "Zip" if zip_code != "" else "City" if city != "" else "Abb"
-            df = df.loc[df[val] == (zip_code or city or state)]
-    except:
-        pass
-
-    try:
-        if len(selected_cohorts) > 0:
-            df = df.loc[df['Cohort'].isin(selected_ctss)]
-                
-    except:
-        pass
-
-    #value_counts = df["FCName"].value_counts()
-    #df2 = pd.DataFrame(value_counts)
-    #df2 = df2.reset_index()
-    #df2.columns = ['FCName', 'Count']
-    #df = df.merge(df2, left_on = 'FCName', right_on = 'FCName',how='left')
-
-    return df
-
-def mapF(data):
-    print(data)
-    geo = data.iloc[0]
-    print(geo)
-    #print(geo[7], geo[8])
-    #lat = float(geo['lat'])
-    #lon = float(geo['lon'])
-    lat = float(geo[7])
-    lon = float(geo[8])
-    max = data['Count'].max()
-    view_state = pdk.ViewState(
-        pitch = 40.5,
-        bearing = -27.36,
-        latitude = lat,
-        longitude = lon,
-        zoom = 4,
-    )
+def ShowCityDataframe(DATA_URL, SEARCH_DATA):
+    df = pd.read_csv(DATA_URL)
+    st.title("City FIPS, Location, and Population")
+    st.text("Search for any city in the United States:")
 
-    layer = pdk.Layer(
-            'ColumnLayer',
-            data=data,
-            get_position='[lon, lat]',
-            pickable=True,
-            extruded=True,
-            auto_highlight=True,
-            get_elevation="Count",
-            # cell_size=200,
-            radius = 1000,
-            elevation_scale=1000000/ max,
-            get_fill_color='[255, 255 - (Count/3500) * 255, 0,100]',
-            coverage=10
-    )                 
+    # Search query for existing data
+    search_query = st.text_input(label="City Name", value="")
+    if search_query != "":
+        df = df[df["city"].str.contains(search_query, case=False)]
 
-    r = pdk.Deck(
-        # map_style = x,
-        layers=[layer],
-        initial_view_state= view_state,
-        tooltip = {
-            "html": "<b>City,State:</b> {City} </br> <b>Services:</b> {Count} </br> <b>F:</b> {FCName}",
-           "style": {
-                "backgroundColor": "steelblue",
-                "color": "white"
-           }
-        },
-    )
-    # st.pydeck_chart(r)
-    components.html(r.to_html(as_string=True), height=600)
+    # Search query for new data
+    search_query2 = st.text_input(label="Zip Code", value="")
+    if search_query2 != "":
+        df = df[df["Zip"].str.contains(search_query2)]
 
-data = load_data(state=queried_state,zip_code=queried_zip_code,city=queried_city,age=queried_age,male=queried_male,female=queried_female)
+    st.subheader("City Detail")
+    st.write(df)
 
-if st.checkbox('Display data ?'):
-    data
+Cities = "uscities.csv"
+SearchData = "US.txt"
+ShowCityDataframe(Cities, SearchData)
 
-mapF(data)
-