Update app.py

app.py

@@ -8,6 +8,7 @@ import time
 import random
 import os
 from huggingface_hub import InferenceClient
+from fpdf import FPDF
 
 def setup_session():
     try:
@@ -112,26 +113,29 @@ def call_api(content, system_message, max_tokens, temperature, top_p):
     total_tokens = response.usage.total_tokens
     return modified_text, input_tokens, output_tokens, total_tokens
 
-def analyze_info(category, topic, references1, references2, references3):
-    return f"선택한 카테고리: {category}\n블로그 주제: {topic}\n참고 글1: {references1}\n참고 글2: {references2}\n참고 글3: {references3}"
+def analyze_info(category, style, topic, references1, references2, references3):
+    return f"선택한 카테고리: {category}\n선택한 포스팅 스타일: {style}\n블로그 주제: {topic}\n참고 글1: {references1}\n참고 글2: {references2}\n참고 글3: {references3}"
 
-def suggest_title(category, topic, references1, references2, references3, system_message, max_tokens, temperature, top_p):
-    full_content = analyze_info(category, topic, references1, references2, references3)
-    modified_text, input_tokens, output_tokens, total_tokens = call_api(full_content, system_message, max_tokens, temperature, top_p)
+def suggest_title(category, style, topic, references1, references2, references3, category_prompt, style_prompt, max_tokens, temperature, top_p):
+    full_content = analyze_info(category, style, topic, references1, references2, references3)
+    combined_prompt = f"{category_prompt}\n\n{style_prompt}"
+    modified_text, input_tokens, output_tokens, total_tokens = call_api(full_content, combined_prompt, max_tokens, temperature, top_p)
     token_usage_message = f"[입력 토큰수: {input_tokens}]\n[출력 토큰수: {output_tokens}]\n[총 토큰수: {total_tokens}]"
     return modified_text, token_usage_message
 
-def generate_outline(category, topic, references1, references2, references3, title, system_message, max_tokens, temperature, top_p):
-    full_content = analyze_info(category, topic, references1, references2, references3)
+def generate_outline(category, style, topic, references1, references2, references3, title, category_prompt, style_prompt, max_tokens, temperature, top_p):
+    full_content = analyze_info(category, style, topic, references1, references2, references3)
     content = f"{full_content}\nTitle: {title}"
-    modified_text, input_tokens, output_tokens, total_tokens = call_api(content, system_message, max_tokens, temperature, top_p)
+    combined_prompt = f"{category_prompt}\n\n{style_prompt}"
+    modified_text, input_tokens, output_tokens, total_tokens = call_api(content, combined_prompt, max_tokens, temperature, top_p)
     token_usage_message = f"[입력 토큰수: {input_tokens}]\n[출력 토큰수: {output_tokens}]\n[총 토큰수: {total_tokens}]"
     return modified_text, token_usage_message
 
-def generate_blog_post(category, topic, references1, references2, references3, title, outline, system_message, max_tokens, temperature, top_p):
-    full_content = analyze_info(category, topic, references1, references2, references3)
+def generate_blog_post(category, style, topic, references1, references2, references3, title, outline, category_prompt, style_prompt, max_tokens, temperature, top_p):
+    full_content = analyze_info(category, style, topic, references1, references2, references3)
     content = f"{full_content}\nTitle: {title}\nOutline: {outline}"
-    modified_text, input_tokens, output_tokens, total_tokens = call_api(content, system_message, max_tokens, temperature, top_p)
+    combined_prompt = f"{category_prompt}\n\n{style_prompt}"
+    modified_text, input_tokens, output_tokens, total_tokens = call_api(content, combined_prompt, max_tokens, temperature, top_p)
     formatted_text = modified_text.replace('\n', '\n\n')
     token_usage_message = f"[입력 토큰수: {input_tokens}]\n[출력 토큰수: {output_tokens}]\n[총 토큰수: {total_tokens}]"
     return formatted_text, token_usage_message
@@ -152,7 +156,7 @@ def fetch_references(topic):
     
     return "참고글 생성 완료", references1_content, references2_content, references3_content
 
-def fetch_references_and_generate_all_steps(category, topic, blog_title, system_message_outline, max_tokens_outline, temperature_outline, top_p_outline, system_message_blog_post, max_tokens_blog_post, temperature_blog_post, top_p_blog_post):
+def fetch_references_and_generate_all_steps(category, style, topic, blog_title, system_message_outline, max_tokens_outline, temperature_outline, top_p_outline, system_message_blog_post, max_tokens_blog_post, temperature_blog_post, top_p_blog_post):
     search_url = generate_naver_search_url(topic)
     session = setup_session()
     if session is None:
@@ -168,10 +172,10 @@ def fetch_references_and_generate_all_steps(category, topic, blog_title, system_
     references3_content = f"제목: {selected_results[2]['제목']}\n내용: {crawl_blog_content(selected_results[2]['링크'], session)}"
     
     # 아웃라인 생성
-    outline_result, outline_token_usage = generate_outline(category, topic, references1_content, references2_content, references3_content, blog_title, system_message_outline, max_tokens_outline, temperature_outline, top_p_outline)
+    outline_result, outline_token_usage = generate_outline(category, style, topic, references1_content, references2_content, references3_content, blog_title, system_message_outline, max_tokens_outline, temperature_outline, top_p_outline)
     
     # 블로그 글 생성
-    blog_post_result, blog_post_token_usage = generate_blog_post(category, topic, references1_content, references2_content, references3_content, blog_title, outline_result, system_message_blog_post, max_tokens_blog_post, temperature_blog_post, top_p_blog_post)
+    blog_post_result, blog_post_token_usage = generate_blog_post(category, style, topic, references1_content, references2_content, references3_content, blog_title, outline_result, system_message_blog_post, max_tokens_blog_post, temperature_blog_post, top_p_blog_post)
     
     return references1_content, references2_content, references3_content, outline_result, outline_token_usage, blog_post_result, blog_post_token_usage
 
@@ -205,24 +209,183 @@ def get_blog_post_prompt(category):
 # 블로그 텍스트 생성 규칙(건강정보)
 """
 
-# Gradio 인터페이스 구성
+def get_style_prompt(style):
+    prompts = {
+        "친근한": """
+#친근한 블로그 포스팅 스타일 가이드
+1. 톤과 어조
+   - 대화하듯 편안하고 친근한 말투 사용
+   - 독자를 "여러분" 또는 "독자님들"로 지칭
+   - 적절한 이모지를 sparse하게 사용하여 친근감 표현
+2. 문장 및 내용 구성
+   - 짧고 간결한 문장 위주로 작성
+   - 구어체 표현 사용 (예: "~했어요", "~인 것 같아요")
+   - '~니다', '~했죠'는 사용하지 말 것.
+   - 개인적인 경험이나 일화로 시작
+   - 실생활에서 쉽게 적용할 수 있는 팁 제공
+   - 독자의 공감을 얻을 수 있는 사례 포함
+3. 용어 및 설명 방식
+   - 전문 용어 대신 쉬운 단어로 풀어서 설명
+   - 비유나 은유를 활용하여 복잡한 개념 설명
+   - 수사의문문 활용하여 독자와 소통하는 느낌 주기
+4. 독자와의 상호작용
+   - 독자의 의견을 물어보는 질문 포함
+   - 댓글 달기를 독려하는 문구 사용
+5. 이모지 활용
+   - 주요 포인트나 새로운 섹션을 시작할 때만 관련 이모지 사용
+   - 전체 글에서 3-5개 정도의 이모지만 사용하여 과도하지 않게 유지
+6. 마무리
+   - 친근하고 격려하는 톤으로 마무리
+   - 다음 포스팅에 대한 기대감 유발
+주의사항: 너무 가벼운 톤은 지양하고, 주제의 중요성을 해치지 않는 선에서 친근함 유지
+예시: "여러분, 오늘 하루는 어떠셨나요? 😊 저는 오늘 재미있는 경험을 했어요. 바로 '미니멀 라이프'에 도전해본 건데요. 처음에는 좀 막막했지만, 생각보다 즐거운 경험이었어요! 여러분도 한번 따라해보시겠어요? 제가 알게 된 꿀팁들을 하나하나 알려드릴게요. 
+""",
+        "일반": """
+#일반적인 블로그 포스팅 스타일 가이드
+1. 톤과 어조
+   - 중립적이고 객관적인 톤 유지
+   - 적절한 존댓말 사용 (예: "~합니다", "~입니다")
+2. 내용 구조 및 전개
+   - 명확한 주제 제시로 시작
+   - 논리적인 순서로 정보 전개
+   - 주요 포인트를 강조하는 소제목 활용
+   - 적절한 길이의 단락으로 구성
+3. 용어 및 설명 방식
+   - 일반적으로 이해하기 쉬운 용어 선택
+   - 필요시 간단한 설명 추가
+   - 객관적인 정보 제공에 중점
+4. 텍스트 구조화
+   - 불릿 포인트나 번호 매기기를 활용하여 정보 구조화
+   - 중요한 정보는 굵은 글씨나 기울임꼴로 강조
+5. 독자 상호작용
+   - 적절히 독자의 생각을 묻는 질문 포함
+   - 추가 정보를 찾을 수 있는 키워드 제시
+6. 마무리
+   - 주요 내용 간단히 요약
+   - 추가 정보에 대한 안내 제공
+주의사항: 너무 딱딱하거나 지루하지 않도록 균형 유지
+예시: "최근 환경 문제가 대두되면서 '제로 웨이스트' 라이프스타일에 대한 관심이 높아지고 있습니다. 제로 웨이스트란 일상생활에서 발생하는 쓰레기를 최소화하는 생활 방식을 말합니다. 이 글에서는 제로 웨이스트의 개념, 실천 방법, 그리고 그 효과에 대해 알아보겠습니다. 먼저 제로 웨이스트의 정의부터 살펴보면...
+""",
+        "전문적인": """
+#전문적인 블로그 포스팅 스타일 가이드
+1. 톤과 구조
+   - 공식적이고 학술적인 톤 사용
+   - 객관적이고 분석적인 접근 유지
+   - 명확한 서론, 본론, 결론 구조
+   - 체계적인 논점 전개
+   - 세부 섹션을 위한 명확한 소제목 사용
+2. 내용 구성 및 전개
+   - 복잡한 개념을 정확히 전달할 수 있는 문장 구조 사용
+   - 논리적 연결을 위한 전환어 ��용
+   - 해당 분야의 전문 용어 적극 활용 (필요시 간략한 설명 제공)
+   - 심층적인 분석과 비판적 사고 전개
+   - 다양한 관점 제시 및 비교
+3. 데이터 및 근거 활용
+   - 통계, 연구 결과, 전문가 의견 등 신뢰할 수 있는 출처 인용
+   - 필요시 각주나 참고문헌 목록 포함
+   - 수치 데이터는 텍스트로 명확히 설명
+4. 텍스트 구조화
+   - 논리적 구조를 강조하기 위해 번호 매기기 사용
+   - 핵심 개념이나 용어는 기울임꼴로 강조
+   - 긴 인용문은 들여쓰기로 구분
+5. 마무리
+   - 핵심 논점 재강조
+   - 향후 연구 방향이나 실무적 함의 제시
+주의사항: 전문성을 유지하되, 완전히 이해하기 어려운 수준은 지양
+예시: "본 연구에서는 인공지능(AI)의 윤리적 함의에 대해 고찰한다. 특히, 자율주행 자동차의 의사결정 알고리즘에서 발생할 수 있는 윤리적 딜레마에 초점을 맞춘다. Bonnefon et al. (2016)의 연구에 따르면, 자율주행 차량의 알고리즘이 직면할 수 있는 윤리적 선택의 복잡성이 지적된 바 있다. 본고에서는 이러한 윤리적 딜레마를 세 가지 주요 관점에서 분석한다: 1) 공리주의적 접근, 2) 의무론적 접근, 3) 덕 윤리적 접근. 각 접근법의 장단점을 비교 분석하고, 이를 바탕으로 자율주행 차량의 윤리적 의사결정 프레임워크를 제안하고자 한다...
+"""
+    }
+    return prompts.get(style, "포스팅 스타일 프롬프트")
+
+def get_style_description(style):
+    descriptions = {
+        "친근한": "독자와 가까운 친구처럼 대화하는 듯한 친근한 스타일입니다.",
+        "일반": "일반적이고 중립적인 톤으로 정보를 전달하는 스타일입니다.",
+        "전문적인": "전문가의 시각에서 깊이 있는 정보를 전달하는 스타일입니다."
+    }
+    return descriptions.get(style, "포스팅 스타일을 선택하세요.")
+
+def update_prompts(category, style):
+    title_prompt = get_title_prompt(category)
+    outline_prompt = get_outline_prompt(category)
+    blog_post_prompt = get_blog_post_prompt(category)
+    style_prompt = get_style_prompt(style)
+    style_description = get_style_description(style)
+    return title_prompt, outline_prompt, blog_post_prompt, style_prompt, style_description
+
+#PDF파일 만들
+class PDF(FPDF):
+    def __init__(self):
+        super().__init__()
+        current_dir = os.path.dirname(__file__)
+        self.add_font("NanumGothic", "", os.path.join(current_dir, "NanumGothic.ttf"), uni=True)
+        self.add_font("NanumGothic", "B", os.path.join(current_dir, "NanumGothicBold.ttf"), uni=True)
+        self.add_font("NanumGothicExtraBold", "", os.path.join(current_dir, "NanumGothicExtraBold.ttf"), uni=True)
+        self.add_font("NanumGothicLight", "", os.path.join(current_dir, "NanumGothicLight.ttf"), uni=True)
+
+    def header(self):
+        self.set_font('NanumGothic', '', 10)
+
+    def footer(self):
+        self.set_y(-15)
+        self.set_font('NanumGothic', '', 8)
+        self.cell(0, 10, f'Page {self.page_no()}', 0, 0, 'C')
+
+def save_to_pdf(blog_post):
+    pdf = PDF()
+    pdf.add_page()
+    
+    lines = blog_post.split('\n')
+    title = lines[0].strip()
+    content = '\n'.join(lines[1:]).strip()
+    
+    filename = format_filename(title) + ".pdf"
+    
+    pdf.set_font("NanumGothic", 'B', size=14)
+    pdf.cell(0, 10, title, ln=True, align='C')
+    pdf.ln(10)
+    
+    pdf.set_font("NanumGothic", '', size=11)
+    pdf.multi_cell(0, 5, content)
+
+    print(f"Saving PDF as: {filename}")
+    pdf.output(filename)
+    return filename
+
+def format_filename(text):
+    text = re.sub(r'[^\w\s-]', '', text)
+    return text[:50].strip()
+
+# save_content_to_pdf 함수도 간단하게 수정
+def save_content_to_pdf(blog_post):
+    return save_to_pdf(blog_post)
+
 title = "정보성 포스팅 자동생성기(제목추천 후 자동)"
 
-def update_prompts(category):
+def update_prompts_and_description(category, style):
     title_prompt = get_title_prompt(category)
     outline_prompt = get_outline_prompt(category)
     blog_post_prompt = get_blog_post_prompt(category)
-    return title_prompt, outline_prompt, blog_post_prompt
+    style_prompt = get_style_prompt(style)
+    style_description = get_style_description(style)
+    return title_prompt, outline_prompt, blog_post_prompt, style_prompt, style_description
 
 with gr.Blocks() as demo:
     gr.Markdown(f"# {title}")
     
-    # 1단계
     gr.Markdown("### 1단계 : 포스팅 카테고리를 지정해주세요")
     category = gr.Radio(choices=["일반", "건강정보"], label="포스팅 카테고리", value="일반")
     
+    gr.Markdown("---\n\n")
+
+    gr.Markdown("### 2단계: 포스팅 스타일을 선택해주세요", elem_id="step-title")
+    style = gr.Radio(choices=["친근한", "일반", "전문적인"], label="포스팅 스타일", value="친근한")
+    style_description = gr.Markdown(f"_{get_style_description('친근한')}_", elem_id="style-description")
+    
+    gr.Markdown("---\n\n")
+
     # 2단계
-    gr.Markdown("### 2단계 : 블로그 주제, 또는 키워드를 상세히 입력하세요")
+    gr.Markdown("### 3단계 : 블로그 주제, 또는 키워드를 상세히 입력하세요")
     topic = gr.Textbox(label="블로그 주제(예시: 오징어 무침회(X), 오징어 무침회 레시피(O))", placeholder="예시: 여행지 추천(X), 8월 국내 여행지 추천(O)")
 
     # 3단계: 참고 글을 위한 변수들 미리 정의
@@ -244,7 +407,7 @@ with gr.Blocks() as demo:
 
     # 제목 추천 버튼
     title_btn = gr.Button("제목 추천하기")
-    title_btn.click(fn=suggest_title, inputs=[category, topic, references1, references2, references3, title_system_message, title_max_tokens, title_temperature, title_top_p], outputs=[title_suggestions, title_token_output])
+    title_btn.click(fn=suggest_title, inputs=[category, style, topic, references1, references2, references3, title_system_message, style, title_max_tokens, title_temperature, title_top_p], outputs=[title_suggestions, title_token_output])
     
     blog_title = gr.Textbox(label="블로그 제목", placeholder="블로그 제목을 입력해주세요")
     
@@ -272,10 +435,16 @@ with gr.Blocks() as demo:
     generate_post_btn = gr.Button("블로그 글 생성하기")
     generate_post_btn.click(
         fn=fetch_references_and_generate_all_steps,
-        inputs=[category, topic, blog_title, outline_system_message, outline_max_tokens, outline_temperature, outline_top_p, blog_system_message, blog_max_tokens, blog_temperature, blog_top_p],
+        inputs=[category, style, topic, blog_title, outline_system_message, outline_max_tokens, outline_temperature, outline_top_p, blog_system_message, blog_max_tokens, blog_temperature, blog_top_p],
         outputs=[references1, references2, references3, outline_result, outline_token_output, output, token_output]
     )
 
-    category.change(fn=update_prompts, inputs=category, outputs=[title_system_message, outline_system_message, blog_system_message])
+    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, outputs=[title_system_message, outline_system_message, blog_system_message, style_description])
+    style.change(fn=update_prompts_and_description, inputs=category, outputs=[title_system_message, outline_system_message, blog_system_message, style_description])
 
 demo.launch()