Update app.py

app.py

@@ -17,6 +17,16 @@ import json
 from huggingface_hub.utils._errors import HfHubHTTPError
 from gradio_client import Client
 from urllib.parse import urljoin
+import requests
+from pexels_api import API
+import logging
+
+# 로깅 설정
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+pexels_api =  "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"
 
 def setup_session():
     try:
@@ -546,55 +556,6 @@ def update_prompts_and_description(category, style):
 WEBHOOK_URL = os.getenv("WEBHOOK_URL") 
 BLOGGER_ID = os.getenv("BLOGGER_ID")
 
-def send_to_blogger(blog_title, blog_content):
-    payload = {
-        "id": BLOGGER_ID,
-        "title": blog_title,
-        "content": blog_content
-    }
-    
-    try:
-        response = requests.post(WEBHOOK_URL, json=payload)
-        if response.status_code == 200:
-            return "포스팅이 성공적으로 전송되었습니다."
-        else:
-            return f"포스팅 전송 실패. 상태 코드: {response.status_code}"
-    except Exception as e:
-        return f"오류 발생: {str(e)}"
-
-
-def process_all_titles(category, style, topic):
-    # 제목 추천
-    title_suggestions, _ = suggest_title(category, style, topic, "", "", "")
-    titles = title_suggestions.split('\n')
-
-    results = []
-    for title in titles[:10]:  # 처음 10개의 제목만 사용
-        try:
-            # 블로그 글 생성
-            _, _, _, _, _, blog_content, _ = fetch_references_and_generate_all_steps(category, style, topic, title)
-            
-            if blog_content.startswith("API 호출 실패"):
-                results.append(f"제목: {title}\n생성 실패: {blog_content}\n\n")
-                continue
-
-            # 포스팅 전송
-            send_result = send_to_blogger(title, blog_content)
-            
-            results.append(f"제목: {title}\n전송 결과: {send_result}\n\n")
-        except Exception as e:
-            results.append(f"제목: {title}\n처리 중 오류 발생: {str(e)}\n\n")
-        
-        time.sleep(5)  # API 호출 사이에 5초 대기
-    
-    return "\n".join(results)
-
-
-BASE_URL = "http://hugpu.ai:7899"  # 실제 서버 주소로 변경하세요
-
-# Pexels API 클라이언트 설정
-PEXELS_API_KEY = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"
-pexels_api = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"
 
 def get_image_url(topic):
     # 주제에서 핵심 키워드 추출 (여기서는 간단히 처음 두 단어를 사용)
@@ -609,13 +570,12 @@ def get_image_url(topic):
             # 고해상도 이미지 URL 반환
             return photos[0].original
         else:
-            print(f"'{keywords}' 관련 이미지를 찾을 수 없습니다.")
+            logger.warning(f"'{keywords}' 관련 이미지를 찾을 수 없습니다.")
             return None
     except Exception as e:
-        print(f"이미지 검색 중 오류 발생: {str(e)}")
+        logger.error(f"이미지 검색 중 오류 발생: {str(e)}")
         return None
 
-# process_all_titles 함수 수정
 def process_all_titles(category, style, topic, num_titles):
     title_suggestions, _ = suggest_title(category, style, topic, "", "", "")
     titles = title_suggestions.split('\n')
@@ -641,14 +601,38 @@ def process_all_titles(category, style, topic, num_titles):
             # 포스팅 전송
             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 if image_url else '이미지 없음'}\n\n")
         except Exception as e:
+            logger.error(f"제목 '{title}' 처리 중 오류 발생: {str(e)}")
             results.append(f"제목: {title}\n처리 중 오류 발생: {str(e)}\n\n")
         
         time.sleep(5)  # API 호출 사이에 5초 대기
     
     return "\n".join(results)
 
+def send_to_blogger(blog_title, blog_content):
+    payload = {
+        "id": BLOGGER_ID,
+        "title": blog_title,
+        "content": blog_content
+    }
+    
+    try:
+        # URL 유효성 검사
+        if "![관련 이미지](None)" in blog_content:
+            blog_content = blog_content.replace("![관련 이미지](None)", "관련 이미지를 찾을 수 없습니다.")
+        
+        payload["content"] = blog_content
+        
+        response = requests.post(WEBHOOK_URL, json=payload)
+        if response.status_code == 200:
+            return "포스팅이 성공적으로 전송되었습니다."
+        else:
+            return f"포스팅 전송 실패. 상태 코드: {response.status_code}"
+    except Exception as e:
+        logger.error(f"블로거에 전송 중 오류 발생: {str(e)}")
+        return f"오류 발생: {str(e)}"
+
     
 with gr.Blocks() as demo:
     gr.Markdown(f"# {title}")
@@ -664,7 +648,7 @@ with gr.Blocks() as demo:
     topic = gr.Textbox(label="블로그 주제", placeholder="예시: 8월 국내 여행지 추천")
 
     gr.Markdown("### 4단계 : 자동 생성 및 전송할 블로그 수(주제는 따르되, 각기 다른 제목과 내용으로 구성)를 선택하세요")
-    num_titles = gr.Slider(minimum=1, maximum=100, value=2, step=1, label="생성할 제목 수")
+    num_titles = gr.Slider(minimum=1, maximum=100, value=2, step=1, label="생성/전송할 블로그그 수")
 
     start_btn = gr.Button("시작")
     result_output = gr.Textbox(label="처리 결과", lines=20)