app.py

from youtube_worksheet import YouTubeWorksheet
from dotenv import load_dotenv
import gradio as gr
import os

def process_video(url):
    # 환경변수에서 API 키 가져오기
    API_KEY = os.getenv('GEMINI_API_KEY')
    
    if not API_KEY:
        return "ERROR: GEMINI_API_KEY가 설정되지 않았습니다.", None
    
    worksheet = YouTubeWorksheet(API_KEY)
    
    # 자막 추출
    transcript = worksheet.get_transcript(url)
    if not transcript:
        return "자막을 추출할 수 없습니다.", None
    
    # 워크시트 생성
    content = worksheet.create_worksheet(transcript)
    
    # DOCX 파일로 저장
    output_file = worksheet.save_to_docx(content)
    
    return f"워크시트가 생성되었습니다. 파일명: {output_file}", output_file

# Gradio 인터페이스 생성
load_dotenv()
iface = gr.Interface(
    fn=process_video,
    inputs=[gr.Textbox(label="YouTube URL을 입력하세요", placeholder="https://www.youtube.com/watch?v=...")],
    outputs=[
        gr.Textbox(label="처리 결과"),
        gr.File(label="생성된 워크시트")
    ],
    title="YouTube 학습 워크시트 생성기",
    description="YouTube 영상의 자막을 이용하여 학습 워크시트를 생성합니다."
)

if __name__ == "__main__":
    iface.launch()