Update app.py

app.py

@@ -1,12 +1,43 @@
 import gradio as gr
 from transformers import pipeline
 import geopy
+import plotly.graph_objects as go
 
 
 binary_classifier = pipeline("text-classification", model="GEODE/bert-base-multilingual-cased-binary-classifier-edda-coords")
 ner_pipeline = pipeline("token-classification", model="GEODE/camembert-base-edda-span-classification", aggregation_strategy="simple")
 generator = pipeline("text2text-generation", model="GEODE/mt5-small-coords-norm")
 
+
+def create_map(lat, long):
+    fig = go.Figure(go.Scattermapbox(
+            #customdata=text_list,
+            lat=[lat],
+            lon=[long],
+            mode='markers',
+            marker=go.scattermapbox.Marker(
+                size=6
+            ),
+            #hoverinfo="text",
+            #hovertemplate='<b>Name</b>: %{customdata[0]}<br><b>Price</b>: $%{customdata[1]}'
+        ))
+
+    fig.update_layout(
+        mapbox_style="open-street-map",
+        hovermode='closest',
+        mapbox=dict(
+            bearing=0,
+            center=go.layout.mapbox.Center(
+                lat=lat,
+                lon=long
+            ),
+            pitch=0,
+            zoom=9
+        ),
+    )
+
+    return fig
+
 def dms_to_dd(dms):
     try:
         point = geopy.Point(dms)
@@ -53,7 +84,7 @@ def norm_coordinates(text):
     
     coords = dms_to_dd(result_text)
     
-    return result_text, coords
+    return result_text, create_map(coords[0], coords[1]) if coords else None
     
 
 examples = [
@@ -88,8 +119,9 @@ with gr.Blocks() as demo:
             run_btn.click(fn=extract_coordinates, inputs=inp, outputs=out_text)
 
             out_text = gr.Textbox(label="Extract and normalize DMS coordinates (fine-tuned mT5)")
-            out_map = gr.Map(label="Coordinates on Map")
-            run_btn.click(fn=norm_coordinates, inputs=inp, outputs=[out_text, out_map])
+            #out_map = gr.Map(label="Coordinates on Map")
+            map = gr.Plot(label="Coordinates on Map")
+            run_btn.click(fn=norm_coordinates, inputs=inp, outputs=[out_text, map])
 
     with gr.Row():