Fix data loading (#8)

- fix (20d397f9ebe2c321b056cb885638073fe58f61be)


Co-authored-by: ABDENNACER BADAOUI <badaoui@users.noreply.huggingface.co>

data.py

@@ -519,13 +519,7 @@ class CIResults:
             logger.info("Loading distant data...")
             new_df, latest_update_msg = get_distant_data()
             self.latest_update_msg = latest_update_msg
-            self.available_dates = get_available_dates()
-            logger.info(f"Available dates: {len(self.available_dates)} dates")
-            if self.available_dates:
-                logger.info(f"Date range: {self.available_dates[-1]} to {self.available_dates[0]}")
-            else:
-                logger.warning("No available dates found")
-                self.available_dates = []
+            self.sample_data = False
         except Exception as e:
             error_msg = [
                 "Loading data failed:",
@@ -538,8 +532,26 @@ class CIResults:
             self.sample_data = True
             new_df, latest_update_msg = get_sample_data()
             self.latest_update_msg = latest_update_msg
-            # Generate fake dates for sample data historical functionality
-            self.available_dates = generate_fake_dates()
+        
+        # Try to get available dates 
+        try:
+            if not self.sample_data:
+                self.available_dates = get_available_dates()
+                logger.info(f"Available dates: {len(self.available_dates)} dates")
+                if self.available_dates:
+                    logger.info(f"Date range: {self.available_dates[-1]} to {self.available_dates[0]}")
+                else:
+                    logger.warning("No available dates found")
+                    self.available_dates = []
+            else:
+                # Generate fake dates for sample data historical functionality
+                self.available_dates = generate_fake_dates()
+        except Exception as e:
+            logger.warning(f"Failed to get available dates: {e}")
+            if self.sample_data:
+                self.available_dates = generate_fake_dates()
+            else:
+                self.available_dates = []
         
         # Update attributes
         self.df = new_df