Update app.py

app.py

@@ -511,37 +511,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     variant="fp16"
 ).to(device)
 
-def generate_image(prompt: str):
-    # 한글 입력 감지 및 번역
-    if any('\uac00' <= char <= '\ud7a3' for char in prompt):
-        translated = translator(prompt, max_length=512)
-        prompt = translated[0]['translation_text']
-    
-    # Hi-res와 3840x2160 스타일 적용
-    prompt = f"hyper-realistic 8K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic"
-    
-    # 고정된 설정값
-    negative_prompt = "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly, (deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, disgusting, amputation"
-    width = 1024
-    height = 1024
-    guidance_scale = 6
-    num_inference_steps = 100
-    seed = random.randint(0, 2**32 - 1)
-    generator = torch.Generator().manual_seed(seed)
-    
-    image = pipe(
-        prompt=prompt,
-        negative_prompt=negative_prompt,
-        width=width,
-        height=height,
-        guidance_scale=guidance_scale,
-        num_inference_steps=num_inference_steps,
-        generator=generator,
-    ).images[0]
-    
-    unique_name = str(uuid.uuid4()) + ".png"
-    image.save(unique_name)
-    return unique_name
+
 
 def generate_images(blog_post):
     images = []
@@ -580,6 +550,10 @@ 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개 미만이라면, 부족한 만큼 마지막 이미지를 복제
+    while len(images) < 5:
+        images.append(images[-1] if images else "")
+    
     return images
 
 
@@ -635,15 +609,14 @@ with gr.Blocks() as demo:
         outputs=[references1, references2, references3, outline_result, outline_token_output, output, token_output]
     )
 
-    # 이미지 생성 버튼 및 출력 추가
-    generate_images_btn = gr.Button("이미지 생성하기")
     image_outputs = [gr.Image(label=f"생성된 이미지 {i+1}") for i in range(5)]
-
     generate_images_btn.click(
         fn=generate_images,
         inputs=[output],
         outputs=image_outputs
     )
+    
+
 
     save_pdf_btn = gr.Button("PDF로 저장하기")
     pdf_output = gr.File(label="생성된 PDF 파일")