Update app.py

app.py

@@ -1,81 +1,124 @@
 import gradio as gr
+import folium
+from folium import plugins
 import requests
-import json
-from PIL import Image
+import pandas as pd
+from datetime import datetime
+import time
+import branca.colormap as cm
+import numpy as np
 import io
 
-# API 기본 URL
-BASE_URL = "https://collectionapi.metmuseum.org/public/collection/v1"
+# OpenSky API URL
+BASE_URL = "https://opensky-network.org/api"
 
-def get_departments():
-    """Get all departments from the Met API"""
-    response = requests.get(f"{BASE_URL}/departments")
-    return response.json()['departments']
+def get_states(bounds=None):
+    """Get current aircraft states from OpenSky Network"""
+    params = {}
+    if bounds:
+        params.update({
+            'lamin': bounds[0],
+            'lomin': bounds[1],
+            'lamax': bounds[2],
+            'lomax': bounds[3]
+        })
+    
+    try:
+        response = requests.get(f"{BASE_URL}/states/all", params=params)
+        data = response.json()
+        return data
+    except Exception as e:
+        print(f"Error fetching data: {e}")
+        return None
 
-def search_artworks(query, department_id=None, is_highlight=False, has_images=True, 
-                   is_on_view=False, medium=None, geo_location=None):
-    """Search artworks with various filters"""
-    search_url = f"{BASE_URL}/search"
-    params = {
-        'q': query,
-        'hasImages': has_images,
-        'isHighlight': is_highlight,
-        'isOnView': is_on_view
+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],
+        "north_america": [25.0, -130.0, 50.0, -60.0],
+        "asia": [10.0, 60.0, 50.0, 150.0]
     }
     
-    if department_id:
-        params['departmentId'] = department_id
-    if medium:
-        params['medium'] = medium
-    if geo_location:
-        params['geoLocation'] = geo_location
-        
-    try:
-        # Get object IDs from search
-        response = requests.get(search_url, params=params)
-        results = response.json()
-        
-        if not results.get('objectIDs'):
-            return [], "No results found."
+    # Get aircraft data
+    data = get_states(bounds.get(region))
+    
+    if not data or 'states' not in data:
+        return 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
+            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'
             
-        # Limit to first 12 objects for performance
-        object_ids = results['objectIDs'][:12]
-        
-        images = []
-        captions = []
-        
-        # Get details for each object
-        for object_id in object_ids:
-            object_url = f"{BASE_URL}/objects/{object_id}"
-            object_response = requests.get(object_url)
-            artwork = object_response.json()
+            # Add to heatmap data
+            heat_data.append([lat, lon, 1])
             
-            if artwork.get('primaryImage'):
-                try:
-                    img_response = requests.get(artwork['primaryImage'], timeout=10)
-                    img = Image.open(io.BytesIO(img_response.content))
-                    
-                    if img.mode in ('RGBA', 'LA') or (img.mode == 'P' and 'transparency' in img.info):
-                        img = img.convert('RGB')
-                    
-                    artwork_info = f"""
-                    Title: {artwork.get('title', 'Unknown')}
-                    Artist: {artwork.get('artistDisplayName', 'Unknown')}
-                    Date: {artwork.get('objectDate', 'Unknown')}
-                    Medium: {artwork.get('medium', 'Unknown')}
-                    Department: {artwork.get('department', 'Unknown')}
-                    """
-                    
-                    images.append(img)
-                    captions.append(artwork_info)
-                except Exception as e:
-                    print(f"Error processing image for object {object_id}: {e}")
-                    continue
-        
-        return images, "\n\n".join(captions)
-        
-    except Exception as e:
-        return [], f"An error occurred: {str(e)}"
+            # Create popup content
+            popup_content = f"""
+            <div style="font-family: Arial; width: 200px;">
+                <h4 style="color: #4a90e2;">Flight Information</h4>
+                <p><b>Callsign:</b> {callsign}</p>
+                <p><b>Altitude:</b> {altitude}m</p>
+                <p><b>Velocity:</b> {velocity}m/s</p>
+                <p><b>Origin:</b> {state[2]}</p>
+            </div>
+            """
+            
+            # Add marker
+            folium.Marker(
+                location=[lat, lon],
+                popup=folium.Popup(popup_content, max_width=300),
+                icon=folium.DivIcon(
+                    html=f'''
+                        <div style="transform: rotate({state[10]}deg)">✈️</div>
+                    ''',
+                    icon_size=(20, 20)
+                )
+            ).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 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
+    
+    stats = f"""
+    📊 Statistics:
+    • Total Aircraft: {total_aircraft}
+    • Countries: {countries}
+    • Average Altitude: {avg_altitude:.0f}m
+    """
+    
+    return html_string, stats
 
 # Custom CSS
 custom_css = """
@@ -84,118 +127,59 @@ custom_css = """
     color: #ffffff !important;
 }
 .gr-button {
-    background: linear-gradient(135deg, #8e2de2, #4a00e0) !important;
+    background: linear-gradient(135deg, #4a90e2, #357abd) !important;
     border: none !important;
     color: white !important;
-    font-weight: bold !important;
-    padding: 10px 20px !important;
-    font-size: 1.1em !important;
-    box-shadow: 0 4px 15px rgba(0,0,0,0.2) !important;
 }
 .gr-button:hover {
+    background: linear-gradient(135deg, #357abd, #4a90e2) !important;
     transform: translateY(-2px);
-    box-shadow: 0 6px 20px rgba(0,0,0,0.3) !important;
-    transition: all 0.3s ease;
-}
-.gr-input, .gr-select {
-    border: 2px solid #4a00e0 !important;
-    background: rgba(255, 255, 255, 0.1) !important;
-    color: white !important;
-    font-size: 1.1em !important;
-    border-radius: 8px !important;
-}
-.gr-gallery {
-    background: rgba(0, 0, 0, 0.3) !important;
-    border-radius: 15px !important;
-    padding: 20px !important;
-    min-height: 800px !important;
-    box-shadow: 0 8px 32px rgba(0,0,0,0.3) !important;
+    box-shadow: 0 5px 15px rgba(74, 144, 226, 0.4) !important;
 }
 .title-text {
     text-align: center !important;
     color: #ffffff !important;
-    font-size: 2.8em !important;
-    margin-bottom: 0.3em !important;
+    font-size: 2.5em !important;
+    margin-bottom: 0.5em !important;
     text-shadow: 2px 2px 4px rgba(0,0,0,0.5) !important;
-    font-weight: bold !important;
-}
-.subtitle-text {
-    text-align: center !important;
-    color: #cccccc !important;
-    font-size: 1.3em !important;
-    margin-bottom: 2em !important;
-    font-style: italic !important;
-    text-shadow: 1px 1px 2px rgba(0,0,0,0.3) !important;
 }
 """
 
-# Get departments for dropdown
-departments = get_departments()
-department_choices = {dept['displayName']: dept['departmentId'] for dept in departments}
-
 # Gradio interface
 with gr.Blocks(css=custom_css) as demo:
     gr.HTML(
         """
-        <div class="title-text">🎨 The Met Art Explorer</div>
-        <div class="subtitle-text">Explore over 5,000 years of art from The Metropolitan Museum of Art's collection</div>
+        <div class="title-text">🛩️ Global Aircraft Tracker</div>
         """
     )
     
     with gr.Row():
-        with gr.Column(scale=3):
-            search_input = gr.Textbox(
-                label="Search artwork",
-                placeholder="Enter keywords (e.g., sunflowers, portrait, landscape...)"
-            )
-        with gr.Column(scale=2):
-            department_dropdown = gr.Dropdown(
-                choices=list(department_choices.keys()),
-                label="Department",
-                value=None
-            )
+        region_select = gr.Dropdown(
+            choices=["world", "europe", "north_america", "asia"],
+            value="world",
+            label="Select Region"
+        )
+        refresh_btn = gr.Button("🔄 Refresh")
     
-    with gr.Row():
-        with gr.Column():
-            highlight_checkbox = gr.Checkbox(label="Show Highlights Only", value=False)
-            on_view_checkbox = gr.Checkbox(label="Currently On View", value=False)
-        with gr.Column():
-            medium_input = gr.Textbox(
-                label="Medium",
-                placeholder="e.g., Paintings, Sculpture, Ceramics"
-            )
-            location_input = gr.Textbox(
-                label="Geographic Location",
-                placeholder="e.g., France, Japan, Egypt"
-            )
-    
-    search_btn = gr.Button("🔍 Search Collection", variant="primary")
-    
-    gallery = gr.Gallery(
-        label="Search Results",
-        show_label=True,
-        elem_id="gallery",
-        columns=3,
-        rows=4,
-        height="800px",
-        object_fit="contain"
-    )
+    map_html = gr.HTML()
+    stats_text = gr.Textbox(label="Statistics", lines=4)
     
-    info = gr.Textbox(
-        label="Artwork Details",
-        lines=10,
-        show_label=True
-    )
+    def update_map(region):
+        return create_map(region)
     
-    def handle_search(query, dept, highlight, on_view, medium, location):
-        dept_id = department_choices.get(dept) if dept else None
-        return search_artworks(query, dept_id, highlight, True, on_view, medium, location)
+    refresh_btn.click(
+        fn=update_map,
+        inputs=[region_select],
+        outputs=[map_html, stats_text]
+    )
     
-    search_btn.click(
-        fn=handle_search,
-        inputs=[search_input, department_dropdown, highlight_checkbox, 
-                on_view_checkbox, medium_input, location_input],
-        outputs=[gallery, info]
+    region_select.change(
+        fn=update_map,
+        inputs=[region_select],
+        outputs=[map_html, stats_text]
     )
+    
+    # Initial map load
+    map_html, stats_text = create_map("world")
 
 demo.launch()