Update app.py

app.py

@@ -146,33 +146,18 @@ def generate_blog_post(query, prompt_template):
         frequency_penalty = 0.5
         presence_penalty = 0.3
 
-        # 불필요한 표현 리스트
-        unwanted_phrases = [
-            '여러분',
-            '최근',
-            '마지막으로',
-            '결론적으로',
-            '결국',
-            '종합적으로',
-            '따라서',
-            '마무리',
-            '요약'            
-        ]
-
-        # 불필요한 표현이 포함된 단어 전체를 제거하는 함수
+        # 불필요한 표현 제거 함수
         def remove_unwanted_phrases(text):
-            # 단어 단위로 텍스트를 분할
+            unwanted_phrases = [
+                '여러분', '최근', '마지막으로', '결론적으로', '결국',
+                '종합적으로', '따라서', '마무리', '요약'
+            ]
             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)
-            # 단어들을 다시 조합하여 결과 텍스트 생성
+            result_words = [word for word in words if not any(phrase in word for phrase in unwanted_phrases)]
             return ' '.join(result_words).replace(' \n ', '\n').replace(' \n', '\n').replace('\n ', '\n')
 
         # 초기 프롬프트 구성
-        base_prompt = prompt_template + f"""
+        initial_prompt = prompt_template + f"""
 주제: {query}
 참고글 1: {ref1}
 참고글 2: {ref2}
@@ -180,56 +165,55 @@ def generate_blog_post(query, prompt_template):
 목표 글자수: {target_char_length}
 """
 
-        for attempt in range(max_attempts):
-            # 첫 번째 시도 이후에는 프롬프트에 추가 지침을 포함
-            if attempt > 0:
-                additional_instructions = f"""
-1. 생성된 글의 길이가 목표 글자수에 미달입니다. 내용의 질을 유지하고 더욱 풍부하게 작성하고, 세부사항을 추가하여 자연스럽게 확장하세요.
-2. 기존에 작성한 글은 반드시 유지하되, 중복되는 내용은 피하고 새로운 정보를 추가하세요.
-3. 반드시 마크다운 형식이 아닌 순수한 텍스트로만 출력하세요.
-4. 반드시 이 표현들은 사용하지 마세요: 여러분, 최근, 마지막으로, 결론적으로, 결국, 종합적으로, 따라서, 마무리, 요약.
-"""
-            else:
-                additional_instructions = ""
-
-            user_prompt = base_prompt + additional_instructions
-
-            # 메시지 구성
-            messages = [
-                {"role": "user", "content": user_prompt}
-            ]
-
-            # OpenAI API 호출
-            response = openai.ChatCompletion.create(
-                model=model_name,
-                messages=messages,
-                temperature=temperature,
-                max_tokens=max_tokens,
-                top_p=top_p,
-                frequency_penalty=frequency_penalty,
-                presence_penalty=presence_penalty,
-            )
-
-            # 모델이 생성한 답변 추출
-            generated_post = response['choices'][0]['message']['content'].strip()
+        # 첫 번째 시도
+        messages = [{"role": "user", "content": initial_prompt}]
+        response = openai.ChatCompletion.create(
+            model=model_name,
+            messages=messages,
+            temperature=temperature,
+            max_tokens=max_tokens,
+            top_p=top_p,
+            frequency_penalty=frequency_penalty,
+            presence_penalty=presence_penalty,
+        )
+        first_attempt = response['choices'][0]['message']['content'].strip()
+
+        # 두 번째 시도를 위한 추가 프롬프트
+        additional_prompt = f"""
+이전에 생성된 글을 기반으로 다음 지침을 반드시 따라서 글을 퇴고(revision)하라
+1. 반드시 이전 글(first_attempt)의 구조와 내용을 유지하면서 내용을 보완하라
+2. 반드시 마크다운 형식이 아닌 순수한 텍스트로만 출력하세요.
+3. 반드시 이 표현들은 사용하지 마세요: 여러분, 최근, 마지막으로, 결론적으로, 결국, 종합적으로, 따라서, 마무리, 요약.
+4. 글의 흐름을 자연스럽게 만들고, 각 단락 간의 연결을 부드럽게 해주세요.
+5. 반드시 목표 글자수({target_char_length}자)를 충족하도록 내용을 확장하세요.
+
+이전 글:
+{first_attempt}
 
-            # 불필요한 표현 제거 함수 적용
-            generated_post = remove_unwanted_phrases(generated_post)
-
-            # 생성된 글의 실제 문자 수 계산
-            actual_char_length = len(generated_post)
+참고글 1: {ref1}
+참고글 2: {ref2}
+참고글 3: {ref3}
+"""
 
-            # 글자수가 목표를 충족하면 루프 종료
-            if actual_char_length >= target_char_length:
-                break
-            else:
-                # 로그에 경고 메시지 출력 (사용자에게는 표시되지 않음)
-                print(f"[시도 {attempt + 1}] 생성된 글의 길이가 목표 글자수에 미달합니다. (현재 글자수: {actual_char_length}자)")
-                if attempt == max_attempts - 1:
-                    print("[경고] 최대 시도 횟수에 도달했으나 목표 글자수에 도달하지 못했습니다.")
+        # 두 번째 시도
+        messages = [{"role": "user", "content": additional_prompt}]
+        response = openai.ChatCompletion.create(
+            model=model_name,
+            messages=messages,
+            temperature=temperature,
+            max_tokens=max_tokens,
+            top_p=top_p,
+            frequency_penalty=frequency_penalty,
+            presence_penalty=presence_penalty,
+        )
+        second_attempt = response['choices'][0]['message']['content'].strip()
+
+        # 불필요한 표현 제거
+        final_post = remove_unwanted_phrases(second_attempt)
 
         # 최종 결과물 구성
-        final_post = f"주제: {query}\n\n{generated_post}"
+        final_post = f"주제: {query}\n\n{final_post}"
+        actual_char_length = len(final_post)
 
         return final_post, ref1, ref2, ref3, actual_char_length