From 5e0cb57eee8d6131e876fca955b3cbaa3e827e6b Mon Sep 17 00:00:00 2001
From: Zeheng Li <imzehengl@gmail.com>
Date: Thu, 2 Nov 2023 02:04:45 -0600
Subject: [PATCH] clean up

---
 streamlit_app.py | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/streamlit_app.py b/streamlit_app.py
index 5b250da..d491689 100644
--- a/streamlit_app.py
+++ b/streamlit_app.py
@@ -60,7 +60,7 @@ def get_tree_coverage_sample(df):
 
 st.header("Questions")
 
-score, total = 0, 6
+score, ans = 0, []
 
 df_area = get_area_sample(df)
 ans_area = st.radio(
@@ -68,6 +68,7 @@ def get_tree_coverage_sample(df):
     df_area["name"],
     index=None,
 )
+ans.append(ans_area)
 if ans_area in df_area["name"][df_area["area"] == df_area["area"].max()].tolist():
     score += 1
 
@@ -77,6 +78,7 @@ def get_tree_coverage_sample(df):
     df_bikeways["name"],
     index=None,
 )
+ans.append(ans_bikeways)
 if (
     ans_bikeways
     in df_bikeways["name"][
@@ -91,6 +93,7 @@ def get_tree_coverage_sample(df):
     df_schools["name"],
     index=None,
 )
+ans.append(ans_schools)
 if (
     ans_schools
     in df_schools["name"][df_schools["schools"] == df_schools["schools"].max()].tolist()
@@ -103,6 +106,7 @@ def get_tree_coverage_sample(df):
     df_recreation_facilities["name"],
     index=None,
 )
+ans.append(ans_recreation_facilities)
 if (
     ans_recreation_facilities
     in df_recreation_facilities["name"][
@@ -118,6 +122,7 @@ def get_tree_coverage_sample(df):
     df_transit_stops["name"],
     index=None,
 )
+ans.append(ans_transit_stops)
 if (
     ans_transit_stops
     in df_transit_stops["name"][
@@ -132,6 +137,7 @@ def get_tree_coverage_sample(df):
     df_tree_coverage["name"],
     index=None,
 )
+ans.append(ans_tree_coverage)
 if (
     ans_tree_coverage
     in df_tree_coverage["name"][
@@ -140,18 +146,13 @@ def get_tree_coverage_sample(df):
 ):
     score += 1
 
+total = len(ans)
 if score == total:
     thumb = ":100:"
 else:
     thumb = ":thumbsup:" if score / total > 0.4 else ":thumbsdown:"
 
-if all(
-    [
-        ans_area,
-        ans_bikeways,
-        ans_schools,
-    ]
-):
+if all(ans):
     st.write(f"You score {score} out of {total} {thumb}")