Update app.py

app.py

@@ -116,24 +116,39 @@ def call_api(content, system_message, max_tokens, temperature, top_p):
 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):
+def suggest_title(category, style, topic, references1, references2, references3):
     full_content = analyze_info(category, style, topic, references1, references2, references3)
+    category_prompt = get_title_prompt(category)
+    style_prompt = get_style_prompt(style)
+    max_tokens = 5000
+    temperature = 0.8
+    top_p = 0.95
     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, style, topic, references1, references2, references3, title, category_prompt, style_prompt, max_tokens, temperature, top_p):
+def generate_outline(category, style, topic, references1, references2, references3, title):
     full_content = analyze_info(category, style, topic, references1, references2, references3)
     content = f"{full_content}\nTitle: {title}"
+    category_prompt = get_outline_prompt(category)
+    style_prompt = get_style_prompt(style)
+    max_tokens = 6000
+    temperature = 0.8
+    top_p = 0.95
     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, style, topic, references1, references2, references3, title, outline, category_prompt, style_prompt, max_tokens, temperature, top_p):
+def generate_blog_post(category, style, topic, references1, references2, references3, title, outline):
     full_content = analyze_info(category, style, topic, references1, references2, references3)
     content = f"{full_content}\nTitle: {title}\nOutline: {outline}"
+    category_prompt = get_blog_post_prompt(category)
+    style_prompt = get_style_prompt(style)
+    max_tokens = 8000
+    temperature = 0.8
+    top_p = 0.95
     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')
@@ -156,7 +171,7 @@ def fetch_references(topic):
     
     return "참고글 생성 완료", references1_content, references2_content, references3_content
 
-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):
+def fetch_references_and_generate_all_steps(category, style, topic, blog_title):
     search_url = generate_naver_search_url(topic)
     session = setup_session()
     if session is None:
@@ -172,10 +187,10 @@ def fetch_references_and_generate_all_steps(category, style, topic, blog_title,
     references3_content = f"제목: {selected_results[2]['제목']}\n내용: {crawl_blog_content(selected_results[2]['링크'], session)}"
     
     # 아웃라인 생성
-    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)
+    outline_result, outline_token_usage = generate_outline(category, style, topic, references1_content, references2_content, references3_content, blog_title)
     
     # 블로그 글 생성
-    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)
+    blog_post_result, blog_post_token_usage = generate_blog_post(category, style, topic, references1_content, references2_content, references3_content, blog_title, outline_result)
     
     return references1_content, references2_content, references3_content, outline_result, outline_token_usage, blog_post_result, blog_post_token_usage
 
@@ -183,150 +198,30 @@ def get_title_prompt(category):
     if (category == "일반"):
         return """
 # 블로그 제목 생성 규칙(일반)
-##[기본규칙]
-1. 반드시 ��국어(한글)로 작성하라.
-2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
-3. 특히 너는 '정보성(Informative)' 전문 블로그 마케팅 전문가이다.
-4. 정보 제공에 초점을 맞추어 작성한다. 
-
-##[블로그 제목 작성 규칙]
-1. 블로그 제목 10개를 작성하고 제목 10개만 출력하라.
-2. 제목은 40자 이내로 작성하라.
-3. 제공된 참고글에 맞춰 블로그 제목 10개를 작성하라.
-4. 반드시 핵심키워드(Topic)가 문장 앞쪽에 들어가도록 작성하라.
-5. 핵심 키워드와 연관성 높은 주제를 포함하여 작성하라.
 """
     elif (category == "건강정보"):
         return """
 # 블로그 제목 생성 규칙(건강정보)
-##[기본규칙]
-1. 반드시 한국어(한글)로 작성하라.
-2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
-3. 특히 너는 '건강, 의학 정보' 전문 블로그 마케터이다.
-4. 정확하고 전문적인 정보 제공에 초점을 맞추어 작성한다.
-
-##[블로그 제목 작성 규칙]
-1. 블로그 제목 10개를 작성하고 제목 10개만 출력하라.
-2. 제목은 40자 이내로 작성하라.
-3. 제공된 참고글에 맞춰 블로그 제목 10개를 작성하라.
-4. 사용자가 입력한 블로그 주제, 핵심키워드(Topic)가 문장 앞쪽에 들어가도록 제목을 작성하라.
-5. 참고글을 분석하여 독자들이 건강한 생활을 유지하는 데 필요한 정보를 반영하라.
 """
 
 def get_outline_prompt(category):
     if (category == "일반"):
         return """
 # 블로그 소주제(Subtopic) 생성 규칙(일반)
-##[기본규칙]
-1. 반드시 한국어(한글)로 작성하라.
-2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
-3. 특히 너는 '정보성(Informative)' 전문 블로그 마케팅 전문가이다.
-4. 정보 제공에 초점을 맞추어 작성한다. 
-
-##[소주제 작성규칙]
-1. [기본규칙]을 기본 적용하라.
-2. 블로그 글을 작성하기 위한 소주제를 작성하라.
-3. 제공된 참고글과 블로그 주제, 제목을 바탕으로 핵심 주제를 파악하여 소주제를 생성하라.
-4. 전체 맥락에 맞게 소주제를 작성하라.
-5. 소제목으로 사용할 수 있도록 20자 내외로 작성하라.
-6. 독자가 얻고자 하는 정보와 흥미로운 정보를 제공하도록 소주제를 작성하라.
-7. 소주제의 본론의 내용이 충분히 작성될 수 있는 소주제로 설정하라. 
-8. 반드시 [소주제 구성]에 맞게 출력하라.
-
-##[소주제 구성]
-1. 반드시 [도입부] - 1개, [본론1~5] - 5개, [결론] - 1개로 구성하여 출력하라.
-2. 반드시 [도입부]와 [결론]의 제목이 중복되지 않도록 작성하라.
 """
     elif (category == "건강정보"):
         return """
 # 블로그 소주제(Subtopic) 생성 규칙(건강정보)
-##[기본규칙]
-1. 반드시 한국어(한글)로 작성하라.
-2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
-3. 특히 너는 '건강, 의학 정보' 전문 블로그 마케터이다.
-4. 정확하고 전문적인 정보 제공에 초점을 맞추어 작성한다.
-
-##[소주제 작성규칙]
-1. [기본규칙]을 기본 적용하라.
-2. 블로그 글을 작성하기 위한 소주제를 작성하라.
-3. 제공된 참고글과 블로그 주제, 제목을 바탕으로 핵심 주제를 파악하여 소주제를 생성하라.
-4. 전체 맥락에 맞게 소주제를 작성하라.
-5. 소제목으로 사용할 수 있도록 20자 내외로 작성하라.
-6. 독자가 얻고자 하는 정확한 정보와 건강한 생활을 유지하는 데 필요한 정보를 제공하도록 소주제를 작성하라.
-7. 소주제의 본론의 내용이 충분히 작성될 수 있는 소주제로 설정하라. 
-8. 반드시 [소주제 구성]에 맞춰 소주제만 출력하라.
-
-##[소주제 구성]
-1. 반드시 [도입부] - 1개, [본론1~5] - 5개, [결론] - 1개로 구성하여 출력하라.
-2. 반드시 [도입부]와 [결론]의 제목이 중복되지 않도록 작성하라.
 """
 
 def get_blog_post_prompt(category):
     if (category == "일반"):
         return """
 # 블로그 텍스트 생성 규칙(일반)
-##[기본규칙]
-1. 반드시 한국어(한글)로 작성하라.
-2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
-3. 특히 너는 '정보성(Informative)' 전문 블로그 마케팅 전문가이다.
-4. 정보 제공에 초점을 맞추어 작성한다. 
-
-##[텍스트 작성 규칙]
-1. 반드시 입력된 [소주제]에 맞게 텍스트를 작성하라.
-2. 소주제의 [본론] 5개를 각각 300자 이상으로 작성하라.
-3. 소주제의 [본론] 이 300자 이상 생성되지 않는다면, 전체글이 2000자 이상되도록 작성하라.
-4. 전체 맥락을 이해하고 문장의 일관성을 유지하라.
-5. 주제에 맞는 닉네임, 페르소나를 적용하여 작성하라.
-6. 제공된 참고글��� 어투를 반영하되, [포스팅 스타일]에 맞게 적용하라.
-7. 절대로 참고글을 한문장 이상 그대로 출력하지 말 것.
-8. 주제와 상황에 맞는 적절한 어휘를 선택하라. 
-9. 한글 어휘의 난이도는 쉽게 작성하라.
-10. 절대 문장의 끝에 '답니다'를 사용하지 말 것.
-
-	###[정보성 블로그 작성 규칙]
-	1. 독자가 얻고자 하는 유용한 정보와 흥미로운 정보를 제공하도록 작성하라.
-	2. 독자의 공감을 이끌어내고 궁금증을 해결하도록 작성하라.
-	3. 독자의 관심사를 충족시키도록 작성하라.
-	4. 독자에게 이득이 되는 정보를 작성하라.
-
-##[제외 규칙]
-1. 반드시 비속어 및 욕설(expletive, abusive language, slang)은 제외하라.
-2. 반드시 참고글의 링크(URL)는 제외하라.
-3. 참고글에서 '링크를 확인해주세요'와 같은 링크 이동의 문구는 제외하라.
-4. 참고글에 있는 작성자, 화자, 유튜버, 기자(Writer, speaker, YouTuber, reporter)의 이름, 애칭, 닉네임(Name, Nkickname)은 반드시 제외하라.
-5. 반드시 문장의 끝부분이 어색한 한국어 표현은 제외하라('예요', '답니다', '해요', '해주죠', '됐죠', '됐어요', '고요' 등.)
 """
     elif (category == "건강정보"):
         return """
 # 블로그 텍스트 생성 규칙(건강정보)
-##[기본규칙]
-1. 반드시 한국어(한글)로 작성하라.
-2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
-3. 특히 너는 '건강, 의학 정보' 전문 블로그 마케터이다.
-4. 정확하고 전문적인 정보 제공에 초점을 맞추어 작성한다.
-
-##[텍스트 작성 규칙]
-1. 반드시 입력된 [소주제]에 맞춰서 텍스트를 작성하라.
-2. 반드시 입력된 [소주제]는 변경하지 말고 그대로 출력하라.
-3. 소주제의 [도입부]는 가볍게 작성하되 공감과 흥미, 문제제기, 글의 목적, 본문으로 자연스럽게 이어지는 전환문장등을 출력하라.
-3. 소주제의 [본론] 5개의 내용이 각각 500자 이상이 되도록 작성하라.
-4. 반드시 너가 작성한 전체글이 2500자 이상이 되도록 작성하라.
-5. 전체 맥락을 이해하고 문장의 일관성을 유지하라.
-6. 참고글을 바탕으로 작성한 글에 맞는 닉네임, 페르소나를 적용하여 작성하라.
-7. 어투는 제공된 참고글의 어투를 반영하되, '건강, 의학 정보' 전문 블로그 마케터로서 작성하라.
-8. 절대로 참고글을 한문장 이상 그대로 출력하지 말 것.
-9. 주제와 상황에 맞는 적절한 어휘를 선택하라. 
-
-	###[정보성 블로그 작성 규칙]
-	1. 독자가 얻고자 하는 정확한 정보와 건강한 생활을 유지하는 데 필요한 정보를 제공하도록 소주제에 맞게 내용을 작성하라.
-	2. 독자의 공감을 이끌어내고 궁금증을 해결하도록 작성하라.
-	3. 독자의 관심사를 충족시키도록 작성하라.
-	4. 독자에게 이득이 되는 정보를 작성하라.
-
-##[제외 규칙]
-1. 반드시 참고글의 링크(URL)는 제외하라.
-2. 참고글에서 '링크를 확인해주세요'와 같은 링크 이동의 문구는 제외하라.
-3. 참고글에 있는 작성자, 화자, 유튜버, 기자(Writer, speaker, YouTuber, reporter)의 이름, 애칭, 닉네임(Name, Nkickname)은 반드시 제외하라.
 """
 
 def get_style_prompt(style):
@@ -515,36 +410,19 @@ with gr.Blocks() as demo:
 
     # 제목 추천
     gr.Markdown("### 4단계 : 제목 추천하기")
-    
-    with gr.Accordion("제목 설정", open=True):
-        title_system_message = gr.Textbox(label="시스템 메시지", value=get_title_prompt("일반"), lines=15)
-        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_suggestions = gr.Textbox(label="제목 추천", lines=10)
     title_token_output = gr.Markdown(label="사용된 토큰 수")
 
     # 제목 추천 버튼
     title_btn = gr.Button("제목 추천하기")
-    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])
+    title_btn.click(fn=suggest_title, inputs=[category, style, topic, references1, references2, references3], outputs=[title_suggestions, title_token_output])
     
     blog_title = gr.Textbox(label="블로그 제목", placeholder="블로그 제목을 입력해주세요")
     
     # 블로그 글 생성
     gr.Markdown("### 5단계 : 블로그 글 생성하기")
     gr.HTML("<span style='color: grey;'>[블로그 글 생성하기 버튼을 클릭하면 아웃라인 생성 및 블로그 글 작성이 자동으로 진행됩니다.]</span>")
-    
-    with gr.Accordion("블로그 글 설정", open=True):
-        outline_system_message = gr.Textbox(label="시스템 메시지", value=get_outline_prompt("일반"), lines=20)
-        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)
-        
-        blog_system_message = gr.Textbox(label="시스템 메시지", value=get_blog_post_prompt("일반"), lines=20)
-        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)
 
     outline_result = gr.Textbox(label="아웃라인 결과", lines=15, visible=False)
     outline_token_output = gr.Markdown(label="사용된 토큰 수", visible=False)
@@ -555,7 +433,7 @@ with gr.Blocks() as demo:
     generate_post_btn = gr.Button("블로그 글 생성하기")
     generate_post_btn.click(
         fn=fetch_references_and_generate_all_steps,
-        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],
+        inputs=[category, style, topic, blog_title],
         outputs=[references1, references2, references3, outline_result, outline_token_output, output, token_output]
     )
 
@@ -564,7 +442,7 @@ with gr.Blocks() as demo:
 
     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])
+    category.change(fn=update_prompts_and_description, inputs=category, outputs=[style_description])
+    style.change(fn=update_prompts_and_description, inputs=category, outputs=[style_description])
 
 demo.launch()