Spaces:
Sleeping
Sleeping
| 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']}\n{r['link']}" for r in shorts]) | |
| title = "Slideo Explorer: Your AI-based short video search engine" | |
| description = "Write down a search query, question or a topic and receive top-5 most relevant short video-answers. The most relevant clip will be displayed in the widget and the whole list will be available in another widget as a list of titles, links to Google Drive." | |
| iface = gr.Interface(fn=search, | |
| inputs="text", | |
| outputs=["playable_video", "text"], | |
| title=title, | |
| description=description) | |
| iface.launch() |