asad231 commited on
Commit
9ddaa85
·
verified ·
1 Parent(s): 8024e38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
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 using HTML + CSS
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">