video / app.py
seawolf2357's picture
Update app.py
24742bd verified
raw
history blame
1.14 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:
# μ΅œμ‹  OpenAI APIλ₯Ό μ‚¬μš©ν•˜μ—¬ ν…μŠ€νŠΈ μ™„μ„± μš”μ²­
response = openai.Completion.create(
model="text-davinci-003", # μ‚¬μš© κ°€λŠ₯ν•œ GPT-3 λͺ¨λΈ
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 μΈν„°νŽ˜μ΄μŠ€ μ„€μ •
import gradio as gr
iface = gr.Interface(
fn=generate_keyword_from_text,
inputs=gr.Textbox(lines=2, label="ν…μŠ€νŠΈ μž…λ ₯"),
outputs="text",
title="GPTλ₯Ό μ΄μš©ν•œ Pexels 검색 ν‚€μ›Œλ“œ 생성",
description="제곡된 ν…μŠ€νŠΈλ₯Ό λ°”νƒ•μœΌλ‘œ Pexels 검색에 μ‚¬μš©ν•  영문 ν‚€μ›Œλ“œλ₯Ό μžλ™ μƒμ„±ν•©λ‹ˆλ‹€."
)
# μΈν„°νŽ˜μ΄μŠ€ μ‹€ν–‰
iface.launch()