Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,142 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import cv2
+import numpy as np
+from PIL import Image
+import time
+import mediapipe as mp
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
+mp_pose = mp.solutions.pose
+pose = mp_pose.Pose(
+    min_detection_confidence=0.5,
+    min_tracking_confidence=0.5,
+    model_complexity=1
+)
+mp_drawing = mp.solutions.drawing_utils
 
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
+def analyze_posture(image):
+    """Analyze posture on the image and return the image with keypoints and analysis text"""
+    image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    results = pose.process(image_rgb)
+    
+    annotated_image = image.copy()
+    if results.pose_landmarks:
+        mp_drawing.draw_landmarks(
+            annotated_image, 
+            results.pose_landmarks, 
+            mp_pose.POSE_CONNECTIONS,
+            mp_drawing.DrawingSpec(color=(0, 255, 0), thickness=2, circle_radius=2),
+            mp_drawing.DrawingSpec(color=(0, 0, 255), thickness=2)
+        )
+        posture_status = check_posture(results.pose_landmarks, image.shape)
+    else:
+        posture_status = "キーポイントが検出されませんでした (Key points not detected)"
+    
+    return annotated_image, posture_status
 
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
+def check_posture(landmarks, image_shape):
+    """Analyze posture and return a text report"""
+    h, w, _ = image_shape
+    
+    # Get key 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]
+    right_hip = landmarks.landmark[mp_pose.PoseLandmark.RIGHT_HIP]
+    left_ear = landmarks.landmark[mp_pose.PoseLandmark.LEFT_EAR]
+    right_ear = landmarks.landmark[mp_pose.PoseLandmark.RIGHT_EAR]
+    nose = landmarks.landmark[mp_pose.PoseLandmark.NOSE]
+    
+    # Determine posture (sitting/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
+    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)")
+    
+    # Check for rounded shoulders
+    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)")
+    
+    # Check for side tilt
+    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)")
+    
+    # Check pelvis position
+    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)")
+    
+    # Generate final report
+    if messages:
+        report = [
+            f"**{'座り姿勢' if sitting else '立ち姿勢'} - 問題が検出されました ({'Sitting' if sitting else 'Standing'} - problems 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)"
+        ]
+    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)"
+        ]
+    
+    return "\n\n".join(report)
 
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
+def main():
+    st.set_page_config(layout="wide")
+    st.title("📷 姿勢分析アプリ (Posture Analyzer)")
+    st.write("このアプリはリアルタイムで姿勢を分析します (This app analyzes your posture in real time)")
+    
+    # Create two columns
+    col1, col2 = st.columns([2, 1])
+    
+    with col1:
+        st.header("カメラビュー (Camera View)")
+        run = st.checkbox("ウェブカメラをオンにする (Enable webcam)", value=True)
+        FRAME_WINDOW = st.image([])
+    
+    with col2:
+        st.header("姿勢分析 (Posture Analysis)")
+        status_placeholder = st.empty()
+        if not run:
+            status_placeholder.markdown("""
+                **カメラからのデータを待っています (Waiting for camera data)...**
+                
+                姿勢分析のためにウェブカメラをオンにしてください (Please enable webcam for posture analysis)
+            """)
+    
+    camera = cv2.VideoCapture(0)
+    
+    while run:
+        _, frame = camera.read()
+        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        
+        # Posture analysis
+        analyzed_frame, posture_status = analyze_posture(cv2.cvtColor(frame, cv2.COLOR_RGB2BGR))
+        analyzed_frame = cv2.cvtColor(analyzed_frame, cv2.COLOR_BGR2RGB)
+        
+        # Display in columns
+        with col1:
+            FRAME_WINDOW.image(analyzed_frame)
+        
+        with col2:
+            status_placeholder.markdown(posture_status)
+        
+        time.sleep(0.1)
+    
+    camera.release()
+    cv2.destroyAllWindows()
 
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+if __name__ == "__main__":
+    main()