Update app.py

app.py

@@ -130,13 +130,30 @@ def fetch_crawl_results(query):
 
 def generate_blog_post(query, prompt_template):
     try:
+        # 로깅 설정
+        class KST_Formatter(logging.Formatter):
+            def formatTime(self, record, datefmt=None):
+                dt = datetime.fromtimestamp(record.created, tz=ZoneInfo("Asia/Seoul"))
+                return dt.strftime('%Y-%m-%d %H:%M:%S %Z')
+
+        logging.basicConfig(level=logging.INFO)
+        logger = logging.getLogger(__name__)
+        handler = logging.StreamHandler()
+        formatter = KST_Formatter('%(asctime)s - %(levelname)s - %(message)s')
+        handler.setFormatter(formatter)
+        logger.addHandler(handler)
+
+        logger.info(f"블로그 글 생성 시작: 주제 - {query}")
+
         # 목표 글자수 설정 (문자 수)
         target_char_length = 3000
         max_attempts = 2  # 최대 시도 횟수
 
         # 참고글 가져오기
+        logger.info("참고글 가져오기 시작")
         references = fetch_references(query)
         ref1, ref2, ref3 = references
+        logger.info("참고글 가져오기 완료")
 
         # OpenAI API 설정
         model_name = "gpt-4o-mini"
@@ -146,6 +163,8 @@ def generate_blog_post(query, prompt_template):
         frequency_penalty = 0.5
         presence_penalty = 0.3
 
+        logger.info(f"OpenAI API 설정: 모델 - {model_name}, 온도 - {temperature}")
+
         # 불필요한 표현 제거 함수
         def remove_unwanted_phrases(text):
             unwanted_phrases = [
@@ -165,6 +184,7 @@ def generate_blog_post(query, prompt_template):
 목표 글자수: {target_char_length}
 """
 
+        logger.info("첫 번째 시도 시작")
         # 첫 번째 시도
         messages = [{"role": "user", "content": initial_prompt}]
         response = openai.ChatCompletion.create(
@@ -177,26 +197,40 @@ def generate_blog_post(query, prompt_template):
             presence_penalty=presence_penalty,
         )
         first_attempt = response['choices'][0]['message']['content'].strip()
-
-        # 두 번째 시도를 위한 추가 프롬프트
-        additional_prompt = f"""
-이전에 생성된 글을 기반으로 다음 지침을 반드시 따라서 글을 퇴고(revision)하라
-1. 반드시 이전 글(first_attempt)의 구조와 내용을 유지하면서 내용을 보완하라
+        logger.info("첫 번째 시도 완료")
+        
+        # 불필요한 표현 제거 및 글자수 확인
+        first_attempt_cleaned = remove_unwanted_phrases(first_attempt)
+        first_attempt_length = len(first_attempt_cleaned)
+        logger.info(f"첫 번째 시도 정제 후 글자수: {first_attempt_length}")
+
+        # 첫 번째 시도에서 목표 글자수 충족 시
+        if first_attempt_length >= target_char_length:
+            logger.info("첫 번째 시도에서 목표 글자수 충족")
+            final_post = f"주제: {query}\n\n{first_attempt_cleaned}"
+            return final_post, ref1, ref2, ref3, first_attempt_length
+
+        logger.info("두 번째 시도 (퇴고) 준비")
+        # 두 번째 시도 (퇴고)를 위한 추가 프롬프트
+        revision_prompt = f"""
+이전에 생성된 글을 기반으로 다음 지침을 반드시 따라서 글을 퇴고(revision)하세요:
+1. 반드시 이전 글의 구조와 내용을 유지하면서 내용을 보완하세요.
 2. 반드시 마크다운 형식이 아닌 순수한 텍스트로만 출력하세요.
 3. 반드시 이 표현들은 사용하지 마세요: 여러분, 최근, 마지막으로, 결론적으로, 결국, 종합적으로, 따라서, 마무리, 요약.
 4. 글의 흐름을 자연스럽게 만들고, 각 단락 간의 연결을 부드럽게 해주세요.
 5. 반드시 목표 글자수({target_char_length}자)를 충족하도록 내용을 확장하세요.
 
 이전 글:
-{first_attempt}
+{first_attempt_cleaned}
 
 참고글 1: {ref1}
 참고글 2: {ref2}
 참고글 3: {ref3}
 """
 
-        # 두 번째 시도
-        messages = [{"role": "user", "content": additional_prompt}]
+        logger.info("두 번째 시도 (퇴고) 시작")
+        # 두 번째 시도 (퇴고)
+        messages = [{"role": "user", "content": revision_prompt}]
         response = openai.ChatCompletion.create(
             model=model_name,
             messages=messages,
@@ -206,19 +240,22 @@ def generate_blog_post(query, prompt_template):
             frequency_penalty=frequency_penalty,
             presence_penalty=presence_penalty,
         )
-        second_attempt = response['choices'][0]['message']['content'].strip()
+        revised_attempt = response['choices'][0]['message']['content'].strip()
+        logger.info("두 번째 시도 (퇴고) 완료")
 
         # 불필요한 표현 제거
-        final_post = remove_unwanted_phrases(second_attempt)
+        final_post = remove_unwanted_phrases(revised_attempt)
 
         # 최종 결과물 구성
         final_post = f"주제: {query}\n\n{final_post}"
         actual_char_length = len(final_post)
+        logger.info(f"최종 글자수: {actual_char_length}")
 
+        logger.info("블로그 글 생성 완료")
         return final_post, ref1, ref2, ref3, actual_char_length
 
     except Exception as e:
-        print(f"generate_blog_post 함수에서 오류 발생: {str(e)}")
+        logger.error(f"generate_blog_post 함수에서 오류 발생: {str(e)}")
         return f"블로그 글 생성 중 오류 발생: {str(e)}", "", "", "", 0
         
 # PDF 클래스 및 관련 함수 정의