import gradio as gr
import openai
import os

# OpenAI API 키 설정
openai.api_key = os.getenv("OPENAI_API_KEY")

def generate_keyword_from_text(input_text):
    try:
        # 최신 OpenAI API를 사용하여 텍스트 생성 요청
        response = openai.Completion.create(
            model="gpt-3.5-turbo",  # 사용 가능한 최신 GPT 모델 사용
            prompt=f"다음 텍스트에 대한 Pexels 검색을 위한 영문 키워드 생성: '{input_text}'",
            temperature=0.5,
            max_tokens=10
        )
        keyword = response.choices[0].text.strip()
        return keyword
    except Exception as e:
        print(f"에러 발생: {e}")
        return "키워드 생성 중 에러 발생"

# Gradio 인터페이스 설정과 실행
iface = gr.Interface(
    fn=generate_keyword_from_text,
    inputs=gr.Textbox(lines=2, label="텍스트 입력"),
    outputs="text",
    title="GPT를 이용한 Pexels 검색 키워드 생성",
    description="제공된 텍스트를 바탕으로 Pexels 검색에 사용할 영문 키워드를 자동 생성합니다."
)

iface.launch()