Update app.py

app.py

@@ -368,7 +368,7 @@ def flight_tracking(flight_view_level, country, local_time_zone, flight_info, ai
     
     # Filter airports for the selected country and within the view bounds
     airport_country_loc = airport_locations[
-        (airport_locations['Country'].str.contains(str(loc[0]), case=False, na=False)) & 
+        (airport_locations['Country'].str.contains(str(country), case=False)) & 
         (airport_locations['Latitude'] >= lat_min) &
         (airport_locations['Latitude'] <= lat_max) & 
         (airport_locations['Longitude'] >= lon_min) &
@@ -376,7 +376,7 @@ def flight_tracking(flight_view_level, country, local_time_zone, flight_info, ai
     ]
     
     # Get traffic data
-    geo_df = get_traffic_gdf(lat_min, lat_max, lon_min, lon_max, local_time_zone, loc, flight_info)
+    geo_df = get_traffic_gdf(lat_min, lat_max, lon_min, lon_max, local_time_zone, country, flight_info)
     if geo_df is None:
         return
     
@@ -519,11 +519,13 @@ def flight_tracking(flight_view_level, country, local_time_zone, flight_info, ai
                     icon=folium.Icon(icon='plane', prefix='fa', color='blue'),
                     popup=folium.Popup(
                         f"""
-                        <div style="font-size: 14px; font-family: Arial, sans-serif;">
-                            <b>{row['Name']}</b><br>
-                            IATA: {row['IATA/FAA']}<br>
-                            City: {row['City']}<br>
-                            Country: {row['Country']}
+                        <div style='font-family: Arial, sans-serif;'>
+                            <h4 style='margin: 0;'>{row['Name']}</h4>
+                            <p style='margin: 5px 0;'>
+                                <b>IATA:</b> {row['IATA/FAA']}<br>
+                                <b>City:</b> {row['City']}<br>
+                                <b>Country:</b> {row['Country']}
+                            </p>
                         </div>
                         """,
                         max_width=300