Update app.py

app.py

@@ -30,7 +30,27 @@ if all_region == False:
     region = st.selectbox(
          'Which region do you want to predict temparature ?',
          (df["Region"].unique()), index=0)
-    df.drop(df.loc[df['Location'] != region].index, inplace=True)
+    df.drop(df.loc[df['Region'] != region].index, inplace=True)
+
+if all_country == False:
+    region = 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)
+
+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 all_city == False:
+    region = 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)
+    
+    
 
 def mergeStateToCountry():
     df.loc[df['State'].notna(), 'Country'] = df['State']