fix: cleanup and small fixes

app/footer.py

@@ -1,8 +1,7 @@
 import streamlit as st
-
-from streamlit.components.v1 import html
-from htbuilder import HtmlElement, div, a, p, img, styles
+from htbuilder import HtmlElement, a, div, img, p, styles
 from htbuilder.units import percent, px
+from streamlit.components.v1 import html
 
 
 def image(src_as_string, **style):
@@ -50,10 +49,7 @@ def layout(*args):
     st.markdown(style, unsafe_allow_html=True)
 
     for arg in args:
-        if isinstance(arg, str):
-            body(arg)
-
-        elif isinstance(arg, HtmlElement):
+        if isinstance(arg, str) or isinstance(arg, HtmlElement):
             body(arg)
 
     st.markdown(str(foot), unsafe_allow_html=True)

app/header.py

@@ -1,10 +1,9 @@
-import streamlit as st
-
-from helpers import switch_page
-from style import CSS
 import logging
 
+import streamlit as st
+from helpers import switch_page
 from streamlit_option_menu import option_menu
+from style import CSS
 
 logging.basicConfig(
     format="%(asctime)s %(levelname)-8s %(message)s",
@@ -17,7 +16,6 @@ def header(logo_and_title=True):
     if "first_run" not in st.session_state:
         st.session_state.first_run = True
         for key in [
-            "search_results",
             "selected_value",
             "filename",
             "executed",
@@ -30,6 +28,8 @@ def header(logo_and_title=True):
             st.session_state[key] = None
         st.session_state.video_options = []
         st.session_state.tot_delay = 0
+    if "search_results" not in st.session_state:
+        st.session_state.search_results = []
     if "page" not in st.session_state:
         st.session_state.page = "Karaoke"
         switch_page(st.session_state.page)

app/helpers.py

@@ -1,6 +1,8 @@
+import json
 import random
+from base64 import b64encode
 from io import BytesIO
-import json
+from pathlib import Path
 
 import matplotlib.pyplot as plt
 import numpy as np
@@ -8,28 +10,11 @@ import requests
 import streamlit as st
 from PIL import Image
 from pydub import AudioSegment
-from base64 import b64encode
-from pathlib import Path
 from streamlit.runtime.scriptrunner import RerunData, RerunException
 from streamlit.source_util import get_pages
 from streamlit_player import st_player
 
 extensions = ["mp3", "wav", "ogg", "flac"]  # we will look for all those file types.
-example_songs = [1, 2, 3]
-
-
-def img_to_bytes(img_path):
-    img_bytes = Path(img_path).read_bytes()
-    encoded = b64encode(img_bytes).decode()
-    return encoded
-
-
-# @st.cache_data(show_spinner=False)
-def img_to_html(img_path):
-    img_html = "<div style='display: flex; justify-content: center; align-items: center; height: 50vh;'><img src='data:image/png;base64,{}' class='img-fluid' style='max-width: 100%; max-height: 100%;' ></div>".format(
-        img_to_bytes(img_path)
-    )
-    return img_html
 
 
 @st.cache_data(show_spinner=False)
@@ -76,8 +61,13 @@ def plot_audio(_audio_segment: AudioSegment, *args, **kwargs) -> Image.Image:
     return image
 
 
+@st.cache_data(show_spinner=False)
+def load_list_of_songs():
+    return json.load(open("sample_songs.json"))
+
+
 def get_random_song():
-    sample_songs = json.load(open("sample_songs.json"))
+    sample_songs = load_list_of_songs()
     name, url = random.choice(list(sample_songs.items()))
     return name, url
 

app/service/youtube.py

@@ -1,12 +1,17 @@
+import logging
 import os
-from typing import List
-import yt_dlp
+import re
 import string
 import time
-import re
+from typing import List
+
 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" % (
@@ -24,7 +29,7 @@ def download_audio_from_youtube(url, output_path):
 
     with yt_dlp.YoutubeDL() as ydl:
         info_dict = ydl.extract_info(url, download=False)
-    if info_dict.get("duration") > 360:
+    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)