Update app.py

app.py

@@ -130,6 +130,8 @@ def fetch_crawl_results(query):
 
 def generate_blog_post(query, prompt_template):
     try:
+        target_length = 2000  # 내부적으로 목표 글자수 설정
+        
         references = fetch_references(query)
         ref1, ref2, ref3 = references
 
@@ -144,8 +146,8 @@ def generate_blog_post(query, prompt_template):
 {ref2}
 
 참고글3:
-{ref3}"""
-
+{ref3}
+"""
         messages = [
             {"role": "system", "content": prompt_template},
             {"role": "user", "content": user_message}
@@ -163,11 +165,43 @@ def generate_blog_post(query, prompt_template):
         
         generated_post = response.choices[0].message['content']
         
+        # 원치 않는 표현들을 정규표현식 패턴으로 정의
+        unwanted_patterns = [
+            r'\b여러분[,.]?\s*',
+            r'\b(마지막으로|결론적으로|결국|종합적으로|따라서|마무리|요약)[,.]?\s*'
+        ]
+        
+        # 글자수가 부족한 경우 추가 생성
+        while len(generated_post) < target_length or any(re.search(pattern, generated_post, re.IGNORECASE) for pattern in unwanted_patterns):
+            if len(generated_post) < target_length:
+                additional_length = target_length - len(generated_post)
+                reason = f"현재 글자수는 {len(generated_post)}자 이다. 약 {additional_length}자를 추가로 작성하여 총 {target_length}자 이상이 되도록 작성하라"
+            else:
+                reason = "원치 않는 표현이 포함되어 있습니다. 해당 표현을 제거하고 자연스럽게 글을 다시 작성해주세요."
+
+            additional_message = f"{reason} 이전 내용과 자연스럽게 연결되도록 작성해주세요. 앞서 언급한 표현들(여러분, 마지막으로, 결론적으로, 결국, 종합적으로, 따라서, 마무리, 요약)은 사용하지 마세요."
+            
+            messages.append({"role": "assistant", "content": generated_post})
+            messages.append({"role": "user", "content": additional_message})
+            
+            additional_response = openai.ChatCompletion.create(
+                model="gpt-4o-mini",
+                messages=messages,
+                max_tokens=min(10000, max(additional_length * 2, 1000)),
+                temperature=0.85,
+                top_p=0.9,
+                frequency_penalty=0.5,
+                seed=random_seed
+            )
+            
+            generated_post = additional_response.choices[0].message['content']
+
         final_post = f"주제: {query}\n\n{generated_post}"
+        actual_length = len(generated_post)
         
-        return final_post, ref1, ref2, ref3
+        return final_post, ref1, ref2, ref3, actual_length
     except Exception as e:
-        return f"블로그 글 생성 중 오류 발생: {str(e)}", "", "", ""
+        return f"블로그 글 생성 중 오류 발생: {str(e)}", "", "", "", 0
         
 # PDF 클래스 및 관련 함수 정의
 class PDF(FPDF2):