Update app.py

app.py

@@ -14,6 +14,7 @@ import base64
 from twilio.rest import Client
 from collections import Counter
 import uuid
+import pandas as pd
 
 # ======================
 # 模型加载函数（缓存）
@@ -39,7 +40,6 @@ smoke_pipeline = load_smoke_pipeline()
 gender_pipeline = load_gender_pipeline()
 age_pipeline = load_age_pipeline()
 
-
 # ======================
 # remote settings
 # ======================
@@ -52,7 +52,6 @@ client = Client(account_sid, auth_token)
 
 token = client.tokens.create()
 
-
 # ======================
 # 音频加载函数（缓存）
 # ======================
@@ -76,7 +75,6 @@ def load_all_audios():
 # 应用启动时加载所有音频
 audio_data = load_all_audios()
 
-
 # ======================
 # 照片檢測处理函数
 # ======================
@@ -250,7 +248,6 @@ def cover_page():
        - Set up Twilio environment variables (TWILIO_ACCOUNT_SID and TWILIO_AUTH_TOKEN) for WebRTC.
     """)
 
-
 # ======================
 # 照片检测页面
 # ======================
@@ -278,7 +275,6 @@ def photo_detection_page():
             st.image(image, caption="拍攝的圖片", use_container_width=True)
 
     if image is not None:
-
         # 吸烟分类
         with st.spinner("Wait for smoking detection"):
             smoke_result = smoking_detection(image)
@@ -307,12 +303,16 @@ def photo_detection_page():
                 st.error(f"音频文件不存在: {audio_key}.wav")
 
 # ======================
-# 实时检测页面（
+# 实时检测页面
 # ======================
 
 def real_time_detection_page():
     st.title("实时视频检测")
-    st.write("程序在一分钟内捕获5张快照进行图片分类，首先判定是否吸烟。若检测到吸烟的快照超过2次，则展示年龄与性别分类结果。")
+    st.write("程序在一分钟内捕获5张快照进行图片分类，首先判定是否吸烟。若检测到吸烟的快照超过2次，则将结果添加到表格中，包含快照、性别和年龄。")
+
+    # 初始化 session state 用于存储检测结果
+    if 'detection_results' not in st.session_state:
+        st.session_state.detection_results = []
 
     # 创建用于显示进度文字和进度条的占位容器
     capture_text_placeholder = st.empty()
@@ -321,10 +321,12 @@ def real_time_detection_page():
     classification_progress_placeholder = st.empty()
     detection_info_placeholder = st.empty()
 
+    # 显示检测结果表格
+    table_placeholder = st.empty()
+
     # 启动实时视频流
     ctx = webrtc_streamer(key="unique_example", video_transformer_factory=VideoTransformer,
-                          rtc_configuration={"iceServers": token.ice_servers}
-                          )
+                          rtc_configuration={"iceServers": token.ice_servers})
     image_placeholder = st.empty()
     audio_placeholder = st.empty()
 
@@ -371,13 +373,7 @@ def real_time_detection_page():
                     most_common_gender = Counter(gender_results).most_common(1)[0][0]
                     most_common_age = Counter(age_results).most_common(1)[0][0]
 
-                    result_text = (
-                        f"**吸烟状态：** Smoking (检测到 {smoking_count} 次)\n\n"
-                        f"**性别：** {most_common_gender}\n\n"
-                        f"**年龄范围：** {most_common_age}"
-                    )
-                    classification_result_placeholder.markdown(result_text)
-
+                    # 找到第一张吸烟快照
                     smoking_image = None
                     for idx, label in enumerate(smoke_results):
                         if label.lower() == "smoking":
@@ -385,8 +381,31 @@ def real_time_detection_page():
                             break
                     if smoking_image is None:
                         smoking_image = snapshots[0]
-                    image_placeholder.image(smoking_image, caption="捕获的快照示例", use_container_width=True)
 
+                    # 添加结果到 session state
+                    st.session_state.detection_results.append({
+                        "Timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
+                        "Snapshot": smoking_image,
+                        "Gender": most_common_gender,
+                        "Age Range": most_common_age,
+                        "Smoking Count": smoking_count
+                    })
+
+                    # 更新表格显示
+                    df = pd.DataFrame([
+                        {
+                            "Timestamp": result["Timestamp"],
+                            "Gender": result["Gender"],
+                            "Age Range": result["Age Range"],
+                            "Smoking Count": result["Smoking Count"]
+                        } for result in st.session_state.detection_results
+                    ])
+                    table_placeholder.dataframe(df, use_container_width=True)
+
+                    # 显示示例快照
+                    image_placeholder.image(smoking_image, caption="捕获的吸烟快照", use_container_width=True)
+
+                    # 播放音频
                     audio_placeholder.empty()
                     audio_key = f"{most_common_age} {most_common_gender.lower()}"
                     if audio_key in audio_data:
@@ -402,6 +421,18 @@ def real_time_detection_page():
                     classification_text_placeholder.text("分类进度: 分类完成！")
                     classification_progress.progress(100)
 
+                # 更新表格显示，即使没有吸烟检测到
+                if st.session_state.detection_results:
+                    df = pd.DataFrame([
+                        {
+                            "Timestamp": result["Timestamp"],
+                            "Gender": result["Gender"],
+                            "Age Range": result["Age Range"],
+                            "Smoking Count": result["Smoking Count"]
+                        } for result in st.session_state.detection_results
+                    ])
+                    table_placeholder.dataframe(df, use_container_width=True)
+
                 time.sleep(5)
                 classification_progress_placeholder.empty()
                 classification_text_placeholder.empty()
@@ -418,10 +449,10 @@ def real_time_detection_page():
 
 def main():
     st.sidebar.title("导航")
-    page = st.sidebar.selectbox("选择页面", ["coverpage","照片检测", "实时视频检测"])
+    page = st.sidebar.selectbox("选择页面", ["coverpage", "照片检测", "实时视频检测"])
 
     if page == "coverpage":
-            cover_page()
+        cover_page()
     if page == "照片检测":
         photo_detection_page()
     if page == "实时视频检测":