# camera.py

import streamlit as st
import cv2
import numpy as np
from PIL import Image

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("无法打开摄像头")
    cap.release()

if __name__ == "__main__":
    take_picture()