Update app.py
Browse files
app.py
CHANGED
@@ -434,13 +434,15 @@ def download_youtube_video(youtube_url):
|
|
434 |
video_stream = yt.streams.get_lowest_resolution()
|
435 |
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
|
436 |
video_stream.download(filename=temp_file.name)
|
437 |
-
return temp_file.name
|
438 |
except Exception as e:
|
439 |
-
return None
|
440 |
|
441 |
# Select Video Source
|
442 |
video_path = None
|
443 |
video_display_url = None
|
|
|
|
|
444 |
|
445 |
# Process Uploaded Video
|
446 |
if uploaded_video is not None:
|
@@ -449,18 +451,23 @@ if uploaded_video is not None:
|
|
449 |
f.write(uploaded_video.read())
|
450 |
video_path = temp_file.name # Set video path for detection
|
451 |
video_display_url = temp_file.name
|
|
|
452 |
|
453 |
# Process Video from URL (Check if YouTube or Direct Link)
|
454 |
elif video_url:
|
455 |
if "youtube.com" in video_url or "youtu.be" in video_url:
|
456 |
-
video_path = download_youtube_video(video_url) # Download YouTube video
|
457 |
video_display_url = video_url
|
458 |
else:
|
459 |
video_path = download_video(video_url) # Download direct MP4 video
|
460 |
video_display_url = video_url
|
|
|
461 |
|
462 |
-
# ✅ Small Video Display
|
463 |
if video_display_url:
|
|
|
|
|
|
|
464 |
st.markdown(
|
465 |
f"""
|
466 |
<video controls width="300">
|
|
|
434 |
video_stream = yt.streams.get_lowest_resolution()
|
435 |
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
|
436 |
video_stream.download(filename=temp_file.name)
|
437 |
+
return temp_file.name, yt.title, yt.thumbnail_url # Return video title & thumbnail
|
438 |
except Exception as e:
|
439 |
+
return None, None, None
|
440 |
|
441 |
# Select Video Source
|
442 |
video_path = None
|
443 |
video_display_url = None
|
444 |
+
video_title = None
|
445 |
+
video_thumbnail = None
|
446 |
|
447 |
# Process Uploaded Video
|
448 |
if uploaded_video is not None:
|
|
|
451 |
f.write(uploaded_video.read())
|
452 |
video_path = temp_file.name # Set video path for detection
|
453 |
video_display_url = temp_file.name
|
454 |
+
video_title = "Uploaded Video"
|
455 |
|
456 |
# Process Video from URL (Check if YouTube or Direct Link)
|
457 |
elif video_url:
|
458 |
if "youtube.com" in video_url or "youtu.be" in video_url:
|
459 |
+
video_path, video_title, video_thumbnail = download_youtube_video(video_url) # Download YouTube video
|
460 |
video_display_url = video_url
|
461 |
else:
|
462 |
video_path = download_video(video_url) # Download direct MP4 video
|
463 |
video_display_url = video_url
|
464 |
+
video_title = "Online MP4 Video"
|
465 |
|
466 |
+
# ✅ Small Video Display with Title & Thumbnail
|
467 |
if video_display_url:
|
468 |
+
st.markdown(f"### 🎬 {video_title}") # Show video title
|
469 |
+
if video_thumbnail:
|
470 |
+
st.image(video_thumbnail, caption="Video Thumbnail", width=200) # Show thumbnail for YouTube videos
|
471 |
st.markdown(
|
472 |
f"""
|
473 |
<video controls width="300">
|