Update app.py

app.py

@@ -527,19 +527,19 @@ def generate_images(blog_post):
     title = lines[0].strip()
     
     # 1. 제목을 요약하여 이미지 생성
-    images.append(generate_image(f"Summary of '{title}'"))
+    title_prompt = f"Summary of '{title}'"
+    images.append(generate_image(title_prompt))
     
     # 도입부와 본문, 결론 찾기
     intro_index = next((i for i, line in enumerate(lines) if '[도입부]' in line), -1)
     body_index = next((i for i, line in enumerate(lines) if '[본론1]' in line), -1)
     conclusion_index = next((i for i, line in enumerate(lines) if '[결론]' in line), -1)
     
-    # 2. 도입부 요약으로 이미지 2장 생성
+    # 2. 도입부 요약으로 이미지 1장 생성
     if intro_index != -1 and body_index != -1:
         intro_text = ' '.join(lines[intro_index+1:body_index])
         intro_summary = f"Summary of introduction: {intro_text[:100]}..."
         images.append(generate_image(intro_summary))
-        images.append(generate_image(intro_summary))
     
     # 3. 본문 요약으로 이미지 생성
     if body_index != -1 and conclusion_index != -1:
@@ -558,11 +558,12 @@ def generate_images(blog_post):
         value_summary = f"Image symbolizing the value and benefit for readers: {conclusion_text[:100]}..."
         images.append(generate_image(value_summary))
     
-    # 만약 생성된 이미지가 5개 미만이라면, 부족한 만큼 마지막 이미지를 복제
+    # 만약 생성된 이미지가 5개 미만이라면, 부족한 만큼 새로운 이미지를 생성
     while len(images) < 5:
-        images.append(images[-1] if images else "")
+        random_prompt = f"Random image related to: {title}"
+        images.append(generate_image(random_prompt))
     
-    return images
+    return images[:5]  # 최대 5개의 이미지만 반환: 제목 1장, 도입부 1장, 본문 1장, 결론 2장 = 5장
 
 
 with gr.Blocks() as demo: