import gradio as gr
from ultralytics import YOLO
from pytubefix import YouTube  # pytube-fix 사용
import os
import cv2
from typing import Tuple
from datetime import datetime


def download_youtube_video(url, output_path="./"):
    yt = YouTube(url)
    video_stream = yt.streams.filter(file_extension="mp4", res="720p").first()
    if not video_stream:
        video_stream = yt.streams.get_highest_resolution()
    os.makedirs(output_path, exist_ok=True)
    video_path = os.path.join(output_path, yt.title + ".mp4")
    video_stream.download(output_path)
    return video_path


def greet(name):
    working_directory = os.getcwd()
    return f" {working_directory}!!"


# demo = gr.Interface(fn=greet, 
#                     inputs="text", outputs="text")
# demo.launch()

with gr.Blocks() as app:
    working_directory = os.getcwd()
    with gr.Row():
        gr.Textbox(f'{working_directory}')
    with gr.Row():
        gr.Interface(fn=download_youtube_video, inputs='text', output='text')