Update camera.py

camera.py

@@ -1,29 +1,22 @@
-# camera.py
-
-import streamlit as st
 import cv2
-import numpy as np
-from PIL import Image
+import streamlit as st
 
 def take_picture():
-    # 打开摄像头
-    cap = cv2.VideoCapture(0)
-    
-    st.text("摄像头正在启动...")
-    
-    if cap.isOpened():
-        ret, frame = cap.read()
-        if ret:
-            st.image(frame, channels="BGR")  # 显示视频帧
-            # 保存照片
-            img = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
-            img.save("captured_image.jpg")
-            st.success("照片已保存！")
-        else:
-            st.error("无法捕获图像")
-    else:
-        st.error("无法打开摄像头")
+    """
+    使用 OpenCV 拍摄一张照片并返回图像数据。
+    """
+    cap = cv2.VideoCapture(0)  # 打开默认摄像头
+    ret, frame = cap.read()
     cap.release()
+    return frame
+
+def show_picture(img):
+    """
+    在 Streamlit 中显示图像。
+    """
+    st.image(img, channels="BGR")
 
 if __name__ == "__main__":
-    take_picture()
+    if st.button("拍照"):
+        img = take_picture()
+        show_picture(img)