Update app.py

app.py

@@ -79,14 +79,24 @@ if (should_train_model=='1'): #train model
     
     
     # Check if the result is a non-empty DataFrame
-    file_train_df_temp = fetch_and_update_training_data(file_path_train)
-
-    if file_train_df_temp is not None and not file_train_df_temp.empty:
-        file_train_df = file_train_df_temp
-        file_test_df = pd.read_csv(file_path_test)
-    else:
+    try:
+        # Try fetching from the API
+        file_train_df_temp = fetch_and_update_training_data(file_path_train)
+    
+        if file_train_df_temp is not None and not file_train_df_temp.empty:
+            file_train_df = file_train_df_temp
+            file_test_df = pd.read_csv(file_path_test)
+            print("API fetch data.")
+        else:
+            raise ValueError("Empty or None DataFrame received from API.")
+
+    except Exception as e:
+        # On any error, fallback to local CSV
+        print(f"API fetch failed ({e}) — using local CSV.")
         file_train_df = pd.read_csv(file_path_train)
         file_test_df = pd.read_csv(file_path_test)
+
+
     
     #file_train_df = fetch_and_update_training_data(file_path_train)
     #file_test_df = pd.read_csv(file_path_test)