app.py

import os
import gdown
import gradio as gr
from weaviate_explorer import WeaviateExplorer

weaviate_explorer = WeaviateExplorer()
# video_player = gr.Video(label="Search Results")

downloads_dir = "downloads"  # Directory to store downloaded videos

# Create the downloads directory if it doesn't exist
os.makedirs(downloads_dir, exist_ok=True)

def remove_downloaded_videos():
    """Remove all downloaded videos from the 'downloads' directory."""
    for file_name in os.listdir(downloads_dir):
        file_path = os.path.join(downloads_dir, file_name)
        if os.path.isfile(file_path):
            os.remove(file_path)

def search(query):
    # remove_downloaded_videos()  # Remove previously downloaded videos
    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"
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)
iface.launch()