Update app.py

app.py

@@ -3,16 +3,23 @@ import pandas as pd
 import seaborn as sns
 import matplotlib.pyplot as plt
 from dataset_import import load_data
+from data_cleaning import clean_data
 
 st.title("AnalyzeYT Dataset Analysis")
 
-# Load dataset
+# Load and clean dataset
 data = load_data()
 
 if data is not None:
     st.write("Loaded Data Preview:")
     st.write(data.head())
-
+    
+    # Clean data
+    data = clean_data(data)
+    
+    st.write("Cleaned Data Preview:")
+    st.write(data.head())
+    
     # Show data description
     st.write("Data Description:")
     st.write(data.describe())
@@ -21,7 +28,7 @@ if data is not None:
     st.write("Correlation Matrix:")
     corr = data.corr()
     st.write(corr)
-    st.write(sns.heatmap(corr, annot=True, cmap='coolwarm'))
+    sns.heatmap(corr, annot=True, cmap='coolwarm')
     st.pyplot()
 
     # Data visualization options