Update app.py

app.py

@@ -17,10 +17,15 @@ pose = mp_pose.Pose(static_image_mode=False, min_detection_confidence=0.5)
 mp_drawing = mp.solutions.drawing_utils
 
 def process_frame(frame):
-    # 📸 PIL.Image를 numpy로 변환
+    # 1. None 체크
+    if frame is None:
+        return None, "Error: No image received from camera", None
+
+    # 2. PIL.Image → numpy 변환
     if isinstance(frame, Image.Image):
         frame = np.array(frame)
 
+    # 3. RGB 변환 예외 처리
     try:
         image_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
     except Exception as e:
@@ -28,10 +33,12 @@ def process_frame(frame):
 
     results = pose.process(image_rgb)
 
+    # 관절 시각화
     if results.pose_landmarks:
         mp_drawing.draw_landmarks(frame, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)
 
     try:
+        # 모델 입력 전처리
         resized = cv2.resize(image_rgb, (128, 128))
         input_data = resized.flatten().reshape(1, -1)
         prediction_proba = model.predict_proba(input_data)[0]
@@ -42,7 +49,7 @@ def process_frame(frame):
     except Exception as e:
         return frame, f"Prediction error: {e}", None
 
-    # 📊 그래프 생성
+    # 예측 그래프 생성
     try:
         fig, ax = plt.subplots()
         ax.bar(action_labels.values(), prediction_proba, color='skyblue')
@@ -58,18 +65,18 @@ def process_frame(frame):
         graph_img = None
         label += f" (Graph error: {e})"
 
-    # 💬 예측 텍스트 표시
+    # 예측 텍스트 이미지에 표시
     try:
         cv2.putText(frame, label, (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,255,0), 2)
     except:
-        pass  # 일부 모바일 장치에서 오류 발생 가능
+        pass
 
     return frame, label, graph_img
 
-# Gradio 인터페이스
+# Gradio 인터페이스 구성
 demo = gr.Interface(
     fn=process_frame,
-    inputs=gr.Image(source="webcam", streaming=False),
+    inputs=gr.Image(source="webcam", streaming=False),  # ✅ 사진 촬영 버튼 활성화!
     outputs=[
         gr.Image(label="Pose + Prediction"),
         gr.Textbox(label="Predicted Action"),
@@ -77,10 +84,11 @@ demo = gr.Interface(
     ],
     title="Pose Action Classifier",
     description=(
-        "📱 Mobile users: tap the button to take a photo. On Chrome, you can switch to the back camera by tapping the camera icon "
-        "(if available). Internet Explorer is not supported — use Chrome or Safari."
+        "📱 On mobile, tap the button to take a photo. "
+        "You can switch to the back camera by tapping the camera icon. "
+        "Internet Explorer is not supported — use Chrome or Safari."
     )
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()