Update app.py

app.py

@@ -130,69 +130,87 @@ def fetch_crawl_results(query):
 
 def generate_blog_post(query, prompt_template):
     try:
-        target_length = 2000  # 목표 글자수 설정 (문자 수)
-        max_attempts = 2  # 최대 2번 실행 (초기 1번 + 재시도 1번)
+        # 목표 글자수 설정 (문자 수)
+        target_char_length = 2000
         
+        # 한글에서 1토큰은 약 1~1.5글자 정도이므로 토큰 수로 환산
+        estimated_tokens = int(target_char_length / 1.2)
+        
+        # 참고글 가져오기
         references = fetch_references(query)
         ref1, ref2, ref3 = references
 
         # OpenAI API 설정
-        model_name = "gpt-4"
+        model_name = "gpt-4o-mini"
         temperature = 0.85
-        max_tokens = 3000  # 모델의 제한에 맞게 max_tokens 조정
+        max_tokens = min(10000, estimated_tokens)  # 최대 토큰 수는 모델 제한을 초과하지 않도록 설정
         top_p = 0.9
         frequency_penalty = 0.5
         presence_penalty = 0
 
-        # 제거해야 할 표현 패턴 미리 컴파일
+        # 불필요한 표현 패턴 정의
         unwanted_patterns = [
             re.compile(r'\b여러분[,.]?\s*', re.IGNORECASE),
             re.compile(r'\b(마지막으로|결론적으로|결국|종합적으로|따라서|마무리|요약)[,.]?\s*', re.IGNORECASE)
         ]
 
-        for attempt in range(max_attempts):
-            current_prompt = prompt_template + f"""
+        # 시스템 메시지 설정
+        system_message = """
+        당신은 전문 블로그 작가입니다.
+        아래의 지침을 따르세요:
+        - 반드시 한글로 작성하세요.
+        - 주어진 참고글을 바탕으로 상품 리뷰형 블로그 글을 작성하세요.
+        - 글의 내용은 1500단어 이상이 되도록 하세요.
+        - 제목을 상품 리뷰형 블로그에 적합하게 작성하세요.
+        - 마크다운 형식이 아닌 순수한 텍스트로만 작성하세요.
+        - 참고글의 내용을 충분히 반영하되, 그대로 복사하지 마세요.
+        - 불필요한 표현은 사용하지 마세요: 여러분, 마지막으로, 결론적으로, 결국, 종합적으로, 따라서, 마무리, 요약.
+        - 참고글의 링크나 개인 정보는 포함하지 마세요.
+        - 자연스럽고 친근한 어투로 작성하세요.
+        """
+
+        # 사용자 프롬프트 구성
+        user_prompt = prompt_template + f"""
 주제: {query}
 참고글1: {ref1}
 참고글2: {ref2}
 참고글3: {ref3}
-목표 글자수: {target_length}
-"""
-            if attempt > 0:
-                current_prompt += """
-다음 표현은 사용하지 말 것: 여러분, 마지막으로, 결론적으로, 결국, 종합적으로, 따라서, 마무리, 요약
-이전 생성에서 글자수가 부족하거나 원치 않는 표현이 포함되었다. 수정하여 다시 작성하라.
 """
 
-            # 메시지 구성
-            messages = [
-                {"role": "user", "content": current_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()
-
-            # 생성된 글에서 불필요한 공백 제거
-            generated_post = generated_post.strip()
-
-            # 생성된 글이 조건을 만족하는지 확인
-            if len(generated_post) >= target_length and not any(pattern.search(generated_post) for pattern in unwanted_patterns):
-                break
+        # 메시지 구성
+        messages = [
+            {"role": "system", "content": system_message},
+            {"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
 
         final_post = f"주제: {query}\n\n{generated_post}"
-        actual_length = len(generated_post)
         
-        return final_post, ref1, ref2, ref3, actual_length
+        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