added YouTube URL option

app.py

@@ -1,10 +1,7 @@
-import torch
-import tensorflow as tf
 import time
 import os
 import logging
-from pathlib import Path
-from typing import List, NamedTuple
+
 
 import av
 import cv2
@@ -21,6 +18,7 @@ from transformers import pipeline  # Import Hugging Face transformers pipeline
 
 import requests
 from io import BytesIO  # Import for handling byte streams
+import yt_dlp
 
 
 # CHANGE CODE BELOW HERE, USE TO REPLACE WITH YOUR WANTED ANALYSIS.
@@ -132,17 +130,6 @@ def analyze_sentiment(face):
 # Suppress FFmpeg logs
 os.environ["FFMPEG_LOG_LEVEL"] = "quiet"
 
-# Suppress TensorFlow or PyTorch progress bars
-
-tf.get_logger().setLevel("ERROR")
-os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
-
-# Suppress PyTorch logs
-
-logging.getLogger().setLevel(logging.WARNING)
-torch.set_num_threads(1)
-logging.getLogger("torch").setLevel(logging.ERROR)
-
 # Suppress Streamlit logs using the logging module
 logging.getLogger("streamlit").setLevel(logging.ERROR)
 
@@ -239,7 +226,11 @@ with col1:
     st.subheader("Or Enter Image URL")
     image_url = st.text_input("Image URL")
 
-    # File uploader for videos
+    # Text input for YouTube URL
+    st.subheader("Enter a YouTube URL")
+    youtube_url = st.text_input("YouTube URL")
+
+   # File uploader for videos
     st.subheader("Upload a Video")
     uploaded_video = st.file_uploader(
         "Choose a video...", type=["mp4", "avi", "mov", "mkv"]
@@ -358,6 +349,28 @@ def process_video(video_path):
 
     cap.release()  # Release the video capture object
 
+# Function to get the video stream URL from YouTube using yt-dlp
+
+
+def get_youtube_stream_url(youtube_url):
+    ydl_opts = {
+        'format': 'best[ext=mp4]',
+        'quiet': True,
+    }
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        info_dict = ydl.extract_info(youtube_url, download=False)
+        stream_url = info_dict['url']
+    return stream_url
+
+
+# If a YouTube URL is provided, process the video
+if youtube_url:
+    analysis_init()  # Initialize the analysis UI
+
+    stream_url = get_youtube_stream_url(youtube_url)
+
+    process_video(stream_url)  # Process the video
+
 
 # If a video is uploaded or a URL is provided, process the video
 if uploaded_video is not None or video_url:

requirements.txt

@@ -1,11 +1,12 @@
+tensorboard==2.16.2
+tf-keras
 streamlit
 opencv-python-headless
 numpy
 transformers
-torch
-mtcnn
 setuptools
-tensorflow
-tf-keras
 streamlit_webrtc
-twilio
+twilio
+mtcnn
+yt-dlp
+watchdog