import openai
import os

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

def generate_keyword_from_text(input_text):
    try:
        response = openai.Completion.create(
            model="text-davinci-003",  # GPT-4 모델명을 적절히 선택하세요
            prompt=f"Given the following text, generate a relevant English keyword for Pexels search: '{input_text}'",
            temperature=0.5,
            max_tokens=10,
            top_p=1.0,
            frequency_penalty=0.0,
            presence_penalty=0.0
        )
        keyword = response.choices[0].text.strip()
        return keyword
    except Exception as e:
        print(f"An error occurred: {e}")
        return "Error generating keyword"

# Gradio 인터페이스 예시
import gradio as gr

def gradio_interface(input_text):
    keyword = generate_keyword_from_text(input_text)
    return keyword

iface = gr.Interface(
    fn=gradio_interface,
    inputs=gr.inputs.Textbox(lines=2, label="Enter Text"),
    outputs="text",
    title="Generate Pexels Search Keyword with GPT",
    description="This tool generates a keyword for Pexels search based on the provided text input."
)

iface.launch()