Update app.py

app.py

@@ -12,17 +12,19 @@ import xgboost as xgb
 
 # Setting up the page configuration for Streamlit App
 st.set_page_config(
-    page_title="Taxi",    
+    page_title="Taxi",
     # layout="wide",
     initial_sidebar_state="expanded"
 )
 
+
 # Load the XGBoost model
 #@st.cache_data()
 def get_model():
     model = pickle.load(open("models/model_xgb.pkl", "rb"))
     return model
 
+
 # Function to make prediction using the model and input data
 def make_prediction(data):
     model = get_model()
@@ -38,21 +40,17 @@ def make_prediction(data):
     return model.predict(data_matrix)
 
 
+# Get coordinates from address
 def get_coordinates(address):
-    # Создание экземпляра геокодера
     geolocator = Nominatim(user_agent="my_app")
-
-    # Получение координат по адресу
     location = geolocator.geocode(address)
-
-    # Вывод широты и долготы
     return (location.longitude, location.latitude)
 
 
 def show_map(lon_from, lat_from, lon_to, lat_to):
     # Creating a map
     fig = go.Figure(go.Scattermapbox(
-        mode = "markers",        
+        mode = "markers",
         marker = {'size': 15, 'color': 'red'}
     ))
 
@@ -63,11 +61,11 @@ def show_map(lon_from, lat_from, lon_to, lat_to):
         lat = [lat_from, lat_to],
         marker = go.scattermapbox.Marker(
             size=25,
-            color='red'              
+            color='red'
         )
     ))
 
-    # Добавление линии между точками
+    # Adding a line
     fig.add_trace(go.Scattermapbox(
         mode = "lines",
         lon = [lon_from, lon_to],
@@ -75,19 +73,19 @@ def show_map(lon_from, lat_from, lon_to, lat_to):
         line = dict(width=2, color='green')
     ))
 
-    # Настройка отображения карты
+    # Configuring the display of a map
     fig.update_layout(
         mapbox = {
-            'style': "open-street-map",  # Стиль карты
-            'center': {'lon': (lon_from + lon_to) / 2, 'lat': (lat_from + lat_to) / 2},  # Центр карты
-            'zoom': 9,  # Уровень масштабирования карты            
+            'style': "open-street-map",
+            'center': {'lon': (lon_from + lon_to) / 2, 'lat': (lat_from + lat_to) / 2},
+            'zoom': 9,    
         },
         showlegend = False,
-        height = 600,  # Изменение высоты карты
-        width = 1200  # Изменение ширины карты
+        height = 600,
+        width = 1200
     )
-    
-    # Отображение карты
+
+    # Display the map
     return fig
 
 
@@ -123,18 +121,18 @@ def get_haversine_distance(lat1, lng1, lat2, lng2):
 
 
 # User input features
-def user_input_features(lon_from, lat_from, lon_to, lat_to, passenger_count):    
-    current_time = datetime.now()    
+def user_input_features(lon_from, lat_from, lon_to, lat_to, passenger_count):
+    current_time = datetime.now() 
     pickup_hour= current_time.hour
     today = datetime.today()
     pickup_holiday = 1 if today in holidays.USA() else 0
     total_distance,  total_travel_time, number_of_steps = get_total_distance(lon_from, lat_from, lon_to, lat_to)
-    haversine_distance = get_haversine_distance(lat_from, lon_from, lat_to, lon_to)    
+    haversine_distance = get_haversine_distance(lat_from, lon_from, lat_to, lon_to)
     weekday_number = current_time.weekday()
 
     data = {'vendor_id': 1,
             'passenger_count': passenger_count,
-            'pickup_longitude': lon_from,            
+            'pickup_longitude': lon_from,       
             'pickup_latitude': lat_from,
             'dropoff_longitude': lon_to,
             'dropoff_latitude': lat_to,
@@ -168,19 +166,19 @@ def min_max_scaler(data):
     data_scaled = scaler.transform(data)
     return data_scaled
 
-# Main function
-def main(): 
 
+# Main function
+def main():
     if 'btn_predict' not in st.session_state:
-        st.session_state['btn_predict'] = False 
-    
+        st.session_state['btn_predict'] = False
+
     # Sidebar
     st.sidebar.markdown(''' # New York City Taxi Trip Duration''')
     st.sidebar.image("img/taxi_img.png")
     address_from = st.sidebar.text_input("Откуда:", value="New York, 11 Wall Street")
-    address_to = st.sidebar.text_input("Куда:", value="New York, 740 Park Avenue")    
+    address_to = st.sidebar.text_input("Куда:", value="New York, 740 Park Avenue")
     passenger_count = st.sidebar.slider("Количество пассажиров", 1, 4, 1)
-    
+
     st.session_state['btn_predict'] = st.sidebar.button('Start')
 
     if st.session_state['btn_predict']:
@@ -190,7 +188,7 @@ def main():
         user_data = user_input_features(lon_from, lat_from, lon_to, lat_to, passenger_count)
         # st.write(user_data)
         data_scaled = min_max_scaler(user_data)
-        trip_duration = np.exp(make_prediction(data_scaled)) - 1     
+        trip_duration = np.exp(make_prediction(data_scaled)) - 1
         trip_duration = round(float(trip_duration) / 60)
         st.markdown(f"""
                         <div style='background-color: lightgreen; padding: 10px;'>
@@ -198,6 +196,7 @@ def main():
                         </div>
                     """, unsafe_allow_html=True)
 
+
 # Running the main function
 if __name__ == "__main__":
     main()