Update app.py

app.py

@@ -1,24 +1,25 @@
-import gradio as gr
 import openai
 import os
 
-# 환경 변수에서 OpenAI API 키를 불러옵니다.
+# 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()
+    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
@@ -29,10 +30,10 @@ def gradio_interface(input_text):
 
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs="text",
+    inputs=gr.inputs.Textbox(lines=2, label="Enter 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."
+    title="Generate Pexels Search Keyword with GPT",
+    description="This tool generates a keyword for Pexels search based on the provided text input."
 )
 
-iface.launch()
+iface.launch()