Update app.py

app.py

@@ -35,6 +35,7 @@ os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
 
 app.config['DATA_FOLDER'] = UPLOAD_FOLDER
 os.makedirs(app.config['DATA_FOLDER'], exist_ok=True)
+os.makedirs("data", exist_ok=True)
 
 app.config['MODEL_FOLDER'] = UPLOAD_FOLDER
 os.makedirs(app.config['MODEL_FOLDER'], exist_ok=True)
@@ -163,6 +164,10 @@ def predict():
         
         # Process the DataFrame and generate predictions and classification analysis.
         df_pred, dx_class = process_dataframe(df)
+        if df_pred.empty:
+            print("Processed prediction DataFrame is empty. Check the input file and processing logic.", "error")
+            return redirect(request.url)
+
         
         # Save output files with a timestamp (you can also store in session if needed)
         current_date = pd.Timestamp.now().strftime("%Y-%m-%d")