import gradio as gr
import requests
import time

API_HOME = 'https://api.fliki.ai/v1'
END_LANGUAGES = '/languages'
END_DIALECTS = '/dialects'
END_VOICES = '/voices'
END_GENERATE = '/generate'
END_STATUS = '/generate/status'
API_KEY = 'ZZUIQ4OZASNRQ8B8WYHNW'
brand = '파인애플 노트북'

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

# 기존의 함수들...


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

# 언어 리스트를 가져오는 함수.
def get_language_id():
    url = API_HOME + END_LANGUAGES
    headers = set_header()
    response = requests.request("GET", url, headers=headers).json()
    print(f"언어 리스트: {response}")
    # Korean: 61b8b2fa4268666c126babf1
    return response


# 언어지역 리스트를 가져오는 함수.
def get_dialects():
    url = API_HOME + END_DIALECTS
    headers = set_header()
    response = requests.request("GET", url, headers=headers).json()
    print(f"언어지역 리스트: {response}")
    # Korea: 61b8b30f4268666c126bac8d
    return response


# 지역언어 리스트를 가져오는 함수.
def get_voices(language_id, dialect_id):
    url = API_HOME + END_VOICES
    headers = set_header()
    payload = {
        'languageId': language_id,
        'dialectId': dialect_id
    }
    response = requests.request("POST", url, json=payload, headers=headers).json()
    print(f"voice: {response}")
    return response


# 비디오 또는 오디오 컨텐츠 생성하는 함수.

def generate_contents(user_content, brand_name):
    global brand
    brand = brand_name  # 브랜드명 설정
    url = API_HOME + END_GENERATE
    VOICE_ID = '65934ac2bc02ab1c006755fa'  # Korean, Korea, Wolf Sea

    # 각 문장을 별도의 scene으로 처리
    scenes = [{"content": line, "voiceId": VOICE_ID} for line in user_content.split('\n') if line.strip()]

    payload = {
        "format": "video",  # video | audio
        "scenes": scenes,
        "settings": {
            # ... 설정들 ...
        },
        "backgroundMusicKeywords": "happy, lofi, beats"
    }
    
    # ... 나머지 코드 ...
    
    headers = set_header()
    response = requests.request("POST", url, json=payload, headers=headers).json()

    # HTTP 상태코드 확인
    if response['success'] == True:
        # 응답값 예시: {'success': True, 'data': {'id': '6593aa11d153762d8aa022e9'}}
        print(f"정상: {response['data']['id']}")
        return response['data']['id']
    else:
        print(f"request 실패: {response}")
        return 'FAILED'


# /generate API 호출 성공 후 받은 id로 컨텐츠 생성 상태를 확인하는 함수.
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']:
            if response['data']['status'] == 'queued' or response['data']['status'] == 'processing':
                print('컨텐츠 생성 중... 잠시 후 다시 시도하세요.')
                time.sleep(10)  # 10초 대기
            elif response['data']['status'] == 'success':
                print('컨텐츠 생성 완료!')
                return response['data']['file']
        else:
            print(f"request 실패: {response}")
            break
    return 'FAILED'

def gradio_generate_status(content_id):
    download_url = generate_status(content_id)
    if download_url != 'FAILED':
        return download_url, requests.get(download_url).content
    else:
        return "컨텐츠 생성 실패 또는 ID가 잘못되었습니다.", None

status_iface = gr.Interface(
    fn=gradio_generate_status,
    inputs=gr.Textbox(label="컨텐츠 ID"),
    outputs=[
        gr.Textbox(label="다운로드 URL"),
        gr.File(label="다운로드 버튼")
    ]
)

status_iface.launch()


def download_content(url):
    data = requests.get(url, timeout=30)
    save_path = "downloaded_example.mp4"
    with open(save_path, 'wb') as file:
        for chunk in data.iter_content(chunk_size=128):
            file.write(chunk)
        print(f"MP4 파일이 성공적으로 저장되었습니다. 경로: {save_path}")
    return 'success'


# # 실행.

# get_language_id() # Korean: 61b8b2fa4268666c126babf1
# get_dialects() # Korea: 61b8b30f4268666c126bac8d
# get_voices('61b8b2fa4268666c126babf1', '61b8b30f4268666c126bac8d')

try:
    id = generate_contents()
    if id != 'FAILED':
        status = 'PENDING'
        while status == 'PENDING':
            time.sleep(9)  # 9초 기다림
            print('9초 대기 후 상태확인 함수를 호출합니다.')
            status = generate_status(id)
            if status != 'PENDING': break

        print(f"컨텐츠 생성 성공: {status}")
    else:
        print('생성 실패..')
except Exception as e:
    print(f"오류 발생: {e}")

def gradio_generate_contents(user_content, brand_name):
    return generate_contents(user_content, brand_name)

def gradio_generate_status(content_id):
    return generate_status(content_id)

# ... 이전 코드 ...

# Gradio 인터페이스 수정
iface = gr.Interface(
    fn=gradio_generate_contents,  # 컨텐츠 생성 함수
    inputs=[gr.TextArea(label="컨텐츠 텍스트"), gr.Textbox(label="브랜드명")],  # 멀티라인 텍스트 입력 및 브랜드명 입력
    outputs=gr.Textbox(label="컨텐츠 ID")
)

status_iface = gr.Interface(
    fn=gradio_generate_status,  # 컨텐츠 상태 확인 함수
    inputs=gr.Textbox(label="컨텐츠 ID"),
    outputs=gr.Textbox(label="컨텐츠 상태 또는 URL")    
)

# 인터페이스 실행
iface.launch()
status_iface.launch()