Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,35 +1,41 @@
 import streamlit as st
-import av
-import cv2
+import av  # For video frame processing
+import cv2  # OpenCV for image processing
 import numpy as np
-import mediapipe as mp
-from streamlit_webrtc import webrtc_streamer, WebRtcMode
+import mediapipe as mp  # Pose estimation
+from streamlit_webrtc import webrtc_streamer, WebRtcMode  # WebRTC integration
 
-# Initialize MediaPipe Pose
+# Initialize MediaPipe Pose components
 mp_pose = mp.solutions.pose
 mp_drawing = mp.solutions.drawing_utils
 
-# Session state initialization
-if 'camera_access' not in st.session_state:
-    st.session_state.camera_access = False
+# Initialize session state variables
 if 'posture_status' not in st.session_state:
-    st.session_state.posture_status = "カメラを起動してください (Please enable camera)"
-if 'last_status' not in st.session_state:
-    st.session_state.last_status = ""
+    st.session_state.posture_status = "Please enable camera for analysis"
 
 def analyze_posture(image):
-    """Analyze posture on the image and return annotated image and status"""
+    """
+    Analyze posture using MediaPipe Pose
+    Args:
+        image: Input frame from camera
+    Returns:
+        annotated_image: Frame with pose landmarks drawn
+        posture_status: Analysis results text
+    """
     with mp_pose.Pose(
         min_detection_confidence=0.5,
         min_tracking_confidence=0.5,
         model_complexity=1
     ) as pose:
         
+        # Convert color space and process frame
         image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
         results = pose.process(image_rgb)
         
         annotated_image = image.copy()
+        
         if results.pose_landmarks:
+            # Draw pose landmarks on the frame
             mp_drawing.draw_landmarks(
                 annotated_image, 
                 results.pose_landmarks, 
@@ -39,15 +45,22 @@ def analyze_posture(image):
             )
             posture_status = check_posture(results.pose_landmarks, image.shape)
         else:
-            posture_status = "キーポイントが検出されませんでした (Key points not detected)"
+            posture_status = "No pose detected - ensure full body is visible"
         
         return annotated_image, posture_status
 
 def check_posture(landmarks, image_shape):
-    """Analyze posture and return text report"""
+    """
+    Analyze body landmarks and generate posture report
+    Args:
+        landmarks: Detected pose landmarks
+        image_shape: Dimensions of input image
+    Returns:
+        Formatted posture analysis report
+    """
     h, w, _ = image_shape
     
-    # Get key points
+    # Get key body points
     left_shoulder = landmarks.landmark[mp_pose.PoseLandmark.LEFT_SHOULDER]
     right_shoulder = landmarks.landmark[mp_pose.PoseLandmark.RIGHT_SHOULDER]
     left_hip = landmarks.landmark[mp_pose.PoseLandmark.LEFT_HIP]
@@ -56,102 +69,103 @@ def check_posture(landmarks, image_shape):
     right_ear = landmarks.landmark[mp_pose.PoseLandmark.RIGHT_EAR]
     nose = landmarks.landmark[mp_pose.PoseLandmark.NOSE]
     
-    # Determine posture (sitting/standing)
+    # Determine if sitting or standing
     sitting = left_hip.y < left_shoulder.y + 0.1 or right_hip.y < right_shoulder.y + 0.1
     
     messages = []
     
-    # Check for forward head posture
+    # Forward head posture check
     head_forward = (left_ear.y > left_shoulder.y + 0.1 or right_ear.y > right_shoulder.y + 0.1) and \
                    (nose.y > left_shoulder.y or nose.y > right_shoulder.y)
     if head_forward:
-        messages.append("• 頭が前に傾いています (テキストネック) (Head tilted forward - text neck)")
+        messages.append("• Forward head tilt detected (text neck)")
     
-    # Check for rounded shoulders
+    # Rounded shoulders check
     shoulders_rounded = left_shoulder.x > left_hip.x + 0.05 or right_shoulder.x < right_hip.x - 0.05
     if shoulders_rounded:
-        messages.append("• 肩が丸まっています (Rounded shoulders)")
+        messages.append("• Rounded shoulders detected")
     
-    # Check for side tilt
+    # Side tilt check
     shoulder_diff = abs(left_shoulder.y - right_shoulder.y)
     hip_diff = abs(left_hip.y - right_hip.y)
     if shoulder_diff > 0.05 or hip_diff > 0.05:
-        messages.append("• 体が横に傾いています (Asymmetrical posture)")
+        messages.append("• Body leaning to one side")
     
-    # Check pelvis position
+    # Pelvis position check
     if sitting and (left_hip.y < left_shoulder.y + 0.15 or right_hip.y < right_shoulder.y + 0.15):
-        messages.append("• 骨盤が前に傾いています (Pelvis tilted forward)")
+        messages.append("• Pelvis tilted forward (sitting posture)")
     
     # Generate final report
     if messages:
         report = [
-            f"**{'座り姿勢' if sitting else '立ち姿勢'} - 問題が検出されました ({'Sitting' if sitting else 'Standing'} - problems detected):**",
+            f"**{'Sitting' if sitting else 'Standing'} posture issues detected:**",
             *messages,
-            "\n**アドバイス (Recommendations):**",
-            "• 頭をまっすぐに保ち、耳が肩の上にくるように (Keep your head straight, ears over shoulders)",
-            "• 肩を後ろに引き下げて (Pull shoulders back and down)",
-            "• 背中をまっすぐに保ち、横に傾かないように (Keep your back straight, avoid side tilting)",
-            "• 座るときは坐骨で支えるように (When sitting, support your weight on sitting bones)"
+            "\n**Recommendations:**",
+            "• Keep head straight - ears over shoulders",
+            "• Pull shoulders back and down",
+            "• Maintain straight back, avoid side leaning",
+            "• When sitting, support weight on sitting bones"
         ]
     else:
         report = [
-            f"**完璧な姿勢です ({'座り姿勢' if sitting else '立ち姿勢'})! (Perfect posture {'sitting' if sitting else 'standing'})**",
-            "すべてのキーポイントが正しい位置にあります (All key points are in correct position)",
-            "\n**アドバイス (Advice):**",
-            "• 一日中姿勢に気を付けてください (Continue to monitor your posture throughout the day)"
+            f"**Excellent {'sitting' if sitting else 'standing'} posture!**",
+            "All key points are properly aligned",
+            "\n**Tips:**",
+            "• Continue monitoring your posture daily"
         ]
     
     return "\n\n".join(report)
 
 def video_frame_callback(frame):
-    """Process each video frame"""
+    """
+    Callback function for processing each video frame
+    Args:
+        frame: Incoming video frame from WebRTC
+    Returns:
+        Processed video frame with pose landmarks
+    """
     img = frame.to_ndarray(format="bgr24")
     
     try:
+        # Analyze posture and update session state
         analyzed_img, posture_status = analyze_posture(img)
-        if posture_status != st.session_state.last_status:
-            st.session_state.posture_status = posture_status
-            st.session_state.last_status = posture_status
+        st.session_state.posture_status = posture_status
         return av.VideoFrame.from_ndarray(analyzed_img, format="bgr24")
     except Exception as e:
-        st.error(f"処理エラー: {str(e)} (Processing error)")
-        return av.VideoFrame.from_ndarray(img, format="bgr24")
+        st.error(f"Processing error: {str(e)}")
+        return frame
 
 def main():
+    # Configure Streamlit page
     st.set_page_config(layout="wide")
-    st.title("📷 リアルタイム姿勢分析アプリ (Real-time Posture Analyzer)")
+    st.title("📷 Real-time Posture Analysis")
     
-    # Create columns
+    # Create two-column layout
     col1, col2 = st.columns([2, 1])
     
     with col1:
-        st.header("カメラビュー (Camera View)")
+        st.header("Camera Feed")
         
-        if not st.session_state.camera_access:
-            st.warning("⚠️ カメラを使用するには許可が必要です (Camera access requires permission)")
-            if st.button("カメラアクセスを許可 (Allow camera access)"):
-                st.session_state.camera_access = True
-                st.rerun()
-        else:
-            webrtc_ctx = webrtc_streamer(
-                key="posture-analysis",
-                mode=WebRtcMode.SENDRECV,
-                video_frame_callback=video_frame_callback,
-                media_stream_constraints={"video": True, "audio": False},
-                async_processing=True,
-                rtc_configuration={
-                    "iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]
-                }
-            )
-            
-            if not webrtc_ctx.state.playing:
-                st.session_state.posture_status = "カメラが停止しました (Camera stopped)"
-                st.session_state.last_status = ""
+        # WebRTC streamer component
+        webrtc_ctx = webrtc_streamer(
+            key="posture-analysis",
+            mode=WebRtcMode.SENDRECV,
+            video_frame_callback=video_frame_callback,
+            media_stream_constraints={"video": True, "audio": False},
+            async_processing=True,
+            rtc_configuration={
+                "iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]
+            }
+        )
+        
+        # Handle camera state
+        if not webrtc_ctx.state.playing:
+            st.session_state.posture_status = "Camera is off - please enable access"
+            st.warning("Please allow camera permissions when prompted")
     
     with col2:
-        st.header("姿勢分析結果 (Posture Analysis)")
-        status_placeholder = st.empty()
-        status_placeholder.markdown(st.session_state.posture_status)
+        st.header("Posture Analysis")
+        st.markdown(st.session_state.posture_status)
 
 if __name__ == "__main__":
     main()