updated plotly

app.py

@@ -153,8 +153,8 @@ fig.update_layout(height=1000, width=1000, template="plotly")
 
 # load data ready to plot for local testing
 germany = pd.read_csv(MAP_PATH)
-germany["lat"] = germany["lat"].astype(float)
-germany["lon"] = germany["lon"].astype(float)
+germany["lat"] = pd.to_numeric(germany["lat"])
+germany["lon"] = pd.to_numeric(germany["lon"])
 # germany.drop(columns=["lat", "lon"], inplace=True)
 
 # # or generate it directly in this script
@@ -164,15 +164,13 @@ germany["lon"] = germany["lon"].astype(float)
 # germany.to_csv(MAP_PATH_WITH_COORD, index=False)
 
 # # germany need columns with lat and lon as well as hover data
-fig_map = px.scatter_mapbox(
+fig_map = px.scatter_geo(
     germany,
     lat="lat",
     lon="lon",
     hover_name="ORG",
     custom_data=["Count"],
-    # color_discrete_map=["magenta"],
-    zoom=5,
-    height=700,
+    scope="europe",
 )
 # Custom hover template
 fig_map.update_traces(
@@ -183,7 +181,23 @@ fig_map.update_traces(
         ]
     )
 )
-fig_map.update_layout(mapbox_style="carto-positron")
+fig_map.update_layout(
+    geo=dict(
+        showland=True,
+        landcolor="LightGray",
+        showocean=True,
+        oceancolor="LightBlue",
+        # showcountries=True,
+        # countrycolor="Gray",
+        showsubunits=True,
+        # subunitcolor="Gray",
+        fitbounds="locations",  # Fit the map bounds to the locations
+        lataxis=dict(range=[47, 55]),  # Approximate latitude range for Germany
+        lonaxis=dict(range=[5, 16]),  # Approximate longitude range for Germany
+    ),
+    mapbox_style="carto-positron",
+    height=700,
+)
 
 tensors = {}
 with safe_open("corpus_embeddings.pt", framework="pt", device="cpu") as f: