Update app.py

app.py

@@ -298,6 +298,48 @@ Provide a clear and concise answer focusing on the specific information requeste
         response += f"- {key.replace('_', ' ').title()}: {value}\n"
     return response
 
+@st.cache_data(ttl=60)  # Cache for 1 minute
+def get_traffic_gdf(lat_min, lat_max, lon_min, lon_max, local_time_zone, loc, flight_info):
+    # Get cached flight data
+    json_dict = fetch_flight_data(lat_min, lat_max, lon_min, lon_max)
+    
+    if not json_dict or not json_dict.get('states'):
+        st.warning("No flight data available for the selected area.")
+        return None
+
+    try:
+        unix_timestamp = int(json_dict["time"])
+        local_timezone = pytz.timezone(local_time_zone)
+        local_time = datetime.fromtimestamp(unix_timestamp, local_timezone).strftime('%Y-%m-%d %H:%M:%S')
+        
+        # Optimize DataFrame creation
+        state_df = pd.DataFrame(json_dict["states"], columns=columns)
+        state_df['time'] = local_time
+        
+        # Create GeoDataFrame more efficiently
+        gdf = gpd.GeoDataFrame(
+            state_df,
+            geometry=gpd.points_from_xy(state_df.longitude, state_df.latitude),
+            crs="EPSG:4326"
+        )
+        
+        # Display information
+        st.title("Live Flight Tracker")
+        st.subheader('Flight Details', divider='rainbow')
+        st.write('Location: {0}'.format(loc))
+        st.write('Current Local Time: {0}-{1}:'.format(local_time, local_time_zone))
+        st.write("Minimum_latitude is {0} and Maximum_latitude is {1}".format(lat_min, lat_max))
+        st.write("Minimum_longitude is {0} and Maximum_longitude is {1}".format(lon_min, lon_max))
+        st.write('Number of Visible Flights: {}'.format(len(json_dict['states'])))
+        st.write('Plotting the flight: {}'.format(flight_info))
+        st.subheader('Map Visualization', divider='rainbow')
+        st.write('****Click ":orange[Update Map]" Button to Refresh the Map****')
+        return gdf
+        
+    except Exception as e:
+        st.error(f"Error processing flight data: {str(e)}")
+        return None
+
 def flight_tracking(flight_view_level, country, local_time_zone, flight_info, airport, color):
     # Get cached location data
     loc = get_location(country)
@@ -327,49 +369,8 @@ def flight_tracking(flight_view_level, country, local_time_zone, flight_info, ai
                                             (airport_country_loc['Longitude'] >= lon_min) &
                             (airport_country_loc['Longitude'] <= lon_max)]
     
-    def get_traffic_gdf():
-        # Get cached flight data
-        json_dict = fetch_flight_data(lat_min, lat_max, lon_min, lon_max)
-        
-        if not json_dict or not json_dict.get('states'):
-            st.warning("No flight data available for the selected area.")
-            return None
-
-        try:
-        unix_timestamp = int(json_dict["time"])
-            local_timezone = pytz.timezone(local_time_zone)
-        local_time = datetime.fromtimestamp(unix_timestamp, local_timezone).strftime('%Y-%m-%d %H:%M:%S')
-            
-            # Optimize DataFrame creation
-            state_df = pd.DataFrame(json_dict["states"], columns=columns)
-            state_df['time'] = local_time
-            
-            # Create GeoDataFrame more efficiently
-        gdf = gpd.GeoDataFrame(
-                state_df,
-                geometry=gpd.points_from_xy(state_df.longitude, state_df.latitude),
-                crs="EPSG:4326"
-            )
-            
-            # Display information
-        st.title("Live Flight Tracker")
-        st.subheader('Flight Details', divider='rainbow')
-        st.write('Location: {0}'.format(loc))
-        st.write('Current Local Time: {0}-{1}:'.format(local_time, local_time_zone))
-        st.write("Minimum_latitude is {0} and Maximum_latitude is {1}".format(lat_min, lat_max))
-        st.write("Minimum_longitude is {0} and Maximum_longitude is {1}".format(lon_min, lon_max))
-        st.write('Number of Visible Flights: {}'.format(len(json_dict['states'])))
-        st.write('Plotting the flight: {}'.format(flight_info))
-        st.subheader('Map Visualization', divider='rainbow')
-        st.write('****Click ":orange[Update Map]" Button to Refresh the Map****')
-        return gdf
-
-        except Exception as e:
-            st.error(f"Error processing flight data: {str(e)}")
-            return None
-
     # Get traffic data
-    geo_df = get_traffic_gdf()
+    geo_df = get_traffic_gdf(lat_min, lat_max, lon_min, lon_max, local_time_zone, loc, flight_info)
     if geo_df is None:
         return