app.py

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):
    # GPT-4 모델을 사용하여 입력 텍스트에 기반한 키워드 생성 요청
    response = openai.Completion.create(
        model="text-davinci-004",  # GPT-4 모델 지정 (현재 GPT-4 모델명이 'text-davinci-004'라고 가정)
        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
    )
    # 생성된 키워드 반환
    return response.choices[0].text.strip()

# 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="text",
    outputs="text",
    title="Generate Pexels Search Keyword with GPT-4",
    description="Enter some text and generate a relevant English keyword for searching on Pexels."
)

iface.launch()