Update app.py

app.py

@@ -149,12 +149,38 @@ def query_flight_data(geo_df, question):
     if not identifier:
         return "Please specify a flight identifier (callsign or ICAO code) in your question."
     
-    # Try to find the flight by callsign or icao
+    # Clean and normalize the identifier
+    identifier = identifier.strip().upper()
+    
+    # Try to find the flight by callsign or icao (case-insensitive)
     flight_data = None
-    if identifier in geo_df['callsign'].values:
-        flight_data = geo_df[geo_df['callsign'] == identifier]
-    elif identifier in geo_df['icao24'].values:
-        flight_data = geo_df[geo_df['icao24'] == identifier]
+    
+    # First try exact match
+    if identifier in geo_df['callsign'].str.upper().values:
+        flight_data = geo_df[geo_df['callsign'].str.upper() == identifier]
+    elif identifier in geo_df['icao24'].str.upper().values:
+        flight_data = geo_df[geo_df['icao24'].str.upper() == identifier]
+    
+    # If no exact match, try partial match
+    if flight_data is None or flight_data.empty:
+        # Try matching without spaces or special characters
+        clean_identifier = ''.join(filter(str.isalnum, identifier))
+        if not geo_df['callsign'].empty:
+            clean_callsigns = geo_df['callsign'].fillna('').apply(lambda x: ''.join(filter(str.isalnum, str(x).upper())))
+            matches = clean_callsigns == clean_identifier
+            if matches.any():
+                flight_data = geo_df[matches]
+    
+    # If still no match, try fuzzy matching
+    if flight_data is None or flight_data.empty:
+        try:
+            from difflib import get_close_matches
+            all_callsigns = geo_df['callsign'].fillna('').str.upper().unique()
+            close_matches = get_close_matches(identifier, all_callsigns, n=1, cutoff=0.8)
+            if close_matches:
+                flight_data = geo_df[geo_df['callsign'].str.upper() == close_matches[0]]
+        except:
+            pass
     
     if flight_data is None or flight_data.empty:
         return f"Could not find flight information for {identifier}. Please check the flight identifier and try again."