Videos in a row

app.py

@@ -30,34 +30,32 @@ examples = [
 
 def search(query):
     shorts = weaviate_explorer.explore(query)
-    video_paths = []
-    markdown_links = []
-    
-    # Assuming you can get up to 3 videos
-    for short in shorts[:3]:
-        download_url = f"https://drive.google.com/uc?id={short['link'].split('/')[-2]}"
-        video_path = gdown.download(download_url, output=f"./video_{shorts.index(short)}.mp4")
-        video_paths.append(video_path)
-        markdown_links.append(f"- [{short['title']}]({short['link']})")
-    
-    return video_paths, "\n\n".join(markdown_links)
-
-title = "Slideo: Your AI-based short video search engine. Based on YC"
-description = "Enter a search query to receive relevant short video answers."
+    videos = []
+    for i in range(3):
+        download_url = f"https://drive.google.com/uc?id={shorts[0]['link'].split('/')[-2]}"
+        video_path = gdown.download(download_url, output=f"./video_{i}.mp4")
+        videos.append(video_path)
+    return [gr.Video(video_path) for video_path in videos], "\n\n".join([f"- [{r['title']}]({r['link']})" for r in shorts])
+
+title = "Slideo: Your AI-based short video search engine based on @YCombinator"
+description = "Enter a search query, question or topic to receive the top 5 most relevant short video answers. The most relevant clips will be displayed separately, and they will also be included in the list of top 5 video titles with links to Google Drive."
 
 input_text = gr.Textbox(placeholder="Your query, question, topic, ...", label="Query")
-videos = [gr.Video(label=f"Clip {i+1}") for i in range(3)]
-output_text = gr.Markdown(label="Relevant clips")
+
+videos_row = gr.Row()
+output_text = gr.Markdown(label="Top-5 relevant clips")
 
 iface = gr.Interface(
     fn=search,
     inputs=input_text,
-    outputs=[gr.Row(videos), output_text],
+    outputs=[videos_row, output_text],
     title=title,
     description=description,
     examples=[[e] for e in examples],
     cache_examples="lazy",
-    theme="dark"
+    theme="dark-gradients",
+    layout="vertical",
 )
 
+iface.dependencies[0]["show_progress"] = "hidden"
 iface.launch(auth=(os.getenv("username"), os.getenv("password")))