app.py

import gradio as gr
import folium
from folium import plugins
import requests
import pandas as pd
from datetime import datetime
import time
import numpy as np
import plotly.graph_objects as go
from plotly.subplots import make_subplots

# OpenSky API URL
BASE_URL = "https://opensky-network.org/api"

def get_states(bounds=None):
    """Get current aircraft states from OpenSky Network"""
    try:
        response = requests.get(f"{BASE_URL}/states/all", 
                              params=bounds if bounds else {},
                              timeout=10)  # 타임아웃 추가
        if response.status_code == 200:
            return response.json()
        else:
            print(f"API Error: {response.status_code}")
            return None
    except Exception as e:
        print(f"Error fetching data: {e}")
        return None

def create_monitoring_dashboard(states):
    """Create monitoring dashboard using Plotly"""
    if not states:
        return go.Figure()  # 빈 figure 반환

    # 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 (simplified)
    fig.add_trace(
        go.Pie(labels=['Commercial', 'Private', 'Other'], 
               values=[60, 30, 10], name="Categories"),
        row=2, col=2
    )

    fig.update_layout(
        height=600,
        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"""
    # 기본 맵 생성
    m = folium.Map(
        location=[30, 0],
        zoom_start=3,
        tiles='CartoDB dark_matter'
    )

    # 데이터 가져오기
    bounds = {
        "world": None,
        "europe": {"lamin": 35.0, "lomin": -15.0, "lamax": 60.0, "lomax": 40.0},
        "north_america": {"lamin": 25.0, "lomin": -130.0, "lamax": 50.0, "lomax": -60.0},
        "asia": {"lamin": 10.0, "lomin": 60.0, "lamax": 50.0, "lomax": 150.0}
    }

    data = get_states(bounds.get(region))
    
    if not data or 'states' not in data or not data['states']:
        return (
            m._repr_html_(), 
            create_monitoring_dashboard([]),
            "No data available. Please try again later."
        )

    states = data['states']
    heat_data = []

    # Add aircraft markers
    for state in states:
        if state[6] and state[5]:  # latitude and longitude check
            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'
            
            heat_data.append([lat, lon, 1])
            
            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>
            """
            
            folium.Marker(
                location=[lat, lon],
                popup=folium.Popup(popup_content, max_width=300),
                icon=folium.DivIcon(
                    html=f'''
                        <div style="transform: rotate({state[10] if state[10] else 0}deg)">✈️</div>
                    ''',
                    icon_size=(20, 20)
                )
            ).add_to(m)

    # Add heatmap
    plugins.HeatMap(heat_data, radius=15).add_to(m)
    
    # Statistics
    total_aircraft = len(states)
    countries = len(set(state[2] for state in states if state[2]))
    avg_altitude = np.mean([state[7] for state in states if state[7]]) if states else 0
    
    stats = f"""
    📊 Real-time Statistics:
    • Total Aircraft: {total_aircraft}
    • Countries: {countries}
    • Average Altitude: {avg_altitude:.0f}m
    
    🔄 Last Updated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
    """

    return m._repr_html_(), create_monitoring_dashboard(states), stats

# Custom CSS
custom_css = """
.gradio-container {
    background: linear-gradient(135deg, #1a1a1a, #2d2d2d) !important;
}
.gr-button {
    background: linear-gradient(135deg, #4a90e2, #357abd) !important;
    border: none !important;
    color: white !important;
}
.gr-button:hover {
    background: linear-gradient(135deg, #357abd, #4a90e2) !important;
    transform: translateY(-2px);
    box-shadow: 0 5px 15px rgba(74, 144, 226, 0.4) !important;
}
"""

# Gradio interface
with gr.Blocks(css=custom_css) as demo:
    gr.HTML(
        """
        <h1 style="text-align: center; color: white;">🛩️ Global Aircraft Tracker</h1>
        <p style="text-align: center; color: #ccc;">Real-time tracking of aircraft worldwide</p>
        """
    )
    gr.HTML("""<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fimmunobiotech-opensky.hf.space">
               <img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fimmunobiotech-opensky.hf.space&countColor=%23263759" />
               </a>""")        
    
    with gr.Row():
        region_select = gr.Dropdown(
            choices=["world", "europe", "north_america", "asia"],
            value="world",
            label="Select Region"
        )
        refresh_btn = gr.Button("🔄 Refresh")
    
    map_html = gr.HTML()
    stats_text = gr.Textbox(label="Statistics", lines=6)
    dashboard_plot = gr.Plot(label="Monitoring Dashboard")
    
    def update_map(region):
        try:
            return create_map(region)
        except Exception as e:
            print(f"Error updating map: {e}")
            return (
                "<p>Map loading failed. Please try again.</p>",
                go.Figure(),
                f"Error: {str(e)}"
            )
    
    refresh_btn.click(
        fn=update_map,
        inputs=[region_select],
        outputs=[map_html, dashboard_plot, stats_text]
    )
    
    region_select.change(
        fn=update_map,
        inputs=[region_select],
        outputs=[map_html, dashboard_plot, stats_text]
    )

# Launch with specific configurations
demo.launch(
    show_error=True,
    server_name="0.0.0.0",
    server_port=7860,
    share=False
)