| import logging |
| import os |
| import re |
| import string |
| from typing import List |
|
|
| from loguru import logger as log |
| import streamlit as st |
| import yt_dlp |
| from pytube import Search |
|
|
| logger = logging.getLogger("pytube") |
| logger.setLevel(logging.ERROR) |
|
|
|
|
| def _sanitize_filename(filename): |
| safe_chars = "-_.() %s%s" % ( |
| re.escape(string.ascii_letters), |
| re.escape(string.digits), |
| ) |
| safe_filename = re.sub(f"[^{safe_chars}]", "_", filename) |
| return safe_filename.strip() |
|
|
|
|
| @st.cache_data(show_spinner=False) |
| def download_audio_from_youtube(url, output_path): |
| if not os.path.exists(output_path): |
| os.makedirs(output_path) |
|
|
| with yt_dlp.YoutubeDL({"quiet": True}) as ydl: |
| info_dict = ydl.extract_info(url, download=False) |
| if info_dict.get("duration", 0) > 360: |
| st.error("Song is too long. Please use a song no longer than 6 minutes.") |
| return |
| video_title = info_dict.get("title", None) |
| video_title = _sanitize_filename(video_title) |
| ydl_opts = { |
| "format": "bestaudio/best", |
| "postprocessors": [ |
| { |
| "key": "FFmpegExtractAudio", |
| "preferredcodec": "mp3", |
| "preferredquality": "192", |
| } |
| ], |
| "outtmpl": os.path.join(output_path, video_title), |
| "quiet": True, |
| } |
| with yt_dlp.YoutubeDL(ydl_opts) as ydl: |
| ydl.download([url]) |
| return f"{video_title}.mp3" |
|
|
|
|
| @st.cache_data(show_spinner=False, max_entries=10) |
| def query_youtube(query: str) -> Search: |
| return Search(query) |
|
|
|
|
| def search_youtube(query: str, limit=5) -> List: |
| log.info(f"{query}") |
| if len(query) > 3: |
| search = query_youtube(query + " lyrics") |
| st.session_state.search_results = search.results |
| if "search_results" in st.session_state and st.session_state.search_results is not None: |
| video_options = [video.title for video in st.session_state.search_results[:limit]] |
| else: |
| video_options = [] |
| else: |
| video_options = [] |
| st.session_state.video_options = video_options |
| return video_options |
|
|
|
|
| def get_youtube_url(title: str) -> str: |
| video = st.session_state.search_results[st.session_state.video_options.index(title)] |
| return video.embed_url |
|
|
|
|
| def check_if_is_youtube_url(url: str) -> bool: |
| return url.startswith("http") |
|
|
