Update app.py

app.py

@@ -6,320 +6,209 @@ import pandas as pd
 import io
 from PIL import Image
 import numpy as np
-import time
-from osmnx.errors import InsufficientResponseError, EmptyOverpassResponse  # 追加
+
+ox.settings.use_cache = True  # OSMnxのキャッシュを有効化
 
 def meters_to_degrees(meters, lat):
+    """
+    距離（メートル）を緯度・経度の変化量（度）に変換する。
+    """
     lat_deg = meters / 111320
     lon_deg = meters / (111320 * np.cos(np.radians(lat)))
     return lat_deg, lon_deg
 
-def create_artistic_map(
-    lat, lon, distance, dpi, width, height, bg_color, greenery_color, water_color,
-    building_colors, major_road_color, medium_road_color, minor_road_color,
-    display_greenery, display_water, display_buildings,
-    display_major_roads, display_medium_roads, display_minor_roads,
-    progress_bar, time_placeholder
-):
-    start_time = time.time()
-    current_step = 0
-    total_steps = 2  # 基本のステップ数
-
-    if display_greenery:
-        total_steps += 1
-    if display_water:
-        total_steps += 1
-    if display_buildings:
-        total_steps += 1
-    if display_major_roads or display_medium_roads or display_minor_roads:
-        total_steps += 1
+@st.cache_data
+def fetch_osm_data(point, distance):
+    """
+    指定した地点と距離に基づいて、OSMデータ（ノード、エッジ、フィーチャ）を取得する。
+    """
+    tags = {
+        'building': True,
+        'natural': ['water', 'wood'],
+        'landuse': ['grass', 'forest'],
+        'leisure': 'park'
+    }
+    # グラフデータの取得
+    G = ox.graph_from_point(point, dist=distance, network_type='all')
+    nodes, edges = ox.graph_to_gdfs(G)
+    # フィーチャデータの取得
+    features = ox.geometries_from_point(point, tags, dist=distance)
+    return nodes, edges, features
+
+def create_artistic_map(lat, lon, distance, dpi, width, height, colors):
+    """
+    指定されたパラメータに基づいて、カスタマイズ可能なアートマップ画像を作成する。
+    """
+    # カラー設定の取得
+    bg_color = colors['bg_color']
+    greenery_color = colors['greenery_color']
+    water_color = colors['water_color']
+    building_colors = colors['building_colors']
+    major_road_color = colors['major_road_color']
+    medium_road_color = colors['medium_road_color']
+    minor_road_color = colors['minor_road_color']
 
     # 中心点の設定
     point = (lat, lon)
-    
+
     # 距離をもとに緯度経度の幅と高さを計算
     lat_deg, lon_deg = meters_to_degrees(distance, lat)
-    
+
+    # OSMデータ取得
+    nodes, edges, features = fetch_osm_data(point, distance)
+
+    # 建物データ取得
+    buildings = features[features['building'].notnull()]
+
+    # 水域データ取得
+    water = features[features['natural'] == 'water']
+
+    # 緑地データ取得（公園、草地、森林など）
+    greenery = features[
+        (features['leisure'] == 'park') |
+        (features['landuse'].isin(['grass', 'forest'])) |
+        (features['natural'] == 'wood')
+    ]
+
     # 描画設定
     fig, ax = plt.subplots(figsize=(width, height))
-    
+
     # 背景色を設定
     fig.patch.set_facecolor(bg_color)
     ax.set_facecolor(bg_color)
-    
-    # 進捗を更新
-    current_step += 1
-    elapsed_time = time.time() - start_time
-    estimated_total_time = (elapsed_time / current_step) * total_steps
-    remaining_time = estimated_total_time - elapsed_time
-    progress_percent = int((current_step / total_steps) * 100)
-    progress_bar.progress(progress_percent)
-    time_placeholder.text(f"残り時間: 約{int(remaining_time // 60)}分{int(remaining_time % 60)}秒")
 
     # 緑地の描画
-    if display_greenery:
-        # 緑地データ取得
-        greenery_list = []
-        greenery_tags = [
-            {'leisure': 'park'},
-            {'leisure': 'garden'},
-            {'landuse': 'grass'},
-            {'landuse': 'forest'},
-            {'landuse': 'meadow'},
-            {'natural': 'wood'},
-            {'natural': 'scrub'},
-            {'natural': 'grassland'},
-            {'natural': 'heath'}
-        ]
-        for tags in greenery_tags:
-            try:
-                greenery_part = ox.features_from_point(point, tags=tags, dist=distance)
-                if not greenery_part.empty:
-                    greenery_list.append(greenery_part)
-            except (InsufficientResponseError, EmptyOverpassResponse):
-                continue
-            except Exception as e:
-                st.warning(f"緑地データ取得中にエラーが発生しました: {e}")
-                continue
-        if greenery_list:
-            greenery = pd.concat(greenery_list)
-            greenery.plot(ax=ax, facecolor=greenery_color, edgecolor='none')
-        else:
-            st.warning("指定された範囲に緑地が見つかりませんでした。")
+    if not greenery.empty:
+        greenery.plot(ax=ax, facecolor=greenery_color, edgecolor='none')
 
-        # 進捗を更新
-        current_step += 1
-        elapsed_time = time.time() - start_time
-        estimated_total_time = (elapsed_time / current_step) * total_steps
-        remaining_time = estimated_total_time - elapsed_time
-        progress_percent = int((current_step / total_steps) * 100)
-        progress_bar.progress(progress_percent)
-        time_placeholder.text(f"残り時間: 約{int(remaining_time // 60)}分{int(remaining_time % 60)}秒")
-    
     # 水域の描画
-    if display_water:
-        # 水域データ取得
-        water_list = []
-        water_tags = [
-            {'natural': 'water'},
-            {'waterway': 'riverbank'},
-            {'landuse': 'reservoir'},
-            {'landuse': 'basin'},
-            {'natural': 'wetland'}
-        ]
-        for tags in water_tags:
-            try:
-                water_part = ox.features_from_point(point, tags=tags, dist=distance)
-                if not water_part.empty:
-                    water_list.append(water_part)
-            except (InsufficientResponseError, EmptyOverpassResponse):
-                continue
-            except Exception as e:
-                st.warning(f"水域データ取得中にエラーが発生しました: {e}")
-                continue
-        if water_list:
-            water = pd.concat(water_list)
-            water.plot(ax=ax, facecolor=water_color, edgecolor='none')
-        else:
-            st.warning("指定された範囲に水域が見つかりませんでした。")
+    if not water.empty:
+        water.plot(ax=ax, facecolor=water_color, edgecolor='none')
 
-        # 進捗を更新
-        current_step += 1
-        elapsed_time = time.time() - start_time
-        estimated_total_time = (elapsed_time / current_step) * total_steps
-        remaining_time = estimated_total_time - elapsed_time
-        progress_percent = int((current_step / total_steps) * 100)
-        progress_bar.progress(progress_percent)
-        time_placeholder.text(f"残り時間: 約{int(remaining_time // 60)}分{int(remaining_time % 60)}秒")
-    
     # 建物の描画
-    if display_buildings:
-        # 建物データ取得
-        try:
-            buildings = ox.features_from_point(point, tags={'building': True}, dist=distance)
-            if not buildings.empty:
-                if 'building' in buildings.columns:
-                    building_types = buildings['building'].unique()
-                    for i, building_type in enumerate(building_types):
-                        building_subset = buildings[buildings['building'] == building_type]
-                        building_color = building_colors[i % len(building_colors)]
-                        building_subset.plot(ax=ax, facecolor=building_color, edgecolor='none')
-                else:
-                    # 'building'カラムがない場合、すべて同じ色で表示
-                    buildings.plot(ax=ax, facecolor=building_colors[0], edgecolor='none')
-            else:
-                st.warning("指定された範囲に建物が見つかりませんでした。")
-        except (InsufficientResponseError, EmptyOverpassResponse):
-            st.warning("指定された範囲に建物が見つかりませんでした。")
-        except Exception as e:
-            st.warning(f"建物データ取得中にエラーが発生しました: {e}")
+    if not buildings.empty:
+        building_types = buildings['building'].unique()
+        for i, building_type in enumerate(building_types):
+            building_subset = buildings[buildings['building'] == building_type]
+            building_color = building_colors[i % len(building_colors)]
+            building_subset.plot(ax=ax, facecolor=building_color, edgecolor='none')
 
-        # 進捗を更新
-        current_step += 1
-        elapsed_time = time.time() - start_time
-        estimated_total_time = (elapsed_time / current_step) * total_steps
-        remaining_time = estimated_total_time - elapsed_time
-        progress_percent = int((current_step / total_steps) * 100)
-        progress_bar.progress(progress_percent)
-        time_placeholder.text(f"残り時間: 約{int(remaining_time // 60)}分{int(remaining_time % 60)}秒")
-    
     # 道路の描画
-    if display_major_roads or display_medium_roads or display_minor_roads:
-        try:
-            # OSMデータ取得
-            G = ox.graph_from_point(point, dist=distance, network_type='all')
-            nodes, edges = ox.graph_to_gdfs(G)
-            if not edges.empty and 'highway' in edges.columns:
-                # 主要道路の描画
-                if display_major_roads:
-                    major_roads = edges[edges['highway'].isin(['motorway', 'trunk', 'primary'])]
-                    if not major_roads.empty:
-                        major_roads.plot(ax=ax, linewidth=2, edgecolor=major_road_color)
-                    else:
-                        st.warning("指定された範囲に主要道路が見つかりませんでした。")
-            
-                # 中程度の道路の描画
-                if display_medium_roads:
-                    medium_roads = edges[edges['highway'].isin(['secondary', 'tertiary'])]
-                    if not medium_roads.empty:
-                        medium_roads.plot(ax=ax, linewidth=1.5, edgecolor=medium_road_color)
-                    else:
-                        st.warning("指定された範囲に中程度の道路が見つかりませんでした。")
-            
-                # 小規模な道路の描画
-                if display_minor_roads:
-                    minor_roads = edges[~edges['highway'].isin(['motorway', 'trunk', 'primary', 'secondary', 'tertiary'])]
-                    if not minor_roads.empty:
-                        minor_roads.plot(ax=ax, linewidth=1, edgecolor=minor_road_color)
-                    else:
-                        st.warning("指定された範囲に小規模な道路が見つかりませんでした。")
-            else:
-                st.warning("指定された範囲に道路データが見つかりませんでした。")
-        except (InsufficientResponseError, EmptyOverpassResponse):
-            st.warning("指定された範囲に道路データが見つかりませんでした。")
-        except Exception as e:
-            st.warning(f"道路データ取得中にエラーが発生しました: {e}")
+    if not edges.empty and 'highway' in edges.columns:
+        # 主要道路の描画
+        major_roads = edges[edges['highway'].isin(['motorway', 'trunk', 'primary'])]
+        if not major_roads.empty:
+            major_roads.plot(ax=ax, linewidth=2, edgecolor=major_road_color)
+
+        # 中程度の道路の描画
+        medium_roads = edges[edges['highway'].isin(['secondary', 'tertiary'])]
+        if not medium_roads.empty:
+            medium_roads.plot(ax=ax, linewidth=1.5, edgecolor=medium_road_color)
+
+        # 小規模な道路の描画
+        minor_roads = edges[~edges['highway'].isin(['motorway', 'trunk', 'primary', 'secondary', 'tertiary'])]
+        if not minor_roads.empty:
+            minor_roads.plot(ax=ax, linewidth=1, edgecolor=minor_road_color)
 
-        # 進捗を更新
-        current_step += 1
-        elapsed_time = time.time() - start_time
-        estimated_total_time = (elapsed_time / current_step) * total_steps
-        remaining_time = estimated_total_time - elapsed_time
-        progress_percent = int((current_step / total_steps) * 100)
-        progress_bar.progress(progress_percent)
-        time_placeholder.text(f"残り時間: 約{int(remaining_time // 60)}分{int(remaining_time % 60)}秒")
-        
     # 軸をオフに設定
     ax.axis('off')
-        
+
     # 表示範囲を設定
     ax.set_xlim([lon - lon_deg, lon + lon_deg])
     ax.set_ylim([lat - lat_deg, lat + lat_deg])
 
-    # 最終処理の進捗を更新
-    current_step += 1
-    progress_percent = int((current_step / total_steps) * 100)
-    progress_bar.progress(progress_percent)
-    time_placeholder.text("最終処理中...")
-
-    # バッファに保存して、PIL画像として返す
+    # 画像をバッファに保存
     buf = io.BytesIO()
     plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0, dpi=dpi)
     plt.close()
     buf.seek(0)
-    
-    # 進捗を100%に更新
-    progress_bar.progress(100)
-    time_placeholder.text("完了しました！")
-
     return Image.open(buf)
 
 # Streamlit UI
 st.title("カラフルな地図が作成できます")
 
-# デフォルト値を設定
-default_lat = 35.533283
-default_lon = 139.642000
-default_distance = 1.0  # キロメートル
-
 # ユーザー入力
-lat = st.number_input("緯度を入力してください", value=default_lat, format="%.6f")
-lon = st.number_input("経度を入力してください", value=default_lon, format="%.6f")
-distance = st.number_input("距離を選択（km）", value=default_distance, format="%.2f")
+location = st.text_input("場所を入力してください（住所、都市名など）")
 
-# 画像設定
-dpi = st.number_input("画像のDPI（解像度）を選択", value=72, format="%d")
-width = st.slider("画像の幅（インチ）を選択", 5, 30, 20)
-height = st.slider("画像の高さ（インチ）を選択", 5, 30, 20)
-
-# 表示設定
-st.header("表示設定")
-
-# 背景色
-bg_color = st.color_picker("背景色を選択", "#FAF3E0")
-
-# 緑地の設定
-st.subheader("緑地の設定")
-display_greenery = st.checkbox("緑地を表示", value=True)
-if display_greenery:
-    greenery_color = st.color_picker("緑地の色を選択", "#80C341")
-else:
-    greenery_color = None
-
-# 水域の設定
-st.subheader("水域の設定")
-display_water = st.checkbox("水域を表示", value=True)
-if display_water:
-    water_color = st.color_picker("水域の色を選択", "#45A6FF")
-else:
-    water_color = None
-
-# 建物の設定
-st.subheader("建物の設定")
-display_buildings = st.checkbox("建物を表示", value=True)
-if display_buildings:
-    default_building_colors = ['#FF6F61', '#FFAB73', '#FFA07A', '#FFD700', '#F08080']
-    building_colors = [
-        st.color_picker(f"建物の色 {i+1}", default_color)
-        for i, default_color in enumerate(default_building_colors)
-    ]
-else:
-    building_colors = []
-
-# 道路の設定
-st.subheader("道路の設定")
-# 主要道路
-display_major_roads = st.checkbox("主要道路を表示", value=True)
-if display_major_roads:
-    major_road_color = st.color_picker("主要道路の色を選択", "#FF6347")
+default_lat = 35.533283
+default_lon = 139.642000
+default_distance = 1000
+
+if location:
+    try:
+        geocode_result = ox.geocode(location)
+        lat, lon = geocode_result[0], geocode_result[1]
+        st.write(f"緯度: {lat}, 経度: {lon}")
+    except Exception as e:
+        st.error("場所の取得に失敗しました。緯度と経度を直接入力してください。")
+        lat = st.number_input("緯度を入力してください", value=default_lat, format="%.6f")
+        lon = st.number_input("経度を入力してください", value=default_lon, format="%.6f")
 else:
-    major_road_color = None
-
-# 中程度の道路
-display_medium_roads = st.checkbox("中程度の道路を表示", value=True)
-if display_medium_roads:
-    medium_road_color = st.color_picker("中程度の道路の色を選択", "#FF4500")
+    lat = st.number_input("緯度を入力してください", value=default_lat, format="%.6f")
+    lon = st.number_input("経度を入力してください", value=default_lon, format="%.6f")
+
+distance = st.slider("距離を選択（メートル）", 100, 5000, default_distance)
+
+# カラーテーマの設定
+color_themes = {
+    'テーマ1': {
+        'bg_color': '#FAF3E0',
+        'greenery_color': '#80C341',
+        'water_color': '#45A6FF',
+        'building_colors': ['#FF6F61', '#FFAB73', '#FFA07A', '#FFD700', '#F08080'],
+        'major_road_color': '#FF6347',
+        'medium_road_color': '#FF4500',
+        'minor_road_color': '#D2691E'
+    },
+    'テーマ2': {
+        'bg_color': '#FFFFFF',
+        'greenery_color': '#00FF00',
+        'water_color': '#0000FF',
+        'building_colors': ['#A52A2A', '#8B0000', '#B22222', '#FF0000', '#FF6347'],
+        'major_road_color': '#000000',
+        'medium_road_color': '#2F4F4F',
+        'minor_road_color': '#696969'
+    },
+    'カスタム': None
+}
+
+theme_name = st.selectbox("カラーテーマを選択", list(color_themes.keys()))
+
+if theme_name != 'カスタム':
+    colors = color_themes[theme_name]
 else:
-    medium_road_color = None
+    # 色設定
+    with st.expander("カスタムカラーパレット"):
+        bg_color = st.color_picker("背景色を選択", "#FAF3E0")
+        greenery_color = st.color_picker("緑地の色を選択", "#80C341")
+        water_color = st.color_picker("水域の色を選択", "#45A6FF")
+        building_colors = [
+            st.color_picker(f"建物の色 {i+1}", default_color)
+            for i, default_color in enumerate(['#FF6F61', '#FFAB73', '#FFA07A', '#FFD700', '#F08080'])
+        ]
+        major_road_color = st.color_picker("主要道路の色を選択", "#FF6347")
+        medium_road_color = st.color_picker("中程度の道路の色を選択", "#FF4500")
+        minor_road_color = st.color_picker("小規模な道路の色を選択", "#D2691E")
+    colors = {
+        'bg_color': bg_color,
+        'greenery_color': greenery_color,
+        'water_color': water_color,
+        'building_colors': building_colors,
+        'major_road_color': major_road_color,
+        'medium_road_color': medium_road_color,
+        'minor_road_color': minor_road_color
+    }
 
-# 小規模な道路
-display_minor_roads = st.checkbox("小規模な道路を表示", value=True)
-if display_minor_roads:
-    minor_road_color = st.color_picker("小規模な道路の色を選択", "#D2691E")
-else:
-    minor_road_color = None
+# 画像設定
+with st.expander("画像設定（オプション）"):
+    dpi = st.slider("画像のDPI（解像度）を選択", 72, 600, 300)
+    width = st.slider("画像の幅（インチ）を選択", 5, 30, 20)
+    height = st.slider("画像の高さ（インチ）を選択", 5, 30, 20)
 
 # マップ生成ボタン
 if st.button("マップを生成"):
-    # 進捗バーと残り時間のプレースホルダーを初期化
-    progress_bar = st.progress(0)
-    time_placeholder = st.empty()
-    map_image = create_artistic_map(
-        lat, lon, distance*1000, dpi, width, height,
-        bg_color, greenery_color, water_color, building_colors,
-        major_road_color, medium_road_color, minor_road_color,
-        display_greenery, display_water, display_buildings,
-        display_major_roads, display_medium_roads, display_minor_roads,
-        progress_bar, time_placeholder
-    )
-    st.image(map_image, caption="生成されたアートマップ", use_column_width=True)
+    with st.spinner('マップを生成中...'):
+        map_image = create_artistic_map(
+            lat, lon, distance, dpi, width, height, colors
+        )
+        st.image(map_image, caption="生成されたアートマップ", use_column_width=True)