Update app.py

app.py

@@ -9,10 +9,6 @@ import random
 import os
 from huggingface_hub import InferenceClient
 from fpdf import FPDF
-from transformers import pipeline
-import torch
-from diffusers import StableDiffusionXLPipeline
-import uuid
 
 def setup_session():
     try:
@@ -207,7 +203,6 @@ def get_title_prompt(category):
 2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
 3. 특히 너는 '정보성(Informative)' 전문 블로그 마케팅 전문가이다.
 4. 정보 제공에 초점을 맞추어 작성한다. 
-
 ##[블로그 제목 작성 규칙]
 1. 블로그 제목 10개를 작성하고 제목 10개만 출력하라.
 2. 제목은 40자 이내로 작성하라.
@@ -223,7 +218,6 @@ def get_title_prompt(category):
 2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
 3. 특히 너는 '건강, 의학 정보' 전문 블로그 마케터이다.
 4. 정확하고 전문적인 정보 제공에 초점을 맞추어 작성한다.
-
 ##[블로그 제목 작성 규칙]
 1. 블로그 제목 10개를 작성하고 제목 10개만 출력하라.
 2. 제목은 40자 이내로 작성하라.
@@ -241,7 +235,6 @@ def get_outline_prompt(category):
 2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
 3. 특히 너는 '정보성(Informative)' 전문 블로그 마케팅 전문가이다.
 4. 정보 제공에 초점을 맞추어 작성한다. 
-
 ##[소주제 작성규칙]
 1. [기본규칙]을 기본 적용하라.
 2. 블로그 글을 작성하기 위한 소주제를 작성하라.
@@ -251,7 +244,6 @@ def get_outline_prompt(category):
 6. 독자가 얻고자 하는 정보와 흥미로운 정보를 제공하도록 소주제를 작성하라.
 7. 소주제의 본론의 내용이 충분히 작성될 수 있는 소주제로 설정하라. 
 8. 반드시 [소주제 구성]에 맞게 출력하라.
-
 ##[소주제 구성]
 1. 반드시 [도입부] - 1개, [본론1~5] - 5개, [결론] - 1개로 구성하여 출력하라.
 2. 반드시 [도입부]와 [결론]의 제목이 중복되지 않도록 작성하라.
@@ -264,7 +256,6 @@ def get_outline_prompt(category):
 2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
 3. 특히 너는 '건강, 의학 정보' 전문 블로그 마케터이다.
 4. 정확하고 전문적인 정보 제공에 초점을 맞추어 작성한다.
-
 ##[소주제 작성규칙]
 1. [기본규칙]을 기본 적용하라.
 2. 블로그 글을 작성하기 위한 소주제를 작성하라.
@@ -274,7 +265,6 @@ def get_outline_prompt(category):
 6. 독자가 얻고자 하는 정확한 정보와 건강한 생활을 유지하는 데 필요한 정보를 제공하도록 소주제를 작성하라.
 7. 소주제의 본론의 내용이 충분히 작성될 수 있는 소주제로 설정하라. 
 8. 반드시 [소주제 구성]에 맞춰 소주제만 출력하라.
-
 ##[소주제 구성]
 1. 반드시 [도입부] - 1개, [본론1~5] - 5개, [결론] - 1개로 구성하여 출력하라.
 2. 반드시 [도입부]와 [결론]의 제목이 중복되지 않도록 작성하라.
@@ -289,7 +279,6 @@ def get_blog_post_prompt(category):
 2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
 3. 특히 너는 '정보성(Informative)' 전문 블로그 마케팅 전문가이다.
 4. 정보 제공에 초점을 맞추어 작성한다. 
-
 ##[텍스트 작성 규칙]
 1. 반드시 입력된 [소주제]에 맞게 텍스트를 작성하라.
 2. 소주제의 [본론] 5개를 각각 300자 이상으로 작성하라.
@@ -301,13 +290,11 @@ def get_blog_post_prompt(category):
 8. 주제와 상황에 맞는 적절한 어휘를 선택하라. 
 9. 한글 어휘의 난이도는 쉽게 작성하라.
 10. 절대 문장의 끝에 '답니다'를 사용하지 말 것.
-
 	###[정보성 블로그 작성 규칙]
 	1. 독자가 얻고자 하는 유용한 정보와 흥미로운 정보를 제공하도록 작성하라.
 	2. 독자의 공감을 이끌어내고 궁금증을 해결하도록 작성하라.
 	3. 독자의 관심사를 충족시키도록 작성하라.
 	4. 독자에게 이득이 되는 정보를 작성하라.
-
 ##[제외 규칙]
 1. 반드시 비속어 및 욕설(expletive, abusive language, slang)은 제외하라.
 2. 반드시 참고글의 링크(URL)는 제외하라.
@@ -323,7 +310,6 @@ def get_blog_post_prompt(category):
 2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
 3. 특히 너는 '건강, 의학 정보' 전문 블로그 마케터이다.
 4. 정확하고 전문적인 정보 제공에 초점을 맞추어 작성한다.
-
 ##[텍스트 작성 규칙]
 1. 반드시 입력된 [소주제]에 맞춰서 텍스트를 작성하라.
 2. 반드시 입력된 [소주제]는 변경하지 말고 그대로 출력하라.
@@ -335,13 +321,11 @@ def get_blog_post_prompt(category):
 7. 어투는 제공된 참고글의 어투를 반영하되, '건강, 의학 정보' 전문 블로그 마케터로서 작성하라.
 8. 절대로 참고글을 한문장 이상 그대로 출력하지 말 것.
 9. 주제와 상황에 맞는 적절�� 어휘를 선택하라. 
-
 	###[정보성 블로그 작성 규칙]
 	1. 독자가 얻고자 하는 정확한 정보와 건강한 생활을 유지하는 데 필요한 정보를 제공하도록 소주제에 맞게 내용을 작성하라.
 	2. 독자의 공감을 이끌어내고 궁금증을 해결하도록 작성하라.
 	3. 독자의 관심사를 충족시키도록 작성하라.
 	4. 독자에게 이득이 되는 정보를 작성하라.
-
 ##[제외 규칙]
 1. 반드시 참고글의 링크(URL)는 제외하라.
 2. 참고글에서 '링크를 확인해주세요'와 같은 링크 이동의 문구는 제외하라.
@@ -510,60 +494,44 @@ def update_prompts_and_description(category, style):
     style_description = get_style_description(style)
     return style_description
 
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
-
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-pipe = StableDiffusionXLPipeline.from_pretrained(
-    "SG161222/RealVisXL_V4.0",
-    torch_dtype=torch.float16,
-    use_safetensors=True,
-    add_watermarker=False,
-    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']
+def generate_images(blog_post):
+    images = []
+    lines = blog_post.split('\n')
+    title = lines[0].strip()
     
-    # Hi-res와 3840x2160 스타일 적용
-    prompt = f"hyper-realistic 8K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic"
+    # 1. 제목을 요약하여 이미지 생성
+    images.append(generate_image(f"Summary of '{title}'"))
     
-    # 고정된 설정값
-    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)
+    # 도입부와 본문, 결론 찾기
+    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)
     
-    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]
+    # 2. 도입부 요약으로 이미지 2장 생성
+    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:
+        body_text = ' '.join(lines[body_index+1:conclusion_index])
+        body_summary = f"Summary of main content: {body_text[:100]}..."
+        images.append(generate_image(body_summary))
+    
+    # 4. 결론 요약으로 상징적 이미지 생성
+    if conclusion_index != -1:
+        conclusion_text = ' '.join(lines[conclusion_index+1:])
+        conclusion_summary = f"Symbolic image representing: {conclusion_text[:100]}..."
+        images.append(generate_image(conclusion_summary))
+    
+    # 5. 결론 요약으로 가치와 이익을 상징하는 이미지 생성
+    if conclusion_index != -1:
+        value_summary = f"Image symbolizing the value and benefit for readers: {conclusion_text[:100]}..."
+        images.append(generate_image(value_summary))
     
-    unique_name = str(uuid.uuid4()) + ".png"
-    image.save(unique_name)
-    return unique_name
-
-def generate_images(blog_post):
-    images = []
-    lines = blog_post.split('\n')
-    for i in range(5):
-        if i < len(lines):
-            prompt = lines[i]
-        else:
-            prompt = random.choice(lines)
-        image_path = generate_image(prompt)
-        images.append(image_path)
     return images
 
 
@@ -619,14 +587,6 @@ with gr.Blocks() as demo:
         outputs=[references1, references2, references3, outline_result, outline_token_output, output, token_output]
     )
 
-    save_pdf_btn = gr.Button("PDF로 저장하기")
-    pdf_output = gr.File(label="생성된 PDF 파일")
-
-    save_pdf_btn.click(fn=save_content_to_pdf, inputs=[output], outputs=[pdf_output])
-
-    category.change(fn=update_prompts_and_description, inputs=[category, style], outputs=[style_description])
-    style.change(fn=update_prompts_and_description, inputs=[category, style], outputs=[style_description])    
-
     # 이미지 생성 버튼 및 출력 추가
     generate_images_btn = gr.Button("이미지 생성하기")
     image_outputs = [gr.Image(label=f"생성된 이미지 {i+1}") for i in range(5)]
@@ -635,8 +595,8 @@ with gr.Blocks() as demo:
         fn=generate_images,
         inputs=[output],
         outputs=image_outputs
-    )
-
+    )    
+    
     save_pdf_btn = gr.Button("PDF로 저장하기")
     pdf_output = gr.File(label="생성된 PDF 파일")