Update app.py

app.py

@@ -17,17 +17,35 @@ def main():
     st.title("Machine Learning")
 
     with st.expander("1: Add Your Data Source"):
-            uploaded_file = st.file_uploader("Upload your CSV file", type=["csv"])
-            
-            # If no file is uploaded, load example.csv
-            if uploaded_file is None:
-                try:
-                    data = pd.read_csv('example.csv')  # Load example CSV
-                    st.info("Loaded example.csv")
-                except FileNotFoundError:
-                    st.error("Example CSV file not found. Please upload your own CSV file.")
-            else:
-                data = pd.read_csv(uploaded_file)
+        uploaded_file = st.file_uploader("Upload your CSV or Excel file", type=["csv", "xlsx", "xls"])
+    
+        if uploaded_file is None:
+            try:
+                data = pd.read_csv('example.csv')  # Load example CSV
+                st.info("Loaded example.csv")
+            except FileNotFoundError:
+                st.error("Example CSV file not found. Please upload your own CSV or Excel file.")
+            except pd.errors.EmptyDataError:
+                st.error("Example CSV file is empty or invalid.")
+        else:
+            try:
+                if uploaded_file.name.endswith('.csv'):
+                    data = pd.read_csv(uploaded_file)
+                elif uploaded_file.name.endswith(('.xlsx', '.xls')):
+                    data = pd.read_excel(uploaded_file)
+                
+                # Check if the file has content
+                if data.empty:
+                    st.error("Uploaded file is empty. Please upload a valid CSV or Excel file.")
+                else:
+                    st.success("File uploaded successfully!")
+            except pd.errors.EmptyDataError:
+                st.error("The uploaded file is empty or contains no readable data.")
+            except ValueError:
+                st.error("Error in file format. Please ensure the file is a valid CSV or Excel.")
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
+
 
     with st.expander("2: DataSet Preview"):
         if uploaded_file is not None: