Update app.py

app.py

@@ -5,23 +5,20 @@ import os
 # OpenAI API 키 설정
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def generate_keywords_and_search_images(input_text):
-    # 입력 텍스트를 줄별로 분리
-    lines = input_text.strip().split('\n')
-    all_image_urls = []
-
-    for line in lines:
-        # 각 줄에 대한 키워드 생성
-        keyword = generate_keyword_from_text(line)
-        if keyword.startswith("키워드 생성 중 에러 발생"):
-            print(keyword)
-            continue  # 키워드 생성 실패 시, 다음 줄로 넘어감
-
-        # 생성된 키워드로 Pexels 검색
-        image_urls = search_pexels(keyword)
-        all_image_urls.extend(image_urls)  # 검색된 이미지 URL들을 전체 목록에 추가
-
-    return all_image_urls
+def generate_keyword_from_text(input_text):
+    try:
+        response = openai.Completion.create(
+            model="gpt-3.5-turbo",  # 모델명 확인 필요
+            prompt=f"다음 텍스트를 바탕으로 Pexels에서 잘 검색될 영문 키워드 한 줄을 생성하세요: '{input_text}'",
+            temperature=0.5,
+            max_tokens=10
+        )
+        keyword_full_response = response.choices[0].text
+        keyword = keyword_full_response.split('\n', 1)[0].strip()
+        return keyword
+    except Exception as e:
+        print(f"에러 발생: {e}")
+        return f"키워드 생성 중 에러 발생: {e}"
 
 # Gradio 인터페이스 설정과 실행
 iface = gr.Interface(