Update app.py

Updated webcam capture code

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 from fastai.vision.all import *
+import numpy as np
+import cv2
 
 # Load the pre-trained model
 model_path = "assets/model-r34.pkl"  # Replace with the path to your model file
@@ -33,24 +35,29 @@ if uploaded_image is not None:
 # Capture an image from webcam
 capture = st.checkbox("Capture an Image from Webcam")
 
-if capture:
+def capture_image():
     st.write("Click the button to capture the image")
     capture_button = st.button("Capture")
 
     if capture_button:
         # Capture the image from the webcam
-        # You can use Python libraries like OpenCV for this
-        # Replace this with your code for capturing an image
+        cap = cv2.VideoCapture(0)
+        
+        if not cap.isOpened():
+            st.error("Error: Unable to access the camera")
+            return
         
-        # For example:
-        # cap = cv2.VideoCapture(0)
-        # ret, frame = cap.read()
-        # cv2.imwrite("captured_image.jpg", frame)
-        # cap.release()
+        ret, frame = cap.read()
+            if not ret:
+                st.error("Error: Could not capture image")
+                return
         
         # After capturing the image, predict and display it
-        captured_image_path = "captured_image.jpg"  # Replace with the actual path
-        st.image(captured_image_path, caption="Captured Image", use_column_width=True)
+        captured_image_path = "captured_image.jpg"  
+        cv2.imwrite(captured_image_path, frame)
+
+        # Display the captured image
+        st.image(captured_image_path, caption = "Captured Image", use_column_width = True)
         
         # Make predictions
         prediction, confidence = predict(captured_image_path)
@@ -60,6 +67,10 @@ if capture:
         st.write(f"Gesture: {prediction}")
         st.write(f"Confidence: {confidence:.2f}")
 
+# Capture calling
+if capture:
+    capture_image()
+
 # Example images
 st.sidebar.title("Example Images")
 example_images = {