Update app.py

app.py

@@ -7,7 +7,10 @@ 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")
-st.markdown("<style> body { background-color: #f0f2f6; } </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(
@@ -31,16 +34,39 @@ if st.button("Analyze Text 🚀"):
         label = result[0]['label']
         score = result[0]['score']
 
-        # Map label to understandable output
+        # Determine if text is hate speech and update color/background
         if label == "LABEL_0":
             prediction = "Normal Text 🟢"
             color = "#28a745"
         else:
-            prediction = "Hate Speech 🔴"
+            prediction = "Hate Speech Detected 🔴"
             color = "#dc3545"
+            bg_color = "#FFBABA"  # Update background to a red color for hate speech
+
+            # Play alarm sound and display warning
+            st.warning("⚠️ Warning: Hate Speech Detected! ⚠️")
+            st.markdown(
+                """
+                <audio autoplay>
+                    <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)
-        #st.write(f"Confidence: {score * 100:.2f}%")
     else:
         st.warning("Please enter some text to analyze.")