Upload 2 files

app.py

@@ -0,0 +1,214 @@
+import os
+import re
+import gradio as gr
+from geopy.geocoders import Nominatim
+import osmnx as ox
+import folium
+import branca
+import shutil
+
+# --- Initial settings ---
+geolocator = Nominatim(user_agent="gradio_osm_app")
+ox.settings.log_console = False
+
+# Download directory
+DOWNLOAD_DIR = os.path.join(os.getcwd(), "download")
+os.makedirs(DOWNLOAD_DIR, exist_ok=True)
+
+def slugify(name):
+    return re.sub(r"[^0-9A-Za-z]+", "_", name).strip("_")
+
+def ensure_saved(gdf, slug, layer):
+    if gdf.empty:
+        return
+    filename = f"{slug}_{layer}.gpkg"
+    path = os.path.join(DOWNLOAD_DIR, filename)
+    if os.path.exists(path):
+        os.remove(path)
+    gdf.to_file(path, driver="GPKG")
+
+def make_legend(selected_layers):
+    color_map = {
+        "Highways": "yellow",
+        "Buildings": "#ff802a",
+        "School": "blue",
+        "Fire Station": "red",
+        "Hospital": "green",
+        "Police Station": "lightblue",
+        "Restaurants": "orange",
+        "Hotels": "purple",
+        "Monuments": "lightgreen"
+    }
+    html = '''
+    <div style="
+      position: fixed; bottom: 50px; left: 10px;
+      background-color: rgba(0,0,0,0.7); color: white;
+      padding: 10px; font-size:14px; border-radius:5px; z-index:9999;
+    ">
+      <b>Legend</b><br>
+    '''
+    for layer in selected_layers:
+        color = color_map.get(layer, "gray")
+        html += f'''
+        <i style="background: {color}; width:10px;height:10px;display:inline-block;margin-right:5px;"></i>{layer}<br>
+        '''
+    html += '</div>'
+    return branca.element.Element(html)
+
+style_funcs = {
+    "Highways":       lambda f: {"color": "yellow", "weight": 0.5},
+    "Buildings":      lambda f: {"fill": True, "fillColor": "#ff802a", "color": "#ff802a", "weight": 0.5, "fillOpacity": 0.6},
+    "School":         lambda f: {"fill": True, "fillColor": "blue", "color": "blue", "weight": 0.5, "fillOpacity": 0.6},
+    "Fire Station":   lambda f: {"fill": True, "fillColor": "red", "color": "red", "weight": 0.5, "fillOpacity": 0.6},
+    "Hospital":       lambda f: {"fill": True, "fillColor": "green", "color": "green", "weight": 0.5, "fillOpacity": 0.6},
+    "Police Station": lambda f: {"fill": True, "fillColor": "lightblue", "color": "lightblue", "weight": 0.5, "fillOpacity": 0.6},
+    "Restaurants":    lambda f: {"fill": True, "fillColor": "orange", "color": "orange", "weight": 0.5, "fillOpacity": 0.6},
+    "Hotels":         lambda f: {"fill": True, "fillColor": "purple", "color": "purple", "weight": 0.5, "fillOpacity": 0.6},
+    "Monuments":      lambda f: {"fill": True, "fillColor": "lightgreen", "color": "lightgreen", "weight": 0.5, "fillOpacity": 0.6},
+}
+
+def map_with_layers(place_name, cb_highways, cb_buildings, cb_school,
+                    cb_fire, cb_hospital, cb_police, cb_rest, cb_hotels, cb_monuments):
+    if not place_name.strip():
+        yield None, "❌ Please enter a valid place name."
+        return
+    slug = slugify(place_name)
+
+    yield None, "🔄 Geocoding..."
+    try:
+        gdf_place = ox.geocode_to_gdf(place_name)
+        poly = gdf_place.geometry.iloc[0]
+        lat, lon = poly.centroid.y, poly.centroid.x
+    except Exception as e:
+        yield None, f"❌ Geocoding error: {e}"
+        return
+
+    layers = {}
+    if cb_highways:
+        yield None, "🔄 Downloading Highways..."
+        G = ox.graph_from_polygon(poly, network_type="all", simplify=True)
+        gdf = ox.graph_to_gdfs(G, nodes=False, edges=True)
+        layers['Highways'] = gdf
+        ensure_saved(gdf, slug, 'highways')
+
+    tag_map = {
+        'Buildings': {'building': True}, 'School': {'amenity': 'school'},
+        'Fire Station': {'amenity': 'fire_station'}, 'Hospital': {'amenity': 'hospital'},
+        'Police Station': {'amenity': 'police'}, 'Restaurants': {'amenity': 'restaurant'},
+        'Hotels': {'tourism': 'hotel'}, 'Monuments': {'historic': 'monument'}
+    }
+    flags = [cb_buildings, cb_school, cb_fire, cb_hospital,
+             cb_police, cb_rest, cb_hotels, cb_monuments]
+    for (name, tags), flag in zip(tag_map.items(), flags):
+        if flag:
+            yield None, f"🔄 Downloading {name}..."
+            gdf2 = ox.features_from_polygon(poly, tags)
+            gdf2 = gdf2[gdf2.geometry.type.isin(['Polygon', 'MultiPolygon'])]
+            layers[name] = gdf2
+            ensure_saved(gdf2, slug, name.replace(' ', '').lower())
+
+    yield None, "🔄 Rendering map..."
+    m = folium.Map([lat, lon], zoom_start=13, tiles='CartoDB Dark_Matter')
+    for name, gdf in layers.items():
+        if gdf.empty:
+            continue
+        has_name = 'name' in gdf.columns and gdf['name'].notna().any()
+        cols = ['geometry', 'name'] if has_name else ['geometry']
+        gj = folium.GeoJson(gdf[cols], name=name, style_function=style_funcs[name])
+        if has_name:
+            gj.add_child(folium.GeoJsonPopup(fields=['name'], labels=False))
+        gj.add_to(m)
+
+    folium.LayerControl(collapsed=False).add_to(m)
+    m.get_root().html.add_child(make_legend(list(layers.keys())))
+
+    html_path = os.path.join(DOWNLOAD_DIR, f"{slug}_map.html")
+    m.save(html_path)
+    yield m._repr_html_(), '✅ Map ready!'
+
+def download_data(place_name, cb_highways, cb_buildings, cb_school,
+                  cb_fire, cb_hospital, cb_police, cb_rest, cb_hotels, cb_monuments):
+    slug = slugify(place_name)
+    files = []
+    for layer, flag in zip(
+        ['highways', 'buildings', 'school', 'fire_station', 'hospital',
+         'police_station', 'restaurants', 'hotels', 'monuments'],
+        [cb_highways, cb_buildings, cb_school, cb_fire, cb_hospital,
+         cb_police, cb_rest, cb_hotels, cb_monuments]
+    ):
+        if flag:
+            path = os.path.join(DOWNLOAD_DIR, f"{slug}_{layer}.gpkg")
+            if os.path.exists(path):
+                files.append(path)
+    html_path = os.path.join(DOWNLOAD_DIR, f"{slug}_map.html")
+    if os.path.exists(html_path):
+        files.append(html_path)
+    if not files:
+        raise gr.Error('No layer to download.')
+
+    # Limpa a pasta após retorno
+    result = list(files)  # cópia da lista
+    shutil.rmtree(DOWNLOAD_DIR)
+    os.makedirs(DOWNLOAD_DIR, exist_ok=True)
+    return result
+
+# --- Layout com abas ---
+with gr.Blocks(title="Geoeasy View") as demo:
+    gr.HTML("""
+    <style>
+      #logo { float: left; }
+      #app-title { display: inline-block; vertical-align: middle; margin: 0 0 0 1em; }
+      .header-wrapper { overflow: auto; margin-bottom: 1em; }
+    </style>
+    """)
+    with gr.Tabs():
+        with gr.TabItem('Map'):
+            gr.HTML("""
+            <div class="header-wrapper">
+              <h1 id="app-title">OSM View</h1>
+            </div>
+            """)
+            inp = gr.Textbox(label='Place', placeholder='e.g.: Porto, Portugal')
+            with gr.Row():
+                cb_highways  = gr.Checkbox(label='Highways', value=True)
+                cb_buildings = gr.Checkbox(label='Buildings', value=True)
+                cb_school    = gr.Checkbox(label='School')
+                cb_fire      = gr.Checkbox(label='Fire Station')
+                cb_hospital  = gr.Checkbox(label='Hospital')
+            with gr.Row():
+                cb_police    = gr.Checkbox(label='Police Station')
+                cb_rest      = gr.Checkbox(label='Restaurants')
+                cb_hotels    = gr.Checkbox(label='Hotels')
+                cb_monuments = gr.Checkbox(label='Monuments')
+            btn_map = gr.Button('Show Map')
+            btn_dl  = gr.Button('Download Files')
+            out_map, out_status = gr.HTML(), gr.Textbox(interactive=False)
+            out_files = gr.File(file_count='multiple')
+
+            inputs = [inp, cb_highways, cb_buildings, cb_school, cb_fire,
+                      cb_hospital, cb_police, cb_rest, cb_hotels, cb_monuments]
+            btn_map.click(map_with_layers, inputs=inputs, outputs=[out_map, out_status])
+            btn_dl.click(download_data,    inputs=inputs, outputs=out_files)
+
+        with gr.TabItem('Taginfo'):
+            gr.Markdown('''
+**Most used keys (Taginfo)**
+
+| Position | Key        | Approximate usage¹         |
+|----------|------------|----------------------------|
+| 1        | `highway`  | ~58 million geometries     |
+| 2        | `name`     | ~55 million geometries     |
+| 3        | `source`   | ~52 million geometries     |
+| 4        | `building` | ~48 million geometries     |
+| 5        | `landuse`  | ~34 million geometries     |
+| 6        | `natural`  | ~20 million geometries     |
+| 7        | `waterway` | ~18 million geometries     |
+| 8        | `amenity`  | ~16 million geometries     |
+| 9        | `place`    | ~12 million geometries     |
+| 10       | `power`    | ~14 million geometries     |
+
+¹ Approximate values from daily Taginfo reports.
+''')
+
+if __name__ == '__main__':
+    demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+gradio
+osmnx
+geopy
+folium
+branca