Update app.py

app.py

@@ -132,7 +132,7 @@ def generate_blog_post(query, prompt_template):
     try:
         # 목표 글자수 설정 (문자 수)
         target_char_length = 2000
-        max_attempts = 2  # 최대 시도 횟수 (초기 시도 1회 + 재시도 1회)
+        max_attempts = 2  # 최대 시도 횟수
 
         # 참고글 가져오기
         references = fetch_references(query)
@@ -159,15 +159,17 @@ def generate_blog_post(query, prompt_template):
             '요약'
         ]
 
-        # 불필요한 표현 제거 함수
+        # 불필요한 표현이 포함된 단어 전체를 제거하는 함수
         def remove_unwanted_phrases(text):
-            for phrase in unwanted_phrases:
-                # 정규식 패턴 생성: 불필요한 표현 + 뒤의 공백들
-                pattern = re.compile(re.escape(phrase) + r'\s*')
-                text = pattern.sub('', text)
-            # 연속된 공백을 하나의 공백으로 치환
-            text = re.sub(r'\s+', ' ', text).strip()
-            return text
+            # 단어 단위로 텍스트를 분할
+            words = re.findall(r'\S+|\n', text)
+            result_words = []
+            for word in words:
+                # 불필요한 표현이 단어에 포함되어 있는지 확인
+                if not any(phrase in word for phrase in unwanted_phrases):
+                    result_words.append(word)
+            # 단어들을 다시 조합하여 결과 텍스트 생성
+            return ' '.join(result_words).replace(' \n ', '\n').replace(' \n', '\n').replace('\n ', '\n')
 
         # 초기 프롬프트 구성
         base_prompt = prompt_template + f"""
@@ -182,10 +184,10 @@ def generate_blog_post(query, prompt_template):
             # 첫 번째 시도 이후에는 프롬프트에 추가 지침을 포함
             if attempt > 0:
                 additional_instructions = f"""
-1. 생성된 글의 길이가 목표 글자수에 미달, 내용의 질을 유지하고 더욱 풍부하게 작성하고, 세부사항을 추가하여 자연스럽게 확장하라
-2. 중복되는 내용은 피하고 새로운 정보를 추가하라
-3. 반드시 마크다운 형식이 아닌 순수한 텍스트로만 출력하라
-4. 반드시 이 표현들은 삭제하라: 여러분, 최근, 마지막으로, 결론적으로, 결국, 종합적으로, 따라서, 마무리, 요약.
+1. 생성된 글의 길이가 목표 글자수에 미달입니다. 내용의 질을 유지하고 더욱 풍부하게 작성하고, 세부사항을 추가하여 자연스럽게 확장하세요.
+2. 중복되는 내용은 피하고 새로운 정보를 추가하세요.
+3. 반드시 마크다운 형식이 아닌 순수한 텍스트로만 출력하세요.
+4. 반드시 이 표현들은 사용하지 마세요: 여러분, 최근, 마지막으로, 결론적으로, 결국, 종합적으로, 따라서, 마무리, 요약.
 """
             else:
                 additional_instructions = ""