Modify

app.py

@@ -30,25 +30,34 @@ examples = [
 
 def search(query):
     shorts = weaviate_explorer.explore(query)
-    download_url = f"https://drive.google.com/uc?id={shorts[0]['link'].split('/')[-2]}"
-    video_path = gdown.download(download_url, output=f"./video.mp4")
-
-    return video_path, "\n\n".join([f"- [{r['title']}]({r['link']})" for r in shorts])
-
-title = "Slideo Explorer: 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 clip will be displayed separately, and it 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")
-
-video = gr.Video(label = "Most relevant clip")
-output_text = gr.Markdown(label = "Top-5 relevants clips")
-
-iface = gr.Interface(fn=search, 
-                     inputs=input_text, 
-                     outputs=[video, output_text],
-                     title=title,
-                     description=description,
-                     examples=[[e] for e in examples],
-                     cache_examples="lazy")
+    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."
+
+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")
+
+iface = gr.Interface(
+    fn=search,
+    inputs=input_text,
+    outputs=[gr.Row(videos), output_text],
+    title=title,
+    description=description,
+    examples=[[e] for e in examples],
+    cache_examples="lazy",
+    theme="dark"
+)
 
 iface.launch(auth=(os.getenv("username"), os.getenv("password")))