Update app.py

app.py

@@ -556,35 +556,25 @@ def generate_images(blog_post):
     conclusion_index = next((i for i, line in enumerate(lines) if '[결론]' in line), -1)
     
     # 2. 도입부에 대한 요약으로 흥미 유발 목적 이미지 생성
-    if intro_index != -1 and body_index != -1:
-        intro_text = ' '.join(lines[intro_index+1:body_index])
-        intro_summary = f"Engaging and intriguing image to spark interest: {intro_text[:100]}... Eye-catching, thought-provoking, visually appealing"
-        images.append(generate_image(intro_summary))
+    intro_text = ' '.join(lines[intro_index+1:body_index]) if intro_index != -1 and body_index != -1 else title
+    intro_summary = f"Engaging and intriguing image to spark interest: {intro_text[:100]}... Eye-catching, thought-provoking, visually appealing"
+    images.append(generate_image(intro_summary))
     
     # 3. 본문 내용 요약으로 이미지 생성
-    if body_index != -1 and conclusion_index != -1:
-        body_text = ' '.join(lines[body_index+1:conclusion_index])
-        body_summary = f"Detailed illustration of main content: {body_text[:100]}... Informative, clear visualization, key points highlighted"
-        images.append(generate_image(body_summary))
+    body_text = ' '.join(lines[body_index+1:conclusion_index]) if body_index != -1 and conclusion_index != -1 else title
+    body_summary = f"Detailed illustration of main content: {body_text[:100]}... Informative, clear visualization, key points highlighted"
+    images.append(generate_image(body_summary))
     
     # 4. 결론 내용 요약으로 독자에게 제공되는 가치와 이익을 이미지로 생성
-    if conclusion_index != -1:
-        conclusion_text = ' '.join(lines[conclusion_index+1:])
-        value_summary = f"Image symbolizing the value and benefits for readers: {conclusion_text[:100]}... Positive outcomes, achievements, growth visualization"
-        images.append(generate_image(value_summary))
+    conclusion_text = ' '.join(lines[conclusion_index+1:]) if conclusion_index != -1 else title
+    value_summary = f"Image symbolizing the value and benefits for readers: {conclusion_text[:100]}... Positive outcomes, achievements, growth visualization"
+    images.append(generate_image(value_summary))
     
     # 5. 결론 내용 요약으로 상징을 이미지로 생성
-    if conclusion_index != -1:
-        symbolic_summary = f"Symbolic representation of the conclusion: {conclusion_text[:100]}... Abstract, metaphorical, thought-provoking imagery"
-        images.append(generate_image(symbolic_summary))
+    symbolic_summary = f"Symbolic representation of the conclusion: {conclusion_text[:100]}... Abstract, metaphorical, thought-provoking imagery"
+    images.append(generate_image(symbolic_summary))
     
-    # 만약 생성된 이미지가 5개 미만이라면, 부족한 만큼 새로운 이미지를 생성
-    while len(images) < 5:
-        random_prompt = f"Supplementary image related to: {title}. Diverse, relevant, complementary visual"
-        images.append(generate_image(random_prompt))
-    
-    return images[:5]  # 최대 5개의 이미지만 반환
-
+    return images  # 항상 5개의 이미지를 반환
 
 with gr.Blocks() as demo:
     gr.Markdown(f"# {title}")