Update app.py

app.py

@@ -592,24 +592,27 @@ def process_all_titles(category, style, topic):
 
 BASE_URL = "http://hugpu.ai:7899"  # 실제 서버 주소로 변경하세요
 
-def generate_image(prompt):
+# Pexels API 클라이언트 설정
+PEXELS_API_KEY = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"
+pexels_api = API(PEXELS_API_KEY)
+
+def get_image_url(topic):
+    # 주제에서 핵심 키워드 추출 (여기서는 간단히 처음 두 단어를 사용)
+    keywords = ' '.join(topic.split()[:2])
+    
     try:
-        result = image_client.predict(
-            prompt,
-            api_name="/generate"
-        )
-        # API 응답에서 이미지 경로를 추출합니다.
-        image_path = result[0] if isinstance(result, list) else result
+        # Pexels API를 사용하여 이미지 검색
+        pexels_api.search(keywords, page=1, results_per_page=1)
+        photos = pexels_api.get_entries()
         
-        # 로컬 파일 경로를 웹 URL로 변환
-        if image_path.startswith('/'):
-            image_url = urljoin(BASE_URL, image_path)
+        if photos:
+            # 고해상도 이미지 URL 반환
+            return photos[0].original
         else:
-            image_url = image_path
-        
-        return image_url
+            print(f"'{keywords}' 관련 이미지를 찾을 수 없습니다.")
+            return None
     except Exception as e:
-        print(f"이미지 생성 중 오류 발생: {str(e)}")
+        print(f"이미지 검색 중 오류 발생: {str(e)}")
         return None
 
 # process_all_titles 함수 수정
@@ -626,20 +629,19 @@ def process_all_titles(category, style, topic, num_titles):
                 results.append(f"제목: {title}\n생성 실패: {blog_content}\n\n")
                 continue
 
-            # 이미지 생성
-            image_url = generate_image(title)
-
+            # Pexels API를 사용하여 이미지 URL 가져오기
+            image_url = get_image_url(topic)
+            
             if image_url:
                 # 이미지 URL을 블로그 내용 끝에 추가
-                blog_content += f"\n\n![생성된 이미지]({image_url})"
+                blog_content += f"\n\n![관련 이미지]({image_url})"
             else:
-                blog_content += "\n\n이미지 생성에 실패했습니다."
+                blog_content += "\n\n관련 이미지를 찾을 수 없습니다."
 
             # 포스팅 전송
             send_result = send_to_blogger(title, blog_content)
-        
-            results.append(f"제목: {title}\n전송 결과: {send_result}\n생성된 이미지 URL: {image_url}\n\n")
-
+            
+            results.append(f"제목: {title}\n전송 결과: {send_result}\n사용된 이미지 URL: {image_url}\n\n")
         except Exception as e:
             results.append(f"제목: {title}\n처리 중 오류 발생: {str(e)}\n\n")
         
@@ -647,6 +649,7 @@ def process_all_titles(category, style, topic, num_titles):
     
     return "\n".join(results)
 
+    
 with gr.Blocks() as demo:
     gr.Markdown(f"# {title}")