Update app.py

app.py

@@ -1,18 +1,28 @@
 import gradio as gr
+from PIL import Image
+import cv2
 
-# 定义一个简单的拍照函数
-def capture_image(image):
-    return image
+def capture_image():
+    # 打开摄像头
+    cap = cv2.VideoCapture(0)
+    ret, frame = cap.read()
+    if ret:
+        # 将BGR图像转换为RGB图像
+        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        # 转换为Pillow图像
+        img = Image.fromarray(frame)
+        cap.release()
+        return img
+    else:
+        cap.release()
+        return "无法打开摄像头"
 
-# 使用 Gradio 创建一个摄像头界面
-demo = gr.Interface(
-    fn=capture_image,
-    inputs=gr.Image(source="webcam", tool="editor"),
-    outputs="image",
-    live=True,
-    title="摄像头拍照",
-    description="使用摄像头进行拍照"
-)
+# 创建gradio接口
+with gr.Blocks() as demo:
+    webcam_image = gr.Image(label="摄像头拍摄的图片")
+    capture_button = gr.Button("拍照")
+    
+    capture_button.click(capture_image, outputs=webcam_image)
 
-# 启动 Gradio 应用
+# 启动gradio界面
 demo.launch()