Update app.py

app.py

@@ -130,37 +130,38 @@ def load_ibtracs_data():
         with open(CACHE_FILE, 'rb') as f:
             return pickle.load(f)
     
-    # Define both paths we might need
+    # Define the path for the all basins file
     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, 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
+    try:
+        # 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, all_basins_path)
+            # Load the full ALL dataset
+            ibtracs = tracks.TrackDataset(source='ibtracs', ibtracs_url=all_basins_path)
+        
+        # Save to cache
+        with open(CACHE_FILE, 'wb') as f:
+            pickle.dump(ibtracs, f)
+        
+        return ibtracs
+        
+    except Exception as e:
+        print(f"Error loading IBTrACS data: {e}")
+        # As a fallback, try loading just the default data that comes with tropycal
+        print("Attempting to load default dataset...")
+        ibtracs = tracks.TrackDataset(basin='all')
+        with open(CACHE_FILE, 'wb') as f:
+            pickle.dump(ibtracs, f)
+        return ibtracs
 
 def convert_typhoondata(input_file, output_file):
     with open(input_file, 'r') as infile: