Update app.py

app.py

@@ -5,26 +5,34 @@ import os
 # 환경 변수에서 OpenAI API 키를 불러옵니다.
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def generate_keywords(scene_description):
+
+def generate_keyword_from_text(input_text):
+    # GPT-4 모델을 사용하여 입력 텍스트에 기반한 키워드 생성 요청
     response = openai.Completion.create(
-        model="text-davinci-003",  # 사용할 모델을 지정합니다.
-        prompt=f"Generate a representative English keyword for the following scene description: {scene_description}",
-        max_tokens=60,
-        temperature=0.7
+        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
     )
-    keyword = response.choices[0].text.strip()
+    # 생성된 키워드 반환
+    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
-    
-# Gradio 앱 정의
-def gradio_app():
-    with gr.Blocks() as demo:
-        with gr.Row():
-            scene_description = gr.Textbox(label="Scene Description")
-            keyword_output = gr.Textbox(label="Generated Keyword")
-        gr.Button("Generate Keyword").click(
-            generate_keywords, inputs=[scene_description], outputs=[keyword_output]
-        )
-    return demo
 
-app = gradio_app()
-app.launch()
+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()