Update app.py

app.py

@@ -1,7 +1,6 @@
 # -*- coding: utf-8 -*-
 """
 Created on Tue May 20 11:00:14 2025
-
 @author: ColinWang
 """
 import streamlit as st
@@ -200,11 +199,64 @@ class VideoTransformer(VideoTransformerBase):
             st.write(f"已捕获快照 {len(self.snapshots)}/20")
         return img  # 返回原始帧以供前端显示
 
+# ======================
+# Cover Page
+# ======================
+
+def cover_page():
+    """Display the cover page with project overview and usage instructions."""
+    st.title("Smoking Detection System")
+    
+    st.header("Project Overview")
+    st.write("""
+    The Smoking Detection System is a Streamlit-based web application designed to detect smoking behavior
+    in images or real-time video streams. It leverages advanced machine learning models to classify images
+    for smoking activity, gender, and age range. The system is structured to provide both static image analysis
+    and real-time video processing, with audio feedback for detected smoking incidents.
+    
+    **Purpose**: The primary goal is to identify smoking behavior in public or controlled environments,
+    providing insights into the demographics (gender and age) of individuals engaged in smoking. This can
+    be used for monitoring compliance with no-smoking policies or conducting behavioral studies.
+    
+    **Significance**: The application promotes public health by enabling automated monitoring of smoking
+    activities, potentially aiding in the enforcement of smoking regulations and raising awareness about
+    smoking prevalence across different demographics.
+    
+    **Structure**:
+    - **Cover Page**: Provides an overview and usage instructions.
+    - **Photo Detection**: Analyzes a single uploaded or captured image for smoking, gender, and age.
+    - **Real-Time Video Detection**: Processes video streams, capturing snapshots to detect smoking and
+      analyze demographics if smoking is detected.
+    """)
+    
+    st.header("Usage Instructions")
+    st.write("""
+    1. **Navigation**: Use the sidebar to select a page:
+       - **Cover Page**: View this project overview.
+       - **Photo Detection**: Upload an image or use the camera to capture a photo for analysis.
+       - **Real-Time Video Detection**: Enable the webcam for continuous monitoring.
+    2. **Photo Detection**:
+       - Choose to upload an image or capture one using the camera.
+       - The system will classify the image for smoking. If smoking is detected, it will further analyze
+         gender and age, and play an audio alert based on the results.
+    3. **Real-Time Video Detection**:
+       - Start the webcam to capture 20 snapshots over one minute.
+       - The system analyzes each snapshot for smoking. If smoking is detected in more than two snapshots,
+         it performs gender and age classification and displays the results.
+       - An audio alert is played if smoking is confirmed, based on the most common gender and age range.
+    4. **Requirements**:
+       - Ensure the 'audio' directory contains .wav files named in the format '<age_range> <gender>.wav'
+         (e.g., '10-19 male.wav') for audio feedback.
+       - Set up Twilio environment variables (TWILIO_ACCOUNT_SID and TWILIO_AUTH_TOKEN) for WebRTC.
+    """)
+
+
 # ======================
 # 照片检测页面
 # ======================
 
 def photo_detection_page():
+    audio_placeholder = st.empty()
     st.title("照片检测")
     st.write("上传一张图片或使用摄像头拍摄，检测是否吸烟，若检测到吸烟则进一步分析性别和年龄。")
 
@@ -245,9 +297,17 @@ def photo_detection_page():
                 age_result = age_detection(image)
                 st.success("The age result is:")
             st.write(age_result)
+            
+            audio_placeholder.empty()
+            audio_key = f"{age_result} {gender_result.lower()}"
+            if audio_key in audio_data:
+                audio_bytes = audio_data[audio_key]
+                play_audio_via_js(audio_bytes)
+            else:
+                st.error(f"音频文件不存在: {audio_key}.wav")
 
 # ======================
-# 实时检测页面（原主函数）
+# 实时检测页面（
 # ======================
 
 def real_time_detection_page():
@@ -358,11 +418,13 @@ def real_time_detection_page():
 
 def main():
     st.sidebar.title("导航")
-    page = st.sidebar.selectbox("选择页面", ["照片检测", "实时视频检测"])
+    page = st.sidebar.selectbox("选择页面", ["coverpage","照片检测", "实时视频检测"])
 
+    if page == "coverpage":
+            cover_page()
     if page == "照片检测":
         photo_detection_page()
-    else:
+    if page == "实时视频检测":
         real_time_detection_page()
 
 if __name__ == "__main__":