Update app.py

app.py

@@ -84,15 +84,32 @@ def plot_feature_importance(importance):
     plt.close()
     return buf
 
-# Main analysis function
 def analyze_file(file):
     try:
         # Step 1: Read file
         df = read_file(file)
 
+        # Check if the dataframe is empty
+        if df.empty:
+            return (
+                "The uploaded file is empty or has no valid data.",
+                None,
+                None,
+                None,
+            )
+
         # Step 2: Clean data
         df_cleaned = clean_data(df)
 
+        # Check if the cleaned dataframe is still empty
+        if df_cleaned.empty:
+            return (
+                "The dataset contains no valid data after cleaning.",
+                None,
+                None,
+                None,
+            )
+
         # Step 3: Generate summary statistics
         summary = generate_summary(df_cleaned)
 
@@ -110,8 +127,22 @@ def analyze_file(file):
             importance.head(10),  # Top 10 important features
             importance_plot_buf,
         )
+    except ValueError as ve:
+        # Handle file format issues or parsing errors
+        return (
+            f"ValueError: {str(ve)}",
+            None,
+            None,
+            None,
+        )
     except Exception as e:
-        return str(e)
+        # Catch any other unforeseen issues
+        return (
+            f"An unexpected error occurred: {str(e)}",
+            None,
+            None,
+            None,
+        )
 
 # Gradio Interface
 def gradio_interface():
@@ -139,4 +170,4 @@ def gradio_interface():
 
 # Launch the Gradio interface
 interface = gradio_interface()
-interface.launch(debug=True)
+        interface.launch(debug=True)