video / app.py
seawolf2357's picture
Update app.py
1dc42f8 verified
raw
history blame
1.2 kB
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:
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo", # ν˜„μž¬ μ‚¬μš© κ°€λŠ₯ν•œ GPT λͺ¨λΈλͺ…μœΌλ‘œ μ—…λ°μ΄νŠΈν•˜μ„Έμš”.
messages=[{"role": "system", "content": "Generate a relevant English keyword for the following description."},
{"role": "user", "content": f"{input_text}"}]
)
keyword = response['choices'][0]['message']['content']
return keyword.strip()
except Exception as e:
print(f"An error occurred: {e}")
return "Error generating keyword"
def gradio_interface(input_text):
keyword = generate_keyword_from_text(input_text)
return keyword
# Gradio μΈν„°νŽ˜μ΄μŠ€ μ„€μ •
import gradio as gr
iface = gr.Interface(
fn=gradio_interface,
inputs=gr.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()