Update app.py

app.py

@@ -5,6 +5,15 @@ from PIL import Image
 import urllib.request
 import io
 from utils import *
+from IPython.display import display, Javascript, Image
+from google.colab.output import eval_js
+from base64 import b64decode, b64encode
+import cv2
+import numpy as np
+import html
+from google.colab.patches import cv2_imshow
+from IPython.display import clear_output
+import matplotlib.pyplot as plt
 
 
 # Initialize labels and model
@@ -25,8 +34,49 @@ st.markdown('''
   </div>
 ''', unsafe_allow_html=True)
 
+# Function to take a photo using the webcam
+def take_photo(filename='photo.jpg', quality=0.8):
+    js = Javascript('''
+        async function takePhoto(quality) {
+            const div = document.createElement('div');
+            const capture = document.createElement('button');
+            capture.textContent = 'Capture';
+            div.appendChild(capture);
+
+            const video = document.createElement('video');
+            video.style.display = 'block';
+            const stream = await navigator.mediaDevices.getUserMedia({ video: true });
+
+            document.body.appendChild(div);
+            div.appendChild(video);
+            video.srcObject = stream;
+            await video.play();
+
+            // Resize the output to fit the video element.
+            google.colab.output.setIframeHeight(document.documentElement.scrollHeight, true);
+
+            // Wait for Capture to be clicked.
+            await new Promise((resolve) => capture.onclick = resolve);
+
+            const canvas = document.createElement('canvas');
+            canvas.width = video.videoWidth;
+            canvas.height = video.videoHeight;
+            canvas.getContext('2d').drawImage(video, 0, 0);
+            stream.getVideoTracks()[0].stop();
+            div.remove();
+            return canvas.toDataURL('image/jpeg', quality);
+        }
+    ''')
+    display(js)
+    data = eval_js('takePhoto({})'.format(quality))
+    binary = b64decode(data.split(',')[1])
+    with open(filename, 'wb') as f:
+        f.write(binary)
+    return filename
+    
+
 opt = st.selectbox("How do you want to upload the image for classification?",
-                    ('Please Select', 'Upload image via link', 'Upload image from device'))
+                    ('Please Select', 'Upload image via link', 'Upload image from device', 'Capture a picture'))
 
 # Image processing based on user selection
 image = None
@@ -47,6 +97,9 @@ elif opt == 'Upload image via link':
         except ValueError:
             st.error("Please Enter a valid Image Address!")
 
+elif opt == 'Capture a picture':
+    take_photo()
+
 try:
     if image is not None:
         st.image(image, width=256, caption='Uploaded Image')