Update app.py

app.py

@@ -1,22 +1,29 @@
+import gradio as gr
 import openai
 import os
 
-# OpenAI API 키 설정
-openai.api_key = os.getenv("OPENAI_API_KEY")  # 환경 변수에서 API 키를 불러옵니다.
+# 환경 변수에서 OpenAI API 키를 불러옵니다.
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def generate_keywords_with_gpt(scene_descriptions):
-    responses = []
-    for description in scene_descriptions:
-        response = openai.Completion.create(
-            engine="text-davinci-003",  # 혹은 다른 GPT 모델을 선택할 수 있습니다.
-            prompt=f"Generate a representative keyword for the following scene description: {description}",
-            max_tokens=60,  # 필요한 토큰 수를 조절할 수 있습니다.
-            temperature=0.7
+def generate_keywords(scene_description):
+    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
+    )
+    keyword = response.choices[0].text.strip()
+    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]
         )
-        responses.append(response.choices[0].text.strip())
-    return responses
+    return demo
 
-# 사용 예시
-scene_descriptions = ["Enter description for Scene 1", "Enter description for Scene 2"]
-keywords = generate_keywords_with_gpt(scene_descriptions)
-print(keywords)
+app = gradio_app()
+app.launch()