video / app.py
seawolf2357's picture
Update app.py
79eed57 verified
raw
history blame
No virus
1.2 kB
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()