Update app.py

app.py

@@ -586,7 +586,58 @@ def process_all_titles(category, style, topic):
         time.sleep(5)  # API 호출 사이에 5초 대기
     
     return "\n".join(results)
+
+
+# 번역 모델 설정
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+
+# 이미지 생성 모델 설정
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-3-medium", torch_dtype=torch.float16)
+pipe = pipe.to(device)
+
+def translate_title(title):
+    translated = translator(title, max_length=128)[0]['translation_text']
+    return translated
+
+def generate_image(prompt):
+    image = pipe(prompt).images[0]
+    image_path = f"generated_image_{uuid.uuid4()}.png"
+    image.save(image_path)
+    return image_path
+
+def process_all_titles(category, style, topic, num_titles):
+    title_suggestions, _ = suggest_title(category, style, topic, "", "", "")
+    titles = title_suggestions.split('\n')
+
+    results = []
+    for title in titles[:num_titles]:
+        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
+
+            # 제목 번역 및 이미지 생성
+            translated_title = translate_title(title)
+            image_path = generate_image(translated_title)
+            
+            # 이미지 링크를 블로그 내용 끝에 추가
+            blog_content += f"\n\n[생성된 이미지]({image_path})"
+
+            # 포스팅 전송
+            send_result = send_to_blogger(title, blog_content)
+            
+            results.append(f"제목: {title}\n전송 결과: {send_result}\n생성된 이미지: {image_path}\n\n")
+        except Exception as e:
+            results.append(f"제목: {title}\n처리 중 오류 발생: {str(e)}\n\n")
         
+        time.sleep(5)  # API 호출 사이에 5초 대기
+    
+    return "\n".join(results)
+
+
 with gr.Blocks() as demo:
     gr.Markdown(f"# {title}")
     
@@ -600,13 +651,17 @@ with gr.Blocks() as demo:
     gr.Markdown("### 3단계 : 블로그 주제, 또는 키워드를 상세히 입력하세요")
     topic = gr.Textbox(label="블로그 주제", placeholder="예시: 8월 국내 여행지 추천")
 
+    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)
+    generated_image = gr.Image(label="생성된 이미지")
 
     start_btn.click(
         fn=process_all_titles,
-        inputs=[category, style, topic],
-        outputs=[result_output]
+        inputs=[category, style, topic, num_titles],
+        outputs=[result_output, generated_image]
     )
 
     category.change(fn=update_prompts_and_description, inputs=[category, style], outputs=[style_description])