Update app.py

app.py

@@ -5,38 +5,31 @@ import os
 # OpenAI API 키 설정
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def generate_keyword_from_text(input_text):
-    try:
-        # OpenAI ChatCompletion 을 사용하여 한 줄의 키워드 요청
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=[{
-                'role': 'user',
-                'content': f"다음 텍스트를 바탕으로 Pexels 에서 잘 검색될 영문 키워드 한 줄을 생성하세요: '{input_text}'"
-            }]
-        )
-        # 첫 번째 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()
+def generate_keywords_and_search_images(input_text):
+    # 입력 텍스트를 줄별로 분리
+    lines = input_text.strip().split('\n')
+    all_image_urls = []
 
-        return keyword
-    except Exception as e:
-        print(f"에러 발생: {e}")
-        return f"키워드 생성 중 에러 발생: {e}"
+    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
 
 # Gradio 인터페이스 설정과 실행
 iface = gr.Interface(
-    fn=generate_keyword_from_text,
-    inputs=gr.Textbox(lines=2, placeholder="여기에 Pexels 검색을 위한 텍스트를 입력하세요."),
-    outputs="text",
-    title="GPT를 이용한 Pexels 검색 키워드 생성",
-    description="제공된 텍스트를 바탕으로 Pexels 검색에 사용할 영문 키워드를 자동 생성합니다."
+    fn=generate_keywords_and_search_images,
+    inputs=gr.Textbox(lines=5, placeholder="여기에 Pexels 검색을 위한 텍스트를 입력하세요. 각 줄마다 별도의 검색 키워드가 생성됩니다."),
+    outputs=gr.Gallery(),
+    title="GPT 및 Pexels API를 이용한 다중 이미지 검색",
+    description="제공된 텍스트의 각 줄을 바탕으로 Pexels 검색에 사용할 영문 키워드를 자동 생성하고, 해당 키워드로 Pexels에서 이미지를 검색합니다."
 )
 
 iface.launch()