app.py

import gradio as gr
import random
import os
import re
from huggingface_hub import InferenceClient
from fpdf import FPDF
from datetime import datetime

def create_client(model_name):
    return InferenceClient(model_name, token=os.getenv("HF_TOKEN"))

client = create_client("CohereForAI/c4ai-command-r-plus")

def call_api(content, system_message, max_tokens, temperature, top_p):
    messages = [{"role": "system", "content": system_message}, {"role": "user", "content": content}]
    random_seed = random.randint(0, 1000000)
    response = client.chat_completion(messages=messages, max_tokens=max_tokens, temperature=temperature, top_p=top_p, seed=random_seed)
    return response.choices[0].message.content

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, 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 = call_api(full_content, combined_prompt, max_tokens, temperature, top_p)
    return modified_text

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}"
    combined_prompt = f"{category_prompt}\n\n{style_prompt}"
    modified_text = call_api(content, combined_prompt, max_tokens, temperature, top_p)
    return modified_text

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}"
    combined_prompt = f"{category_prompt}\n\n{style_prompt}"
    modified_text = call_api(content, combined_prompt, max_tokens, temperature, top_p)
    formatted_text = modified_text.replace('\n', '\n\n')
    return formatted_text

def get_title_prompt(category):
    if (category == "고객반응형"):
        return """
        #블로그 제목 생성 규칙(상품리뷰_고객반응형)
        ##[기본규칙]
        1. 반드시 한국어(한글)로 작성하라.
        2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
        3. 특히 너는 '고객반응 상품리뷰' 전문 블로그 마케터이다.
        4. 상품 리뷰를 작성하되, 고객의 실제 반응이 포함된 글을 작성하라.
        5. 선택된 포스팅 스타일을 적용하여 글을 작성하라.
        
        ##[블로그 제목 작성 규칙]
        1. 상품리뷰 블로그 포스팅중 고객의 실제 반응에 특화된 제목을 작성하라.
        2. 참고글을 바탕으로 블로그 제목 10개를 작성하고 제목 10개만 출력하라.
        3. 제목은 40자 이내로 작성하라.
        4. 사용자가 입력한 블로그 주제, 핵심키워드(Topic), 상품명이 문장 앞쪽에 들어가도록 제목을 작성하라.
        5. '쿠팡'이란 단어는 제목에 표현하지 말 것.
        """

def get_outline_prompt(category):
    if (category == "고객반응형"):
        return """
        #블로그 소주제(Subtitle) 아웃라인 생성 규칙
        ##[기본규칙]
        1. 반드시 한국어(한글)로 작성하라.
        2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
        3. 특히 너는 '고객반응 상품리뷰' 전문 블로그 마케터이다.
        4. 상품 리뷰를 작성하되, 고객의 실제 반응이 포함된 글을 작성하라.
        5. 선택된 포스팅 스타일을 적용하여 글을 작성하라.
        
        ##[소주제 작성규칙]
        1. 블로그 글을 작성하기 위한 소주제를 작성하라.
        2. 반드시 상품의 고객의 실제 반응이 포함된 주제로만 소주제를 생성하라.
        3. 소제목으로 사용할 수 있도록 30자 이내로 작성하라.
        4. 독자가 얻고자 하는 흥미로운 정보와 고객의 실제 반응을 제공하도록 소주제를 작성하라.
        5. 반드시 [소주제 구성]에 맞게 출력하라.
        6. 상품명은 소주제에 포함하지 말 것.
        
        ##[소주제 구성]
        1. 반드시 [도입부] - 1개, [본론1~5] - 5개, [결론] - 1개로 구성하여 출력하라.
        2. 반드시 [도입부]와 [결론]의 제목이 중복되지 않도록 작성하라.
        3. [도입부]에는 고객의 실제 리뷰를 분석하여 공감, 문제제기, 구매포인트 등을 제시하라.
        """

def get_blog_post_prompt(category):
    if (category == "고객반응형"):
        return """
        #블로그 텍스트 생성 규칙(상품리뷰_고객반응형)
        ##[기본규칙]
        1. 반드시 한국어(한글)로 작성하라.
        2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
        3. 특히 너는 '고객반응 상품리뷰' 전문 블로그 마케터이다.
        4. 상품 리뷰를 작성하되, 고객의 실제 반응이 포함된 글을 작성하라.
        5. 선택된 포스팅 스타일을 적용하여 글을 작성하라.
                
        ##기능집중형 상품리뷰 생성규칙]
        1. 반드시 입력된 [소주제]에 맞게 텍스트를 작성하라.
        2. 블로그주제와 참고글을 바탕으로 상품명을 파악하여 기억하고, 소주제에서 반드시 상품명을 최대 1회만 출력되게하라.
        3. 소주제별 각각 400자 이상으로 작성하고, 반드시 전체글이 2200자 이상 되도록 작성하라.
        4. 전체 맥락을 이해하고 문장의 일관성을 유지하되, 내용이 중복되지 않게 작성하라.
        5. 절대로 참고글을 한문장 이상 그대로 출력하지 말 것.
        6. 상품에 대한 분석(수치를 포함), 기능과 사용법등의 정보를 작성하고 그에 대한 고객의 실제 반응을 포함하여 리뷰를 작성하라.
        7. 내용은 긍정적으로 작성하되, 고객의 실제 반응이 돋보이도록 작성하라.
        8. 고객의 실제 리뷰를 바탕으로 포스팅 스타일에 맞는 상황을 연출하라.
        9. 고객의 실제 리뷰를 문장에 반영할때는 '구매자의 리뷰', '고객의 리뷰', '사용자 리뷰' 등의 다양한 표현을 사용하라.
                
        ##[제외 규칙]
        1. 반드시 참고글의 링크(URL)는 제외하라.
        2. 참고글에서 '링크를 확인해주세요'와 같은 링크 이동의 문구는 제외하라.
        3. 참고글에 있는 작성자, 화자, 유튜버, 기자(Writer, speaker, YouTuber, reporter)의 이름, 애칭, 닉네임(Name, Nkickname)은 반드시 제외하라.
        4. '업체로 부터 제공 받아서 작성', '쿠팡 파트너스'등의 표현을 반드시 제외하라.
        """

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. 마무리
   - 핵심 논점 재강조
   - 향후 연구 방향이나 실무적 함의 제시

주의사항: 전문성을 유지하되, 완전히 이해하기 어려운 수준은 지양
        """
    }
    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

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"))
        self.add_font("NanumGothic", "B", os.path.join(current_dir, "NanumGothicBold.ttf"))
        self.add_font("NanumGothicExtraBold", "", os.path.join(current_dir, "NanumGothicExtraBold.ttf"))
        self.add_font("NanumGothicLight", "", os.path.join(current_dir, "NanumGothicLight.ttf"))

    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 format_filename(text):
    text = re.sub(r'[^\w\s-]', '', text)
    return text[:50].strip()

def extract_first_recommended_title(blog_post):
    match = re.search(r'## 추천 제목\s*\n\s*(?:\n\s*)*\d+\.\s*(.*?)\s*(?:\n|$)', blog_post, re.DOTALL)
    if match:
        title = match.group(1).strip()
        print(f"Extracted title: {title}")
        return title
    print("No title found")
    return "블로그_글"

def save_to_pdf(blog_post):
    pdf = PDF()
    
    pdf.add_page()
    pdf.set_font("NanumGothic", 'B', size=14)
    pdf.cell(0, 10, "블로그 글", ln=True, align='C')
    pdf.ln(10)
    pdf.set_font("NanumGothic", '', size=11)
    pdf.multi_cell(0, 5, blog_post)

    title = extract_first_recommended_title(blog_post)
    filename = format_filename(title) + ".pdf"
    print(f"Saving PDF as: {filename}")
    pdf.output(filename)
    return filename

def save_content_to_pdf(blog_post):
    filename = save_to_pdf(blog_post)
    return filename

title = "상품리뷰 고객반응형_수동"

with gr.Blocks() as demo:
    gr.Markdown(f"# {title}")
    
    gr.Markdown("### 1단계: 포스팅 카테고리를 지정해주세요", elem_id="step-title")
    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")

    gr.Markdown("### 3단계: 블로그 주제, 또는 키워드를 상세히 입력하세요", elem_id="step-title")
    topic = gr.Textbox(label="블로그 주제(예시: 오징어 무침회(X), 오징어 무침회 레시피(O))", placeholder="예시: 여행지 추천(X), 8월 국내 여행지 추천(O)")

    gr.Markdown("---\n\n")

    gr.Markdown("### 4단계: 참고 글을 입력하세요", elem_id="step-title")
    references1 = gr.Textbox(label="참고 글 1", placeholder="참고할 블로그 포스팅글을 복사하여 붙여넣으세요", lines=10, visible=True)
    references2 = gr.Textbox(label="참고 글 2", placeholder="참고할 블로그 포스팅글을 복사하여 붙여넣으세요", lines=10, visible=True)
    references3 = gr.Textbox(label="참고 글 3", placeholder="참고할 블로그 포스팅글을 복사하여 붙여넣으세요", lines=10, visible=True)

    gr.Markdown("---\n\n")

    gr.Markdown("### 5단계: 블로그 제목을 입력하세요", elem_id="step-title")
    
    with gr.Accordion("제목 설정", visible=True):
        title_system_message = gr.Textbox(label="카테고리 프롬프트", value=get_title_prompt("고객반응형"), lines=15, visible=True)
        title_max_tokens = gr.Slider(label="Max Tokens", minimum=1000, maximum=8000, value=5000, step=1000)
        title_temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, value=0.8, step=0.1)
        title_top_p = gr.Slider(label="Top P", minimum=0.1, maximum=1.0, value=0.95, step=0.05)

    title_btn = gr.Button("제목 추천하기")
    title_suggestions = gr.Textbox(label="제목 추천", lines=10)
    
    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])
    gr.HTML("<span style='color: grey;'>[제목 추천 항목을 참고하여 하나를 복사하여 사용하셔도 됩니다.]</span>")
    blog_title = gr.Textbox(label="블로그 제목", placeholder="블로그 제목을 입력해주세요")
    
    gr.Markdown("---\n\n")

    gr.Markdown("### 6단계: 아웃라인을 작성해주세요", elem_id="step-title")
    gr.HTML("<span style='color: grey;'>[아웃라인에서 나온 결과를 수정해서 사용해주세요]</span>")
    
    with gr.Accordion("아웃라인 설정", visible=True):
        outline_system_message = gr.Textbox(label="카테고리 프롬프트", value=get_outline_prompt("고객반응형"), lines=20, visible=True)
        outline_max_tokens = gr.Slider(label="Max Tokens", minimum=1000, maximum=8000, value=6000, step=1000)
        outline_temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, value=0.8, step=0.1)
        outline_top_p = gr.Slider(label="Top P", minimum=0.1, maximum=1.0, value=0.95, step=0.05)
    
    outline_generate_btn = gr.Button("아웃라인 생성하기")
    outline_result = gr.Textbox(label="아웃라인 결과", lines=15)
    outline_input = gr.Textbox(label="작성할 아웃라인을 입력해주세요", placeholder="생성된 아웃라인 복사, 수정해서 사용하세요", lines=10)

    outline_generate_btn.click(fn=generate_outline, inputs=[category, style, topic, references1, references2, references3, blog_title, outline_system_message, style, outline_max_tokens, outline_temperature, outline_top_p], outputs=[outline_result])
    
    gr.Markdown("---\n\n")

    gr.Markdown("### 7단계: 글 생성하기", elem_id="step-title")
    gr.HTML("<span style='color: grey;'>[아웃라인 별 텍스트량을 정하고 글 생성하기 버튼을 선택해주세요]</span>")
    
    with gr.Accordion("블로그 글 설정", visible=True):
        blog_system_message = gr.Textbox(label="카테고리 프롬프트", value=get_blog_post_prompt("고객반응형"), lines=20, visible=True)
        blog_max_tokens = gr.Slider(label="Max Tokens", minimum=1000, maximum=12000, value=8000, step=1000)
        blog_temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, value=0.8, step=0.1)
        blog_top_p = gr.Slider(label="Top P", minimum=0.1, maximum=1.0, value=0.95, step=0.05)
        
    generate_btn = gr.Button("블로그 글 생성하기")
    output = gr.Textbox(label="생성된 블로그 글", lines=30)
    
    generate_btn.click(fn=generate_blog_post, inputs=[category, style, topic, references1, references2, references3, blog_title, outline_input, blog_system_message, style, blog_max_tokens, blog_temperature, blog_top_p], outputs=[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])

    def update_prompts_and_description(category, style):
        title_prompt, outline_prompt, blog_post_prompt, style_prompt, style_description = update_prompts(category, style)
        return title_prompt, outline_prompt, blog_post_prompt, style_description
    
    category.change(fn=update_prompts_and_description, inputs=[category, style], outputs=[title_system_message, outline_system_message, blog_system_message, style_description])
    style.change(fn=update_prompts_and_description, inputs=[category, style], outputs=[title_system_message, outline_system_message, blog_system_message, style_description])

demo.launch()

gr.HTML("""
<style>
#step-title {
    font-size: 1.7em;
    font-weight: bold;
}
#style-description {
    font-size: 1.2em;
    font-style: italic;
}
</style>
""")