Update app.py

app.py

@@ -21,14 +21,12 @@ merge = st.checkbox("Merge US country with State ?")
 
 all_region = st.checkbox("Take all Region ?", value=True)
 all_country = st.checkbox("Take all Country ?", value=True)
-
-if merge == False:
-    all_state = st.checkbox("Take all State ?", value=True)
-
 all_city = st.checkbox("Take all City ?", value=True)
 
 df = pd.read_csv("city_temperature.csv")
 
+country = "all"
+
 if all_region == False:
     region = st.selectbox(
          'Which region do you want to predict temparature ?',
@@ -36,22 +34,25 @@ if all_region == False:
     df.drop(df.loc[df['Region'] != region].index, inplace=True)
 
 if all_country == False:
-    region = st.selectbox(
+    country = st.selectbox(
          'Which country do you want to predict temparature ?',
          (df["Country"].unique()), index=0)
-    df.drop(df.loc[df['Country'] != region].index, inplace=True)
+    df.drop(df.loc[df['Country'] != country].index, inplace=True)
 
-if merge == False and all_state == False:
-    state = st.selectbox(
-         'Which state do you want to predict temparature ?',
-         (df["State"].unique()), index=0)
-    df.drop(df.loc[df['State'] != region].index, inplace=True)
+if merge == False and country == "US":
+    all_state = st.checkbox("Take all State ?", value=True)
+    
+    if all_state == False:
+        state = st.selectbox(
+             'Which state do you want to predict temparature ?',
+             (df["State"].unique()), index=0)
+        df.drop(df.loc[df['State'] != state].index, inplace=True)
 
 if all_city == False:
-    region = st.selectbox(
+    city = st.selectbox(
          'Which city do you want to predict temparature ?',
          (df["City"].unique()), index=0)
-    df.drop(df.loc[df['City'] != region].index, inplace=True)
+    df.drop(df.loc[df['City'] != city].index, inplace=True)