Update app.py

app.py

@@ -8,21 +8,9 @@ classifier = pipeline("text-classification", model=model_name)
 # Configure the Streamlit page
 st.set_page_config(page_title="Amharic Hate Speech Detector", page_icon="🕵️‍♂️", layout="centered")
 
-# Add CSS to manage the background color change
-def set_background(color):
-    st.markdown(
-        f"""
-        <style>
-            body {{
-                background-color: {color};
-            }}
-        </style>
-        """,
-        unsafe_allow_html=True
-    )
-
 # Set default background color
 default_bg_color = "#f0f2f6"
+bg_color = default_bg_color
 
 # Add a stylish header with a logo
 st.markdown(
@@ -50,25 +38,26 @@ if st.button("Analyze Text 🚀"):
         if label == "LABEL_0":
             prediction = "Normal Text 🟢"
             color = "#28a745"
-            set_background(default_bg_color)  # Reset background for normal text
         else:
             prediction = "Hate Speech Detected 🔴"
             color = "#dc3545"
-            set_background("#FFBABA")  # Set red background for hate speech
+            bg_color = "#FFBABA"  # Update background to a red color for hate speech
 
-            # Play alarm sound and display warning
+            # Display warning
             st.warning("⚠️ Warning: Hate Speech Detected! ⚠️")
-            st.markdown(
-                """
-                <audio autoplay="true" loop="false">
-                    <source src="beep-warning-6387.mp3" type="audio/mp3">
-                    Your browser does not support the audio element.
-                </audio>
-                """,
-                unsafe_allow_html=True
-            )
+
+        # Apply the background color dynamically
+        st.markdown(
+            f"""
+            <style>
+                body {{ background-color: {bg_color}; }}
+                .stAlert {{ text-align: center; }}
+            </style>
+            """,
+            unsafe_allow_html=True
+        )
 
         # Display the result with styled message
         st.markdown(f"<h2 style='text-align: center; color: {color};'>{prediction}</h2>", unsafe_allow_html=True)
     else:
-        st.warning("Please enter some text to analyze.")
+        st.warning("Please enter some text to analyze.")