Discussions

from flask import Flask, request, jsonify, render_template, redirect, url_for
import requests
import os
import time
import logging

# 로깅 설정
logging.basicConfig(level=logging.DEBUG)

app = Flask(__name__)

# HeyGen API 설정
HEYGEN_API_KEY = os.environ.get('HEYGEN_API_KEY', 'your_api_key_here')
AVATAR_ID = os.environ.get('AVATAR_ID', 'your_avatar_id_here')
VOICE_ID = os.environ.get('VOICE_ID', 'your_voice_id_here')

@app.route('/')
def index():
    # 문장 선택 페이지 렌더링
    return render_template('index.html')

@app.route('/video')
def video_page():
    # video_url 파라미터를 받아 video.html 렌더링
    video_url = request.args.get('video_url')
    return render_template('video.html', video_url=video_url)

@app.route('/generate_video', methods=['POST'])
def generate_video():
    try:
        data = request.json
        app.logger.debug(f"Received data: {data}")
        
        text = data.get('texts', [''])[0]
        if not text:
            return jsonify({'error': '텍스트가 제공되지 않았습니다.'}), 400
        
        app.logger.debug(f"HEYGEN_API_KEY: {'Set' if HEYGEN_API_KEY != 'your_api_key_here' else 'NOT SET'}")
        app.logger.debug(f"AVATAR_ID: {'Set' if AVATAR_ID != 'your_avatar_id_here' else 'NOT SET'}")
        app.logger.debug(f"VOICE_ID: {'Set' if VOICE_ID != 'your_voice_id_here' else 'NOT SET'}")

        payload = {
            "caption" : False,
            "video_inputs": [
                {
                    "character": {
                        "type": "avatar",
                        "avatar_id": AVATAR_ID
                    },
                    "voice": {
                        "type": "text",
                        "input_text": text,
                        "voice_id": VOICE_ID,
                        "voice_settings": {
                            "stability": 0.5,
                            "similarity_boost": 0.75
                        }
                    }
                }
            ],
            "dimension": {
                "width": 1280,
                "height": 720
            }
        }

        headers = {
            "accept": "application/json",
            "content-type": "application/json",
            "x-api-key": HEYGEN_API_KEY
        }
        
        app.logger.debug("Sending request to HeyGen API (v2 /video/generate)")
        response = requests.post(
            'https://api.heygen.com/v2/video/generate',
            headers=headers,
            json=payload
        )
        
        app.logger.debug(f"HeyGen API response status: {response.status_code}")
        app.logger.debug(f"HeyGen API response: {response.text}")
        
        if response.status_code != 200:
            return jsonify({'error': f'HeyGen API 오류: {response.text}'}), 500
        
        video_data = response.json()
        video_id = video_data.get('data', {}).get('video_id')
        if not video_id:
            return jsonify({'error': '비디오 ID를 찾을 수 없습니다.'}), 500
        
        # 무한 루프로 상태 폴링 (생성 완료 시까지)
        video_url = poll_video_status(video_id, headers)
        # 생성 완료되면 /video 페이지로 리다이렉트하기 위해 video_url 반환
        return jsonify({'video_url': video_url})
    
    except Exception as e:
        app.logger.error(f'오류 발생: {str(e)}', exc_info=True)
        return jsonify({'error': f'서버 오류: {str(e)}'}), 500

def poll_video_status(video_id, headers, delay=5):
    """비디오 생성 상태 확인 및 완료될 때까지 무한 반복 대기"""
    while True:
        try:
            app.logger.debug("Polling video status")
            status_url = f'https://api.heygen.com/v1/video_status.get?video_id={video_id}'
            app.logger.debug(f"Checking status URL: {status_url}")
            status_response = requests.get(status_url, headers=headers)
            
            app.logger.debug(f"Status response code: {status_response.status_code}")
            app.logger.debug(f"Status response text: {status_response.text}")
            
            if status_response.status_code != 200:
                app.logger.debug(f"Status check failed: {status_response.status_code}")
                time.sleep(delay)
                continue
            
            status_data = status_response.json()
            status = status_data.get('data', {}).get('status')
            app.logger.debug(f"Video status: {status}")
            
            if status == 'completed':
                video_url = status_data['data']['video_url']
                app.logger.debug(f"Video completed, URL: {video_url}")
                return video_url
            
            if status == 'failed':
                app.logger.error(f"Video generation failed: {status_data}")
                raise Exception('비디오 생성에 실패했습니다.')
            
            time.sleep(delay)
            
        except Exception as e:
            app.logger.error(f"폴링 중 오류: {str(e)}", exc_info=True)
            time.sleep(delay)

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5010, debug=True)