Optimize the User Exprence

emotion.py

@@ -16,7 +16,7 @@ ANALYZE_EVERY_SEC = 0.8
 DOWNSAMPLE_W = 720
 EMA_ALPHA = 0.45
 CONF_TH = 40.0
-SWITCH_CONFIRM_SEC = 2.5
+SWITCH_CONFIRM_SEC = 1.5
 IOU_TH = 0.25
 ZOOM_FACTOR = 1.5
 
@@ -92,6 +92,7 @@ class AppState:
 
     final_score: int = 3  
 
+    frozen_frame: Optional[np.ndarray] = None
 
 # =========================
 # UI HTML (強制黑色文字)
@@ -265,6 +266,7 @@ def _reset_state_for_camera(st: AppState) -> AppState:
     st.final_emo = None
     st.final_conf = None
     st.final_score = 3
+    st.frozen_frame = None
     return st
 
 
@@ -273,7 +275,7 @@ def _reset_state_for_camera(st: AppState) -> AppState:
 # =========================
 def on_stream(frame_rgb: np.ndarray, st: AppState):
     if st.finished or (not st.running):
-        cam_visible = False
+        out_img = st.frozen_frame if st.frozen_frame is not None else None
 
         if st.final_emo:
             result_html = _result_card_html(st.final_emo, st.final_conf)
@@ -285,7 +287,7 @@ def on_stream(frame_rgb: np.ndarray, st: AppState):
             hint_html = _camera_hint_stopped()
 
         return (
-            gr.update(visible=cam_visible),
+            gr.update(visible=True, value=out_img),
             gr.update(value=result_html),
             gr.update(value=mood_val),
             gr.update(value=hint_html),
@@ -411,9 +413,10 @@ def on_stream(frame_rgb: np.ndarray, st: AppState):
 
         st.running = False
         st.finished = True
+        st.frozen_frame = frame_rgb # 把現在這一幀畫面存起來
 
         return (
-            gr.update(visible=False),
+            gr.update(visible=True, value=st.frozen_frame),
             gr.update(value=_result_card_html(stable_key, top_conf)),
             gr.update(value=_score_radio_value(EMO_TO_SCORE.get(stable_key, 3))),
             gr.update(value=_camera_hint_done()),