Update app.py

app.py

@@ -626,58 +626,46 @@ def convert_markdown_images(content):
     return content
 
 
-def process_all_titles(category, style, topic, num_titles):
+def process_all_titles(category, style, topic, num_titles, progress=gr.Progress()):
     title_suggestions, _ = suggest_title(category, style, topic, "", "", "")
     titles = title_suggestions.split('\n')
     
-    # 빈 문자열이나 '###' 등의 불필요한 문자열 제거
     titles = [title.strip() for title in titles if title.strip() and not title.strip().startswith('###')]
-    
-    # "블로그"와 "제목"이라는 단어가 포함된 제목을 제거
     titles = [title for title in titles if "블로그" not in title and "제목" not in title]
 
     results = []
-    for title in titles[:num_titles]:
+    for i, title in enumerate(titles[:num_titles], 1):
+        progress(i / num_titles, desc=f"처리 중: {i}/{num_titles}")
         try:
-
-            
-            # 첫 번째 이미지 URL 가져오기
             image_url_1 = get_image_url(topic)
-            
-            # 두 번째 이미지 URL 가져오기
             image_url_2 = get_image_url(topic)
             
-            # 블로그 내용 생성
             _, _, _, _, _, 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
-
-            if image_url_1:
-                img_tag_1 = f'<img src="{image_url_1}" alt="{topic} 1" style="max-width:100%; height:auto;">'
-                blog_content = f'{img_tag_1}\n\n{blog_content}'
-
-            # 두 번째 이미지 URL을 블로그 내용 끝 부분에 추가
-            if image_url_2:
-                img_tag_2 = f'<img src="{image_url_2}" alt="{topic} 2" style="max-width:100%; height:auto;">'
-                blog_content = f'{blog_content}\n\n{img_tag_2}'
-            
+                result = f"제목: {title}\n생성 실패: {blog_content}\n\n"
+            else:
+                if image_url_1:
+                    blog_content = f'<img src="{image_url_1}" alt="{topic} 1" style="max-width:100%; height:auto;">\n\n{blog_content}'
+                if image_url_2:
+                    blog_content = f'{blog_content}\n\n<img src="{image_url_2}" alt="{topic} 2" style="max-width:100%; height:auto;">'
 
+                send_result = send_to_blogger(title, blog_content)
+                result = f"제목: {title}\n전송 결과: {send_result}\n사용된 이미지 URL 1: {image_url_1 if image_url_1 else '이미지 없음'}\n사용된 이미지 URL 2: {image_url_2 if image_url_2 else '이미지 없음'}\n\n"
             
-            # 포스팅 전송
-            send_result = send_to_blogger(title, blog_content)
-            
-            results.append(f"제목: {title}\n전송 결과: {send_result}\n사용된 이미지 URL 1: {image_url_1 if image_url_1 else '이미지 없음'}\n사용된 이미지 URL 2: {image_url_2 if image_url_2 else '이미지 없음'}\n\n")
+            results.append(result)
+            yield f"진행 상황: {i}/{num_titles}\n\n" + "\n".join(results)
         except Exception as e:
             logger.error("제목 '%s' 처리 중 오류 발생: %s", title, str(e))
-            results.append(f"제목: {title}\n처리 중 오류 발생: {str(e)}\n\n")
+            result = f"제목: {title}\n처리 중 오류 발생: {str(e)}\n\n"
+            results.append(result)
+            yield f"진행 상황: {i}/{num_titles}\n\n" + "\n".join(results)
         
-        time.sleep(5)  # API 호출 사이에 5초 대기
+        time.sleep(5)
     
-    return "\n".join(results)
+    yield f"완료: {num_titles}/{num_titles}\n\n" + "\n".join(results)
+
 
-import html
 
 def send_to_blogger(blog_title, blog_content):
     if not WEBHOOK_URL:
@@ -723,8 +711,8 @@ with gr.Blocks() as demo:
     gr.Markdown("### 3단계 : 블로그 주제, 또는 키워드를 상세히 입력하세요")
     topic = gr.Textbox(label="블로그 주제", placeholder="예시: 8월 국내 여행지 추천")
 
-    gr.Markdown("### 4단계 : 자동 생성 및 전송할 블로그 수(계정당 일일 포스팅 업로드 제한 60건까지 )")
-    num_titles = gr.Slider(minimum=1, maximum=60, value=2, step=1, label="포스팅 건")
+    gr.Markdown("### 4단계 : 자동 생성 및 전송할 블로그 수를 선택하세요")
+    num_titles = gr.Slider(minimum=1, maximum=100, value=2, step=1, label="포스팅 건")
 
     start_btn = gr.Button("시작")
     result_output = gr.Textbox(label="처리 결과", lines=20)