app.py

import gradio as gr
import requests
import json

API_URL = "https://api.openai.com/v1/chat/completions"

# 이미지 생성용 프롬프트를 작성하는 함수
def create_image_prompts(script, openai_api_key):
    # 스크립트의 각 줄을 분석하여 프롬프트 생성 (여기서는 단순화된 예시로 구현)
    lines = script.split('\n')
    prompts = []
    for line in lines:
        # 주요 행위 키워드 추출 (여기서는 단순화된 형태로 구현)
        keyword = "adventure"  # 예시 키워드, 실제 구현에서는 분석을 통해 추출
        prompt = f"3d style, 4K, like Brad Pitt young boy, {keyword}"
        prompts.append(prompt)
    return "\n".join(prompts)

def predict(inputs, top_p, temperature, openai_api_key):
    # OpenAI API를 사용하여 스크립트 생성
    narration_prompt = f"아동용 애니메이션 동영상에 사용할 스크립트를 작성하라. 반드시 한글로 작성할것. 일체의 지문이나 지시어, 설명 부호 등을 포함시키지 말것.  입력: '{inputs}'"
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {openai_api_key}"
    }
    payload = {
        "model": "gpt-4-1106-preview",
        "messages": [{"role": "system", "content": narration_prompt}],
        "temperature": temperature,
        "top_p": top_p,
        "n": 1,
        "max_tokens": 1000
    }
    response = requests.post(API_URL, headers=headers, json=payload)
    if response.status_code == 200:
        response_data = response.json()
        generated_text = response_data['choices'][0]['message']['content']
        return generated_text
    else:
        return "Error: Unable to generate response."

def create_detailed_image_prompts(script):
    # 스크립트의 각 줄을 분석하여 프롬프트 생성
    lines = script.split('\n')
    detailed_prompts = []
    for line in lines:
        # 여기서는 예시로 몇 가지 간단한 패턴을 사용합니다.
        # 실제 구현에서는 더 복잡한 자연어 처리를 적용해야 할 수 있습니다.
        if "야구장에 갔다" in line:
            action = "go to the baseball (background)"
            prompt = f"3d style, 4K, like Brad Pitt young boy, {action}"
        elif "숲을 탐험했다" in line:
            action = "explore the forest (background)"
            prompt = f"3d style, 4K, like Brad Pitt young boy, {action}"
        else:
            # 일치하는 패턴이 없을 경우 기본 프롬프트 사용
            prompt = f"3d style, 4K, like Brad Pitt young boy, adventure"
        detailed_prompts.append(prompt)
    return "\n".join(detailed_prompts)

# Gradio 인터페이스 수정
with gr.Blocks() as demo:
    gr.Markdown("<h1 align='center'>토리의 모험: 3D 애니메이션 생성기</h1>")
    with gr.Row():
        openai_api_key = gr.Textbox(type='password', label="Enter your OpenAI API key here")
        inputs = gr.Textbox(placeholder="여기에 입력하세요.", label="아동용 애니메이션 스크립트를 생성하고 싶은 주제어나 문장을 입력하세요.")
        top_p = gr.Slider(minimum=0, maximum=1.0, value=1.0, step=0.05, label="Top-p (nucleus sampling)")
        temperature = gr.Slider(minimum=0, maximum=5.0, value=1.0, step=0.1, label="Temperature")
    output = gr.Textbox(label="Generated Script", readonly=True)
    prompts_output = gr.TextArea(label="Image Generation Prompts", readonly=True)
    submit_button = gr.Button("Generate Script")
    prompts_button = gr.Button("Generate Image Prompts")

    def generate_prompts(script):
        return create_detailed_image_prompts(script)

    submit_button.click(fn=predict, inputs=[inputs, top_p, temperature, openai_api_key], outputs=output)
    prompts_button.click(fn=generate_prompts, inputs=[output], outputs=prompts_output)

demo.launch()