Map update

Former-commit-id: 2e9ad04139c6f154fb12ee54a66a703a3bb43e13
Former-commit-id: e63c943e2fa038179e19e9f6d2ad8a0c029aa193

main.py

@@ -7,6 +7,7 @@ import io
 from model_execute import preprocess_images, output_to_names
 from summarization import init_model_and_tokenizer, summarize
 from wikipedia_api import getWikipedia
+from mapbox_map import plot_map
 
 @st.cache_resource
 def load_recognition_model():
@@ -70,22 +71,24 @@ st.title("Распознавание достопримечательносте
 # Images input.
 images = load_images()
 
+summarize_checkbox = st.checkbox("Короткое описание")
 result = st.button('Распознать')
 
-if result:
+if images and result:
     # Get predictions
     names = predict_images(images, landmark_model)
     st.write(names)
 
     # Request descriptions and coordinates from Wikipedia.
     wiki_data = getWikipedia(names)
-
+    st.write("Загружены данные с википедии.")
     # Summarize descriptions for each landmark.
-    for landmark in wiki_data:
-        description = landmark['summary']
-        summarized = summarize(description, summarizer, tokenizer)
-        landmark['summarized'] = summarized
-
+    if summarize_checkbox:
+        for landmark in wiki_data:
+            description = landmark['summary']
+            summarized = summarize(description, summarizer, tokenizer)
+            landmark['summarized'] = summarized
     st.write(wiki_data)
 
     # Draw a map.
+    plot_map(wiki_data)

map.py

@@ -1,52 +0,0 @@
-import streamlit as st
-import pandas as pd
-import plotly.graph_objects as go
-
-# Загрузка данных
-data = pd.DataFrame({
-    'latitude': [37.7749, 34.0522],  # Пример координат для точек
-    'longitude': [-122.4194, -118.2437],
-    'name': ['San Francisco', 'Los Angeles']
-})
-
-# Заголовок приложения
-st.title('Приложение с картой MapBox')
-
-# Создание маркеров
-marker_size = 12  # Размер маркера
-
-fig = go.Figure()
-
-for i, row in data.iterrows():
-    fig.add_trace(go.Scattermapbox(
-        lat=[row['latitude']],
-        lon=[row['longitude']],
-        mode='markers',
-        marker=dict(
-            size=marker_size,
-            color='red',
-            sizemode='diameter',  # Размер маркера в диаметрах
-        ),
-        text=row['name']
-    ))
-
-# Вычисление границы области, содержащей все маркеры
-min_lat, max_lat = data['latitude'].min(), data['latitude'].max()
-min_lon, max_lon = data['longitude'].min(), data['longitude'].max()
-
-# Рассчитываем центр и масштаб для отображения всех маркеров
-center_lat = (min_lat + max_lat) / 2
-center_lon = (min_lon + max_lon) / 2
-zoom_level = 1  # Масштаб (может потребоваться настройка)
-
-# Настройка карты
-fig.update_layout(
-    mapbox_style="open-street-map",
-    mapbox=dict(
-        center=dict(lat=center_lat, lon=center_lon),
-        zoom=zoom_level,
-    ),
-)
-
-# Отображение карты в Streamlit
-st.plotly_chart(fig)

mapbox_map.py

@@ -0,0 +1,59 @@
+import streamlit as st
+import pandas as pd
+import plotly.graph_objects as go
+
+def preprocess_df(df):
+    """
+    Убрать строки без координат, преобразовать координаты в float.
+    """
+    no_coords = df['latitude'] == ''
+    df = df.drop(df[no_coords].index)
+    df['latitude'] = df['latitude'].astype('float')
+    df['longitude'] = df['longitude'].astype('float')
+    return df
+
+def plot_map(landmarks):
+    # Загрузка данных
+    data = pd.DataFrame(landmarks)
+    data = preprocess_df(data)
+    st.write(data)
+    # Создание маркеров
+    marker_size = 12  # Размер маркера
+
+    fig = go.Figure()
+
+    for i, row in data.iterrows():
+        if type(row['latitude']) is not float:
+            continue
+        fig.add_trace(go.Scattermapbox(
+            lat=[row['latitude']],
+            lon=[row['longitude']],
+            mode='markers',
+            marker=dict(
+                size=marker_size,
+                color='red',
+                sizemode='diameter',  # Размер маркера в диаметрах
+            ),
+            text=row['find']
+        ))
+
+    # Вычисление границы области, содержащей все маркеры
+    min_lat, max_lat = data['latitude'].min(), data['latitude'].max()
+    min_lon, max_lon = data['longitude'].min(), data['longitude'].max()
+
+    # Рассчитываем центр и масштаб для отображения всех маркеров
+    center_lat = (min_lat + max_lat) / 2
+    center_lon = (min_lon + max_lon) / 2
+    zoom_level = 12  # Масштаб (может потребоваться настройка)
+
+    # Настройка карты
+    fig.update_layout(
+        mapbox_style="open-street-map",
+        mapbox=dict(
+            center=dict(lat=center_lat, lon=center_lon),
+            zoom=zoom_level,
+        ),
+    )
+
+    # Отображение карты в Streamlit
+    st.plotly_chart(fig)

wikipedia_api.py

@@ -26,12 +26,16 @@ def getWikipedia(pList):
          longitude = find['data-lon']
     else:
       summary = 'Ничего не найдено'
-
+      
     res = {'find':i,
-             'summary':summary,
-             'latitude':latitude,
-             'longitude':longitude,
+             'summary': summary,
+             'latitude': latitude,
+             'longitude': longitude,
              }
     result.append(res)
 
   return result
+
+if __name__ == "__main__":
+
+  print(getWikipedia(["Новый Арбат"]))