Update app.py

app.py

@@ -109,21 +109,18 @@ def video_feed(video_source):
         text_placeholder.markdown(f"<h3 style='text-align: center;'>{result_text}</h3>", unsafe_allow_html=True)
 
 # Sidebar for video or image upload
-upload_choice = st.sidebar.radio("Choose input source", ["Upload Video", "Upload Image", "Camera"])
+upload_choice = st.sidebar.radio("Choose input source", ["Upload Image", "Upload Video", "Camera"])
 
 if upload_choice == "Camera":
-    # Access camera
-    video_source = cv2.VideoCapture(0)
-    video_feed(video_source)
+    # Use Streamlit's built-in camera input widget for capturing images from the webcam
+    image = st.camera_input("Take a picture")
 
-elif upload_choice == "Upload Video":
-    uploaded_video = st.file_uploader("Upload Video", type=["mp4", "mov", "avi", "mkv", "webm"])
-    if uploaded_video:
-        # Temporarily save the video to disk
-        with tempfile.NamedTemporaryFile(delete=False) as tfile:
-            tfile.write(uploaded_video.read())
-            video_source = cv2.VideoCapture(tfile.name)
-            video_feed(video_source)
+    if image is not None:
+        # Convert the image to a numpy array
+        frame = np.array(Image.open(image))
+        frame, result_text = process_frame(frame)
+        st.image(frame, caption='Processed Image', use_column_width=True)
+        st.markdown(f"<h3 style='text-align: center;'>{result_text}</h3>", unsafe_allow_html=True)
 
 elif upload_choice == "Upload Image":
     uploaded_image = st.file_uploader("Upload Image", type=["png", "jpg", "jpeg", "gif"])
@@ -134,4 +131,13 @@ elif upload_choice == "Upload Image":
         st.image(frame, caption='Processed Image', use_column_width=True)
         st.markdown(f"<h3 style='text-align: center;'>{result_text}</h3>", unsafe_allow_html=True)
 
+elif upload_choice == "Upload Video":
+    uploaded_video = st.file_uploader("Upload Video", type=["mp4", "mov", "avi", "mkv", "webm"])
+    if uploaded_video:
+        # Temporarily save the video to disk
+        with tempfile.NamedTemporaryFile(delete=False) as tfile:
+            tfile.write(uploaded_video.read())
+            video_source = cv2.VideoCapture(tfile.name)
+            video_feed(video_source)
+
 st.sidebar.write("Emotion Labels: Angry, Fear, Happy, Neutral, Sad, Surprise")