Update app.py

app.py

@@ -33,7 +33,7 @@ DATA_PATH = args.data_path
 
 ONI_DATA_PATH = os.path.join(DATA_PATH, 'oni_data.csv')
 TYPHOON_DATA_PATH = os.path.join(DATA_PATH, 'processed_typhoon_data.csv')
-LOCAL_iBtrace_PATH = os.path.join(DATA_PATH, 'ibtracs.WP.list.v04r01.csv')
+LOCAL_iBtrace_PATH = os.path.join(DATA_PATH, 'ibtracs.ALL.list.v04r01.csv') 
 iBtrace_uri = 'https://www.ncei.noaa.gov/data/international-best-track-archive-for-climate-stewardship-ibtracs/v04r01/access/csv/ibtracs.ALL.list.v04r01.csv'
 CACHE_FILE = 'ibtracs_cache.pkl'
 CACHE_EXPIRY_DAYS = 1
@@ -129,19 +129,37 @@ def load_ibtracs_data():
     if os.path.exists(CACHE_FILE) and (datetime.now() - datetime.fromtimestamp(os.path.getmtime(CACHE_FILE))).days < CACHE_EXPIRY_DAYS:
         with open(CACHE_FILE, 'rb') as f:
             return pickle.load(f)
-    if os.path.exists(LOCAL_iBtrace_PATH):
-        # Remove the basin='west_pacific' parameter to load all basins
-        ibtracs = tracks.TrackDataset(source='ibtracs', ibtracs_url=LOCAL_iBtrace_PATH)
+    
+    # Define both paths we might need
+    all_basins_path = os.path.join(DATA_PATH, 'ibtracs.ALL.list.v04r01.csv')
+    
+    # Try to load all basins file first
+    if os.path.exists(all_basins_path):
+        print("Loading ALL basins file...")
+        ibtracs = tracks.TrackDataset(source='ibtracs', ibtracs_url=all_basins_path)
     else:
+        print("Downloading ALL basins file...")
         response = requests.get(iBtrace_uri)
         response.raise_for_status()
         with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.csv') as temp_file:
             temp_file.write(response.text)
-            shutil.move(temp_file.name, LOCAL_iBtrace_PATH)
-        # Remove the basin='west_pacific' parameter here as well
-        ibtracs = tracks.TrackDataset(source='ibtracs', ibtracs_url=LOCAL_iBtrace_PATH)
+            shutil.move(temp_file.name, all_basins_path)
+        # Make sure we load the full ALL dataset
+        ibtracs = tracks.TrackDataset(source='ibtracs', ibtracs_url=all_basins_path)
+    
+    # Check if we have all basins
+    available_basins = set()
+    for storm_id in ibtracs.keys():
+        if len(storm_id) >= 2:
+            basin_code = storm_id[:2]  # First two characters are usually basin code
+            available_basins.add(basin_code)
+    
+    print(f"Available basin codes: {available_basins}")
+    
+    # Save to cache
     with open(CACHE_FILE, 'wb') as f:
         pickle.dump(ibtracs, f)
+    
     return ibtracs
 
 def convert_typhoondata(input_file, output_file):