#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
YouTube Shorts Maker - 웹 인터페이스
Flask 기반 간단한 웹 UI
"""
from flask import Flask, render_template_string, request, jsonify, send_file
import os
from pathlib import Path
import threading
from youtube_shorts_maker import YouTubeShortsConverter
app = Flask(__name__)
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 # 16MB max
# 전역 변수
converter = YouTubeShortsConverter(output_dir="./web_shorts_output")
processing_status = {}
# HTML 템플릿
HTML_TEMPLATE = """
YouTube Shorts Maker
🎬 YouTube Shorts Maker
유튜브 동영상을 1분 쇼츠로 자동 변환
📌 예제 URL (테스트용)
Me at the zoo (짧은 동영상)
"""
@app.route('/')
def index():
"""메인 페이지"""
return render_template_string(HTML_TEMPLATE)
@app.route('/convert', methods=['POST'])
def convert_video():
"""동영상 변환 API"""
try:
data = request.get_json()
url = data.get('url')
duration = data.get('duration', 60)
watermark = data.get('watermark', False)
if not url:
return jsonify({
'success': False,
'message': 'URL이 필요합니다.'
}), 400
# 변환 실행
shorts_path = converter.convert_url_to_shorts(
youtube_url=url,
target_duration=duration,
add_watermark=watermark,
keep_temp=False
)
filename = os.path.basename(shorts_path)
return jsonify({
'success': True,
'message': f'쇼츠가 성공적으로 생성되었습니다! ({duration}초)',
'filename': filename,
'path': shorts_path
})
except Exception as e:
return jsonify({
'success': False,
'message': f'오류: {str(e)}'
}), 500
@app.route('/download/')
def download_file(filename):
"""파일 다운로드"""
try:
file_path = converter.output_dir / filename
if not file_path.exists():
return jsonify({
'success': False,
'message': '파일을 찾을 수 없습니다.'
}), 404
return send_file(
file_path,
as_attachment=True,
download_name=filename,
mimetype='video/mp4'
)
except Exception as e:
return jsonify({
'success': False,
'message': f'다운로드 오류: {str(e)}'
}), 500
def main():
"""웹 서버 실행"""
print("=" * 60)
print("🌐 YouTube Shorts Maker - 웹 인터페이스")
print("=" * 60)
print("\n📍 브라우저에서 http://localhost:5000 접속")
print("🛑 종료하려면 Ctrl+C 누르기\n")
app.run(
host='0.0.0.0',
port=5000,
debug=True
)
if __name__ == '__main__':
main()