Update app.py

app.py

@@ -3,43 +3,32 @@ import cv2
 import numpy as np
 from PIL import Image
 
-# 打开摄像头
-def load_camera():
-    cap = cv2.VideoCapture(0)  # 0 是默认摄像头的索引
+# 创建一个摄像头捕捉函数
+def capture_camera():
+    cap = cv2.VideoCapture(0)  # 参数0表示默认摄像头
     if not cap.isOpened():
-        st.error("Cannot open camera")
-        return None
-    return cap
-
-# 捕获图片
-def capture_image(cap):
+        st.error("无法打开摄像头")
+        return None, None
     ret, frame = cap.read()
+    cap.release()
     if not ret:
-        st.error("Can't receive frame (stream end?). Exiting ...")
-        return None
-    return frame
+        st.error("无法从摄像头获取图像")
+        return None, None
+    return frame, cap
 
-# 主程序
 def main():
-    st.title("Camera Capture with Streamlit")
-    
-    # 摄像头对象
-    cap = load_camera()
+    st.title("摄像头拍照应用")
 
-    if cap:
-        st.sidebar.button("Click to Capture", key="capture")
-        frame = capture_image(cap)
-        
+    # 添加一个按钮用于拍照
+    if st.button("打开摄像头拍照"):
+        frame, cap = capture_camera()
         if frame is not None:
-            # 将图像从BGR转换为RGB
-            frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-            img = Image.fromarray(frame_rgb)
-
-            # 显示图像
-            st.image(img, caption="Captured Image", use_column_width=True)
+            # 转换颜色从BGR到RGB
+            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            # 将图像显示在Streamlit界面上
+            st.image(frame, use_column_width=True, caption="捕捉到的图像")
 
-        # 释放摄像头资源
-        cap.release()
+    st.text("点击上方按钮打开摄像头并拍照。")
 
 if __name__ == "__main__":
     main()