app.py

import gradio as gr
import requests
import time

API_HOME = 'https://api.fliki.ai/v1'
END_GENERATE = '/generate'
END_STATUS = '/generate/status'
API_KEY = 'ZZUIQ4OZASNRQ8B8WYHNW'

def set_header():
    header = {
        "Content-Type": "application/json",
        "Authorization": "Bearer " + API_KEY,
    }
    return header

# 비디오 또는 오디오 컨텐츠 생성하는 함수
def generate_contents(user_content, brand_name, aspect_ratio='landscape'):
    url = API_HOME + END_GENERATE
    VOICE_ID = '64ea14306a00ac991dcb8a3f'  # Korean, Korea, Wolf Sea

    # 멀티 라인 텍스트를 분리하여 각 줄을 별도의 씬으로 처리
    scenes = [{"content": line.strip(), "voiceId": VOICE_ID} for line in user_content.split('\n') if line.strip()]

    payload = {
        "format": "video",  # video | audio
        "scenes": scenes,
        "settings": {
            'aspectRatio': aspect_ratio,
            'subtitle': {
                'fontColor': 'yellow',
                'backgroundColor': 'black',
                'placement': 'bottom',
                'display': 'phrase',
            },
        },
        "backgroundMusicKeywords": "happy, lofi, beats"
    }
    
    headers = set_header()
    response = requests.request("POST", url, json=payload, headers=headers).json()

    if response['success']:
        return response['data']['id']
    else:
        return 'FAILED'

# 컨텐츠 생성 상태를 확인하는 함수
def generate_status(content_id):
    url = API_HOME + END_STATUS
    payload = {"id": content_id}
    headers = set_header()

    while True:
        response = requests.request("POST", url, json=payload, headers=headers).json()

        if response['success']:
            status = response['data']['status']
            if status == 'queued' or status == 'processing':
                time.sleep(10)  # 10초 대기
            elif status == 'success':
                return response['data']['file']
        else:
            break
    return 'FAILED'

# Gradio 인터페이스 함수
def gradio_generate_contents(user_content, brand_name, aspect_ratio):
    return generate_contents(user_content, brand_name, aspect_ratio)

def gradio_generate_status(content_id):
    download_url = generate_status(content_id)
    if download_url != 'FAILED':
        display_text = download_url[:10] + '...' if len(download_url) > 10 else download_url
        html_link = f'<a href="{download_url}" target="_blank">{display_text}</a>'
        html_video = f'<iframe src="{download_url}" width="640" height="360" frameborder="0" allowfullscreen></iframe>'
        return html_link + '<br>' + html_video
    else:
        return "컨텐츠 생성 실패 또는 ID가 잘못되었습니다."

# 컨텐츠 생성 인터페이스
iface_create = gr.Interface(
    fn=gradio_generate_contents,
    inputs=[
        gr.TextArea(label="컨텐츠 텍스트"),
        gr.Textbox(label="브랜드명"),
        gr.Dropdown(label="화면 비율", choices=['portrait', 'square', 'landscape'], value='landscape')
    ],
    outputs=gr.Textbox(label="컨텐츠 ID")
)

# 상태 확인 및 비디오 다운로드 인터페이스
iface_status = gr.Interface(
    fn=gradio_generate_status,
    inputs=gr.Textbox(label="컨텐츠 ID"),
    outputs=gr.HTML(label="다운로드 링크 및 비디오")
)

# 두 인터페이스를 탭으로 구성
iface_combined = gr.TabbedInterface([iface_create, iface_status], ["컨텐츠 생성", "상태 확인 및 다운로드"])
iface_combined.launch()