Ubuntu
commited on
Commit
·
162a7ae
1
Parent(s):
02b5df3
update lgic for id
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def download_youtube_audio(url: str, output_dir: Optional[str] = None) -> str:
|
|
15 |
output_dir = tempfile.gettempdir()
|
16 |
|
17 |
yt_dlp_command = [
|
18 |
-
"
|
19 |
"-f", "bestaudio/best",
|
20 |
"-x", # Extract audio
|
21 |
"--audio-format", "mp3",
|
@@ -80,6 +80,8 @@ def embed_youtube(youtube_url):
|
|
80 |
try:
|
81 |
#video_id = YoutubeDL().extract_info(youtube_url, download=False)['id']\
|
82 |
video_id = youtube_url.split("v=")[1]
|
|
|
|
|
83 |
print(video_id)
|
84 |
embed_html = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>'
|
85 |
return gr.update(value=embed_html, visible=True), "", None
|
|
|
15 |
output_dir = tempfile.gettempdir()
|
16 |
|
17 |
yt_dlp_command = [
|
18 |
+
"yt-dlp",
|
19 |
"-f", "bestaudio/best",
|
20 |
"-x", # Extract audio
|
21 |
"--audio-format", "mp3",
|
|
|
80 |
try:
|
81 |
#video_id = YoutubeDL().extract_info(youtube_url, download=False)['id']\
|
82 |
video_id = youtube_url.split("v=")[1]
|
83 |
+
if 'shorts' in youtube_url:
|
84 |
+
video_id = youtube_url.split('/')[-1]
|
85 |
print(video_id)
|
86 |
embed_html = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>'
|
87 |
return gr.update(value=embed_html, visible=True), "", None
|