Update app.py

app.py

@@ -7,23 +7,33 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 
 def generate_keyword_from_text(input_text):
     try:
-        # OpenAI ChatCompletion 을 사용하여 챗 모델 요청
+        # OpenAI ChatCompletion 을 사용하여 한 줄의 키워드 요청
         response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo", 
-            messages=[
-                {"role": "user", "content": f"다음 텍스트에 대한 Pexels 검색을 위한 영문 키워드 생성: '{input_text}'"}
-            ]
+            model="gpt-3.5-turbo",
+            messages=[{
+                'role': 'user',
+                'content': f"다음 텍스트를 바탕으로 Pexels 에서 잘 검색될 영문 키워드 한 줄을 생성하세요: '{input_text}'"
+            }]
         )
-        keyword = response['choices'][0]['message']['content']
+        # 첫 번째 choice에서 message content를 받아옴
+        keyword_full_response = response['choices'][0]['message']['content']
+        
+        # 가장 첫 번째 라인만 추출
+        keyword = keyword_full_response.split('\n', 1)[0].strip()
+        
+        # '""'로 감싼 키워드 제거
+        if keyword.startswith('"') and keyword.endswith('"'):
+            keyword = keyword[1:-1].strip()
+
         return keyword
     except Exception as e:
         print(f"에러 발생: {e}")
-        return "키워드 생성 중 에러 발생"
+        return f"키워드 생성 중 에러 발생: {e}"
 
 # Gradio 인터페이스 설정과 실행
 iface = gr.Interface(
     fn=generate_keyword_from_text,
-    inputs=gr.Textbox(lines=2, label="텍스트 입력"),
+    inputs=gr.Textbox(lines=2, placeholder="여기에 Pexels 검색을 위한 텍스트를 입력하세요."),
     outputs="text",
     title="GPT를 이용한 Pexels 검색 키워드 생성",
     description="제공된 텍스트를 바탕으로 Pexels 검색에 사용할 영문 키워드를 자동 생성합니다."