Fix

app.py

@@ -30,31 +30,27 @@ examples = [
 
 def search(query):
     shorts = weaviate_explorer.explore(query)
-    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_row = gr.Component(gr.Row(), name="Videos")
-output_text = gr.Markdown(label="Top-5 relevant clips")
-
-iface = gr.Interface(
-    fn=search,
-    inputs=input_text,
-    outputs=[videos_row, output_text],
-    title=title,
-    description=description,
-    examples=[[e] for e in examples],
-    cache_examples="lazy",
-    theme="dark-gradients"
-)
+    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",
+                     theme="dark-gradients")
 
 iface.dependencies[0]["show_progress"] = "hidden"
 iface.launch(auth=(os.getenv("username"), os.getenv("password")))