Update app.py

app.py

@@ -132,6 +132,7 @@ def generate_blog_post(query, prompt_template):
     try:
         # 목표 글자수 설정 (문자 수)
         target_char_length = 2000
+        max_attempts = 2  # 최대 시도 횟수 (초기 시도 1회 + 재시도 1회)
 
         # 참고글 가져오기
         references = fetch_references(query)
@@ -145,14 +146,14 @@ def generate_blog_post(query, prompt_template):
         frequency_penalty = 0.5
         presence_penalty = 0
 
-        # 불필요한 표현 패턴 정의
+        # 불필요한 표현 패턴 정의 (하드코딩)
         unwanted_patterns = [
             re.compile(r'\b여러분[,.]?\s*', re.IGNORECASE),
             re.compile(r'\b(마지막으로|결론적으로|결국|종합적으로|따라서|마무리|요약)[,.]?\s*', re.IGNORECASE)
         ]
 
-        # 사용자 프롬프트 구성 (세 개의 참고글 모두 포함)
-        user_prompt = prompt_template + f"""
+        # 초기 프롬프트 구성
+        base_prompt = prompt_template + f"""
 주제: {query}
 참고글 1: {ref1}
 참고글 2: {ref2}
@@ -160,41 +161,59 @@ def generate_blog_post(query, prompt_template):
 목표 글자수: {target_char_length}
 """
 
-        # 메시지 구성
-        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()
-
-        # 생성된 글에서 불필요한 표현 제거
-        for pattern in unwanted_patterns:
-            generated_post = pattern.sub('', generated_post)
-
-        # 생성된 글의 실제 문자 수 계산
-        actual_char_length = len(generated_post)
-
-        # 목표 글자수 미달 시 경고 메시지 추가
-        if actual_char_length < target_char_length:
-            warning_message = f"\n\n[경고] 생성된 글의 길이가 목표 글자수인 {target_char_length}자에 미달합니다. (현재 글자수: {actual_char_length}자)"
-            generated_post += warning_message
+        for attempt in range(max_attempts):
+            # 첫 번째 시도 이후에는 프롬프트에 추가 지침을 포함
+            if attempt > 0:
+                additional_instructions = f"""
+생성된 글의 길이가 목표 글자수에 미달, 내용의 질을 유지하고 더욱 풍부하게 작성하고, 세부사항을 추가하여 자연스럽게 확장하라
+중복되는 내용은 피하고 새로운 정보를 추가하라
+반드시 불필요한 표현은 사용하지 말 것: 여러분, 최근, 마지막으로, 결론적으로, 결국, 종합적으로, 따라서, 마무리, 요약.
+"""
+            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()
+
+            # 불필요한 표현 제거 (코드에서 하드코딩으로 처리)
+            for pattern in unwanted_patterns:
+                generated_post = pattern.sub('', generated_post)
+
+            # 생성된 글의 실제 문자 수 계산
+            actual_char_length = len(generated_post)
+
+            # 글자수가 목표를 충족하면 루프 종료
+            if actual_char_length >= target_char_length:
+                break
+            else:
+                # 로그에 경고 메시지 출력 (사용자에게는 표시되지 않음)
+                print(f"[시도 {attempt + 1}] 생성된 글의 길이가 목표 글자수에 미달합니다. (현재 글자수: {actual_char_length}자)")
+                if attempt == max_attempts - 1:
+                    print("[경고] 최대 시도 횟수에 도달했으나 목표 글자수에 도달하지 못했습니다.")
 
         # 최종 결과물 구성
         final_post = f"주제: {query}\n\n{generated_post}"
 
         return final_post, ref1, ref2, ref3, actual_char_length
+
     except Exception as e:
         print(f"generate_blog_post 함수에서 오류 발생: {str(e)}")
         return f"블로그 글 생성 중 오류 발생: {str(e)}", "", "", "", 0