Update app.py

app.py

@@ -2,20 +2,22 @@ import gradio as gr
 import folium
 from folium import plugins
 import requests
+from requests.auth import HTTPBasicAuth
 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
+import os
 
-# OpenSky API 인증 정보
-USERNAME = "seawolf2357"
-PASSWORD = "Time2175!@"
+# OpenSky API 인증 정보를 환경 변수에서 가져오기
+USERNAME = os.getenv("ID_AUTH")
+PASSWORD = os.getenv("PW_AUTH")
 
 class OpenSkyApi:
     def __init__(self, username=None, password=None):
-        self.auth = (username, password) if username and password else None
+        self.auth = HTTPBasicAuth(username, password) if username and password else None
         self.base_url = "https://opensky-network.org/api"
     
     def get_states(self, time_secs=0, icao24=None, bbox=None):
@@ -32,15 +34,25 @@ class OpenSkyApi:
                 "lomax": bbox[3]
             })
         
+        headers = {
+            'User-Agent': 'Mozilla/5.0',
+            'Accept': 'application/json'
+        }
+        
         try:
             response = requests.get(
                 f"{self.base_url}/states/all",
                 params=params,
                 auth=self.auth,
+                headers=headers,
                 timeout=15
             )
+            
             if response.status_code == 200:
                 return StateVector(response.json())
+            elif response.status_code == 401:
+                print("Authentication failed. Please check your environment variables ID_AUTH and PW_AUTH.")
+                return None
             else:
                 print(f"Error {response.status_code}: {response.text}")
                 return None
@@ -71,10 +83,10 @@ class State:
         self.velocity = state_array[9]
         self.true_track = state_array[10]
         self.vertical_rate = state_array[11]
-        self.sensors = state_array[12]
-        self.baro_altitude = state_array[13]
-        self.squawk = state_array[14]
-        self.spi = state_array[15]
+        self.sensors = state_array[12] if len(state_array) > 12 else None
+        self.baro_altitude = state_array[13] if len(state_array) > 13 else None
+        self.squawk = state_array[14] if len(state_array) > 14 else None
+        self.spi = state_array[15] if len(state_array) > 15 else None
         self.position_source = state_array[16] if len(state_array) > 16 else None
 
 # OpenSky API 인스턴스 생성
@@ -113,8 +125,6 @@ def get_states(bounds=None, max_retries=3):
             return None
     return None
 
-
-
 def create_monitoring_dashboard(states):
     """Create monitoring dashboard using Plotly"""
     if not states:
@@ -270,8 +280,6 @@ def create_map(region="world"):
 
     return m._repr_html_(), create_monitoring_dashboard(states), stats
 
-
-
 # Custom CSS
 custom_css = """
 .gradio-container {
@@ -336,12 +344,15 @@ with gr.Blocks(css=custom_css) as demo:
         outputs=[map_html, dashboard_plot, stats_text]
     )
 
+# 환경 변수 확인
+if not USERNAME or not PASSWORD:
+    print("Warning: Environment variables ID_AUTH and/or PW_AUTH are not set.")
+    print("The application will run with anonymous access, which has lower rate limits.")
+
 # Launch with specific configurations
 demo.launch(
     show_error=True,
     server_name="0.0.0.0",
     server_port=7860,
     share=False
-)
-
-
+)