Update app.py

app.py

@@ -8,10 +8,27 @@ import time
 import branca.colormap as cm
 import numpy as np
 import io
+from PIL import Image
+import plotly.graph_objects as go
+from plotly.subplots import make_subplots
 
 # OpenSky API URL
 BASE_URL = "https://opensky-network.org/api"
 
+# Aircraft photos API (예시 - 실제 구현시에는 적절한 API로 대체 필요)
+AIRCRAFT_PHOTOS_API = "https://api.planespotters.net/pub/photos/hex/{icao24}"
+
+def get_aircraft_photo(icao24):
+    """Get aircraft photo from Planespotters API"""
+    try:
+        response = requests.get(AIRCRAFT_PHOTOS_API.format(icao24=icao24))
+        data = response.json()
+        if data.get('photos'):
+            return data['photos'][0]['thumbnail_large']['src']
+    except:
+        # 기본 항공기 이미지 URL 반환
+        return "https://example.com/default-aircraft.jpg"
+
 def get_states(bounds=None):
     """Get current aircraft states from OpenSky Network"""
     params = {}
@@ -31,9 +48,60 @@ def get_states(bounds=None):
         print(f"Error fetching data: {e}")
         return None
 
+def create_monitoring_dashboard(data):
+    """Create monitoring dashboard using Plotly"""
+    if not data or 'states' not in data:
+        return None
+
+    states = data['states']
+    
+    # Create subplots
+    fig = make_subplots(
+        rows=2, cols=2,
+        subplot_titles=('Altitude Distribution', 'Speed Distribution', 
+                       'Aircraft by Country', 'Aircraft Categories')
+    )
+
+    # Altitude distribution
+    altitudes = [state[7] for state in states if state[7]]
+    fig.add_trace(
+        go.Histogram(x=altitudes, name="Altitude"),
+        row=1, col=1
+    )
+
+    # Speed distribution
+    speeds = [state[9] for state in states if state[9]]
+    fig.add_trace(
+        go.Histogram(x=speeds, name="Speed"),
+        row=1, col=2
+    )
+
+    # Aircraft by country
+    countries = pd.Series([state[2] for state in states if state[2]]).value_counts()
+    fig.add_trace(
+        go.Bar(x=countries.index[:10], y=countries.values[:10], name="Countries"),
+        row=2, col=1
+    )
+
+    # Aircraft categories
+    categories = pd.Series([state[17] for state in states if state[17]]).value_counts()
+    fig.add_trace(
+        go.Pie(labels=categories.index, values=categories.values, name="Categories"),
+        row=2, col=2
+    )
+
+    fig.update_layout(
+        height=800,
+        showlegend=False,
+        template="plotly_dark",
+        paper_bgcolor='rgba(0,0,0,0)',
+        plot_bgcolor='rgba(0,0,0,0)'
+    )
+
+    return fig
+
 def create_map(region="world"):
     """Create aircraft tracking map"""
-    # Default bounds for different regions
     bounds = {
         "world": None,
         "europe": [35.0, -15.0, 60.0, 40.0],
@@ -41,51 +109,46 @@ def create_map(region="world"):
         "asia": [10.0, 60.0, 50.0, 150.0]
     }
     
-    # Get aircraft data
     data = get_states(bounds.get(region))
     
     if not data or 'states' not in data:
-        return None, "Failed to fetch aircraft data"
+        return None, None, "Failed to fetch aircraft data"
     
-    # Create base map
     m = folium.Map(
         location=[30, 0],
         zoom_start=3,
         tiles='CartoDB dark_matter'
     )
     
-    # Create aircraft icon
-    aircraft_icon = folium.CustomIcon(
-        'https://raw.githubusercontent.com/pointhi/leaflet-color-markers/master/img/marker-icon-2x-blue.png',
-        icon_size=(25, 41)
-    )
-    
-    # Create heatmap data
     heat_data = []
     
     # Add aircraft markers
     for state in data['states']:
-        if state[6] and state[5]:  # If latitude and longitude exist
+        if state[6] and state[5]:
             lat, lon = state[6], state[5]
             callsign = state[1] if state[1] else 'N/A'
             altitude = state[7] if state[7] else 'N/A'
             velocity = state[9] if state[9] else 'N/A'
+            icao24 = state[0]
             
-            # Add to heatmap data
             heat_data.append([lat, lon, 1])
             
-            # Create popup content
+            # Get aircraft photo
+            photo_url = get_aircraft_photo(icao24)
+            
             popup_content = f"""
-            <div style="font-family: Arial; width: 200px;">
+            <div style="font-family: Arial; width: 300px;">
                 <h4 style="color: #4a90e2;">Flight Information</h4>
+                <img src="{photo_url}" style="width: 100%; max-height: 200px; object-fit: cover; margin-bottom: 10px;">
                 <p><b>Callsign:</b> {callsign}</p>
+                <p><b>ICAO24:</b> {icao24}</p>
                 <p><b>Altitude:</b> {altitude}m</p>
                 <p><b>Velocity:</b> {velocity}m/s</p>
                 <p><b>Origin:</b> {state[2]}</p>
+                <p><b>Status:</b> {'On Ground' if state[8] else 'In Air'}</p>
             </div>
             """
             
-            # Add marker
             folium.Marker(
                 location=[lat, lon],
                 popup=folium.Popup(popup_content, max_width=300),
@@ -97,28 +160,31 @@ def create_map(region="world"):
                 )
             ).add_to(m)
     
-    # Add heatmap layer
     plugins.HeatMap(heat_data, radius=15).add_to(m)
-    
-    # Add layer control
     folium.LayerControl().add_to(m)
     
-    # Save map to HTML string
-    html_string = m._repr_html_()
+    # Create monitoring dashboard
+    dashboard = create_monitoring_dashboard(data)
     
     # Create statistics
     total_aircraft = len(data['states'])
     countries = len(set(state[2] for state in data['states'] if state[2]))
     avg_altitude = np.mean([state[7] for state in data['states'] if state[7]]) if data['states'] else 0
+    in_air = sum(1 for state in data['states'] if not state[8])
+    on_ground = sum(1 for state in data['states'] if state[8])
     
     stats = f"""
-    📊 Statistics:
+    📊 Real-time Statistics:
     • Total Aircraft: {total_aircraft}
+    • Aircraft in Air: {in_air}
+    • Aircraft on Ground: {on_ground}
     • Countries: {countries}
     • Average Altitude: {avg_altitude:.0f}m
+    
+    🔄 Last Updated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
     """
     
-    return html_string, stats
+    return m._repr_html_(), dashboard, stats
 
 # Custom CSS
 custom_css = """
@@ -143,6 +209,11 @@ custom_css = """
     margin-bottom: 0.5em !important;
     text-shadow: 2px 2px 4px rgba(0,0,0,0.5) !important;
 }
+.dashboard {
+    background: rgba(0, 0, 0, 0.3) !important;
+    border-radius: 10px !important;
+    padding: 20px !important;
+}
 """
 
 # Gradio interface
@@ -154,15 +225,24 @@ with gr.Blocks(css=custom_css) as demo:
     )
     
     with gr.Row():
-        region_select = gr.Dropdown(
-            choices=["world", "europe", "north_america", "asia"],
-            value="world",
-            label="Select Region"
-        )
-        refresh_btn = gr.Button("🔄 Refresh")
+        with gr.Column(scale=2):
+            region_select = gr.Dropdown(
+                choices=["world", "europe", "north_america", "asia"],
+                value="world",
+                label="Select Region"
+            )
+        with gr.Column(scale=1):
+            refresh_btn = gr.Button("🔄 Refresh")
+            auto_refresh = gr.Checkbox(label="Auto Refresh (30s)", value=False)
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            map_html = gr.HTML()
+        with gr.Column(scale=1):
+            stats_text = gr.Textbox(label="Statistics", lines=8)
     
-    map_html = gr.HTML()
-    stats_text = gr.Textbox(label="Statistics", lines=4)
+    with gr.Row():
+        dashboard_plot = gr.Plot(label="Monitoring Dashboard")
     
     def update_map(region):
         return create_map(region)
@@ -170,16 +250,25 @@ with gr.Blocks(css=custom_css) as demo:
     refresh_btn.click(
         fn=update_map,
         inputs=[region_select],
-        outputs=[map_html, stats_text]
+        outputs=[map_html, dashboard_plot, stats_text]
     )
     
     region_select.change(
         fn=update_map,
         inputs=[region_select],
-        outputs=[map_html, stats_text]
+        outputs=[map_html, dashboard_plot, stats_text]
     )
     
+    # Auto refresh
+    if auto_refresh:
+        demo.load(
+            fn=update_map,
+            inputs=[region_select],
+            outputs=[map_html, dashboard_plot, stats_text],
+            every=30
+        )
+    
     # Initial map load
-    map_html, stats_text = create_map("world")
+    map_html, dashboard_plot, stats_text = create_map("world")
 
 demo.launch()