Update app.py

app.py

@@ -291,23 +291,19 @@ def download_video(url):
         return None
 
 # Download YouTube video
-def download_youtube_video(youtube_url):
-    try:
-        yt = YouTube(youtube_url)
-        video_stream = yt.streams.filter(file_extension="mp4", res="360p").first()
-        if not video_stream:
-            video_stream = yt.streams.get_lowest_resolution()
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-        video_stream.download(filename=temp_file.name)
-        return temp_file.name, yt.title, yt.thumbnail_url  # Return video title & thumbnail
-    except Exception as e:
-        return None, None, None
+def get_youtube_embed_code(youtube_url):
+    """Generate an embedded iframe for YouTube videos"""
+    return f"""
+        <iframe width="560" height="315" src="https://www.youtube.com/embed/{youtube_url.split('=')[-1]}" 
+        frameborder="0" allowfullscreen></iframe>
+    """
 
 # Select Video Source
 video_path = None
 video_display_url = None
 video_title = None
 video_thumbnail = None
+youtube_embed = None
 
 # Process Uploaded Video
 if uploaded_video is not None:
@@ -321,27 +317,20 @@ if uploaded_video is not None:
 # Process Video from URL (Check if YouTube or Direct Link)
 elif video_url:
     if "youtube.com" in video_url or "youtu.be" in video_url:
-        video_path, video_title, video_thumbnail = download_youtube_video(video_url)  # Download YouTube video
-        video_display_url = video_url
+        youtube_embed = get_youtube_embed_code(video_url)  # Get embedded iframe
+        video_title = "YouTube Video"
     else:
         video_path = download_video(video_url)  # Download direct MP4 video
-        video_display_url = video_url
+        video_display_url = video_path
         video_title = "Online MP4 Video"
 
-# ✅ Small Video Display with Title & Thumbnail
-if video_display_url:
-    st.markdown(f"### 🎬 {video_title}")  # Show video title
-    if video_thumbnail:
-        st.image(video_thumbnail, caption="Video Thumbnail", width=200)  # Show thumbnail for YouTube videos
-    st.markdown(
-        f"""
-        <video controls width="300">
-            <source src="{video_display_url}" type="video/mp4">
-            Your browser does not support the video tag.
-        </video>
-        """,
-        unsafe_allow_html=True,
-    )
+# ✅ Display Video Properly
+if youtube_embed:
+    st.markdown(f"### 🎬 {video_title}")
+    st.markdown(youtube_embed, unsafe_allow_html=True)  # Show YouTube Video
+elif video_display_url:
+    st.markdown(f"### 🎬 {video_title}")
+    st.video(video_display_url)  # Show Uploaded or MP4 URL Video
 
 # ✅ "Analyze Video" بٹن ہمیشہ شو ہوگا
 analyze_button = st.button("Analyze Video")