Update app.py

app.py

@@ -11,8 +11,6 @@ from spacy.cli import download
 import base64
 import numpy as np
 import datetime
-import time
-from streamlit_option_menu import option_menu
 
 # Download necessary NLP models
 nltk.download('wordnet')
@@ -77,20 +75,6 @@ def add_image_to_state(image, caption, capture_time):
     if len(st.session_state.captured_images) < 20:
         st.session_state.captured_images.append((img_str, caption, capture_time))
 
-def capture_images_from_webcam(num_images=5, interval=0.5):
-    captured_images = []
-    cap = cv2.VideoCapture(0)
-    for _ in range(num_images):
-        ret, frame = cap.read()
-        if not ret:
-            break
-        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        pil_image = Image.fromarray(frame_rgb)
-        captured_images.append(pil_image)
-        time.sleep(interval)
-    cap.release()
-    return captured_images
-
 def page_image_captioning():
     st.title("Image Captioning")
     st.write("Your image captioning code here")
@@ -102,14 +86,15 @@ def page_video_captioning():
 def page_webcam_capture():
     st.title("Live Captioning with Webcam")
 
-    if st.button("Capture 5 Images"):
-        captured_images = capture_images_from_webcam()
-        for img in captured_images:
-            img_array = np.array(img)
-            caption = generate_caption(img)
-            capture_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-            add_image_to_state(img_array, caption, capture_time)
-            st.image(img, caption=f"Caption: {caption}")
+    img_file = st.camera_input("Capture an image")
+
+    if img_file:
+        img = Image.open(img_file)
+        img_array = np.array(img)
+        caption = generate_caption(img)
+        capture_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        add_image_to_state(img_array, caption, capture_time)
+        st.image(img, caption=f"Caption: {caption}")
 
     if st.button('Stop'):
         st.write("Camera stopped.")