Update app.py

app.py

@@ -302,6 +302,7 @@ def real_time_detection_page():
     classification_text = st.empty()
     classification_progress = st.empty()
     detection_info = st.empty()
+    status_alert = st.empty()  # New placeholder for status alert
     table = st.empty()
     image_display = st.empty()
     audio = st.empty()
@@ -338,6 +339,7 @@ def real_time_detection_page():
                 classification.progress(33)
 
                 if smoking_count > 2:
+                    status_alert.error("Smoking Detected!")  # Red alert for smoking
                     gender_results = [classify_gender(img) for img in snapshots]
                     classification.progress(66)
                     age_results = [classify_age(img) for img in snapshots]
@@ -382,7 +384,7 @@ def real_time_detection_page():
                     else:
                         st.error(f"Audio file not found: {audio_key}.wav")
                 else:
-                    st.markdown("**Smoking Status:** Not Smoking")
+                    status_alert.success("No Smoking Detected")  # Green alert for no smoking
                     image_display.empty()
                     audio.empty()
                     classification_text.text("Classification Progress: Completed!")
@@ -405,6 +407,7 @@ def real_time_detection_page():
                 classification_progress.empty()
                 classification_text.empty()
                 capture_text.empty()
+                status_alert.empty()  # Clear the alert for the next cycle
                 detection_info.info("Starting detection...")
                 ctx.video_transformer.snapshots = []
                 ctx.video_transformer.last_capture_time = time.time()