Add a loading gif and a quality selector for downloading the video

app.py

@@ -30,6 +30,8 @@ def get_embedding(txt_model, vis_model, query, video):
 
 def find_frames(url, txt_model, vis_model, desc, seconds, top_k):
     text = st.text("Downloading video (Descargando video)...")
+    # gif from https://giphy.com/gifs/alan-DfSXiR60W9MVq
+    gif_runner = st.image("./loading.gif")
     probe = ffmpeg.probe(url)
     video_stream = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None)
     width = int(video_stream['width'])
@@ -51,8 +53,9 @@ def find_frames(url, txt_model, vis_model, desc, seconds, top_k):
     txt_embd, img_embds = get_embedding(txt_model, vis_model, desc, video)
     cos_scores = np.array(util.cos_sim(txt_embd, img_embds))
     ids = np.argsort(cos_scores)[0][-top_k:]
-
     imgs = [Image.fromarray(video[i]) for i in ids]
+
+    gif_runner.empty()
     text.empty()
     st.image(imgs)
 
@@ -87,6 +90,7 @@ def clifs_page(txt_model, vis_model):
         "Top K",
         min_value=1,
         max_value=5,
+        value=3,
         step=1,
     )
     desc = st.sidebar.text_input(
@@ -99,19 +103,24 @@ def clifs_page(txt_model, vis_model):
         value='https://youtu.be/xUv6XgPwGaQ',
         help="Youtube video you want to search (Video de Youtube que desea búscar)",
     )
+    quality = st.sidebar.radio(
+        "Quality of the Video (Calidad del Video)",
+        [144, 240, 360, 480],
+        help="Quality of the video to download. Higher quality takes more time (Calidad del video para descargar. Calidad más alta toma más tiempo)",
+    )
 
     submit_button = st.sidebar.button("Search (Búscar)")
     if submit_button:
-        ydl_opts = {"format": "mp4[height=360]"}
+        ydl_opts = {"format": f"mp4[height={quality}]"}
         with youtube_dl.YoutubeDL(ydl_opts) as ydl:
             info_dict = ydl.extract_info(url, download=False)
             video_url = info_dict.get("url", None)
             find_frames(video_url, txt_model, vis_model, desc, seconds, top_k)
 
 PAGES = {
+    "CLIFS": clifs_page,
     "Home": main_page,
     "Inicio": inicio_pagina,
-    "CLIFS": clifs_page
 }