Update app.py

app.py

@@ -15,7 +15,7 @@ st.markdown("""
             font-family: 'Arial', sans-serif;
         }
         .stApp {
-            background-color: white;
+            background-color: black;
             padding: 20px;
             border-radius: 10px;
             box-shadow: 2px 2px 10px rgba(0, 0, 0, 0.1);
@@ -82,12 +82,15 @@ if uploaded_file is not None:
 
         # Preprocessing function
         def preprocess_text(text):
-            text = text.lower()
-            text = re.sub(r'\s+', ' ', text)
-            text = re.sub(r'[^a-z\s]', '', text)
+            text = text.lower() # Convert to lowercase
+            text = re.sub(r'\s+', ' ', text) # Remove extra spaces
+            text = re.sub(r'[^a-z\s]', '', text) # Remove special characters & numbers
             return text
 
+        # Apply preprocessing and classification
         df['processed_content'] = df['content'].apply(preprocess_text)
+
+        # Classify each record into one of the five classes
         df['class'] = df['processed_content'].apply(lambda x: text_classification_pipeline(x)[0]['label'] if x.strip() else "Unknown")
 
         # Show results