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")
    video_player.value = video_path
    print(video_path)
    return video_player, "\n\n".join([f"{r['title']}\n{r['link']}" for r in shorts])

iface = gr.Interface(fn=search, 
                     inputs="text", 
                     outputs=[video_player, "text"])
iface.launch()