Update app.py

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 
 def get_market_submarket(df, google_ola):
     filtered_row = df[df['google_ola'] == google_ola].iloc[0]
-    return filtered_row['market_costar'], filtered_row['submarket_costar']
+    return filtered_row['prediction'], filtered_row['building_sf'], filtered_row['market_costar'], filtered_row['submarket_costar']
 
 def main():
     st.set_page_config(initial_sidebar_state="collapsed")
@@ -21,12 +21,15 @@ def main():
     
     if 'user_select_value' not in st.session_state:
         st.session_state['user_select_value'] = ""
-    if 'year_buit' not in st.session_state:
-        st.session_state['year_buit'] = 1990
+    if 'property_lsf' not in st.session_state:
+        st.session_state['property_lsf'] = 0
     if 'execution_dates' not in st.session_state:
         st.session_state['execution_dates'] = 2014
     if 'LSF' not in st.session_state:
         st.session_state['LSF'] = 75000
+    if 'prediction' not in st.session_state:
+        st.session_state['prediction'] = 0
+    
     
     st.title("Search")
     df_properties = pd.read_csv("market_underwriting_pred_short_atlanta.csv", encoding='utf-8')
@@ -58,12 +61,15 @@ def main():
         if selected_option == "":
             st.write(":red[Please choose the property address]")
         else:
+            
             st.session_state['user_select_value'] = selected_option
-            st.session_state['year_buit'] = 1990
+            
             st.session_state['execution_dates'] = 2014
-            market_val, submarket_val = get_market_submarket(df_properties, selected_option)
+            prediction, property_lsf, market_val, submarket_val = get_market_submarket(df_properties, selected_option)
             st.session_state['market_val'] = market_val
             st.session_state['submarket_val'] = submarket_val
+            st.session_state['property_lsf'] = property_lsf
+            st.session_state['prediction'] = prediction
 
             st.switch_page("pages/market_rent_estimation.py")