Update app.py

app.py

@@ -74,14 +74,22 @@ def get_naver_answers( message: str ):
     naver_start_time = time.time()
     response = gemini_client.models.generate_content(
         model = genai_model,
-        contents=f"{message}\n 위의 내용을 짧은 제목으로 요약합니다. 제목만 보여주세요. 대답하지 마세요",
+        contents=f"{message}\n 위의 내용을 짧은 제목으로 요약합니다. 제목만 보여주세요. 대답하지 마세요. 한국어로만 답변해주세요!!!",
     )
     query = response.text
     print( "Query: ", query)
 
     context = get_answers( query )
 
-    sorted_answers = ['. '.join(answer['answers']) for answer in context]
+    answer_count = 0
+    sorted_answers = []
+    for answer in context:
+        curr = '. '.join(answer['answers'])
+        answer_count += len(curr)
+        sorted_answers.append(curr)
+        if answer_count > 55000:
+            break
+    # sorted_answers = ['. '.join(answer['answers']) for answer in context]
     naver_end_time = time.time()
     print(f"Time taken to extract from Naver: { naver_end_time - naver_start_time } seconds")
     document = '\n'.join(sorted_answers)
@@ -254,12 +262,14 @@ def compare_answers(message: str):
     return results
 
 def chatFunction( message, history ):
-    content = f'아래 문서를 바탕으로 질문에 답하세요. 답변에서 질문을 따라 출력 하지 마세요. 답변은 한국어로만 해주세요. 찾은 Naver 문서와 다른 문서에서 답변이 없는 내용은 절대 출력하지 마세요 \n 질문: {message}\n 문서: '
+    content = f'아래 문서를 바탕으로 질문에 답하세요. 답변에서 질문을 따라 출력 하지 마세요. 답변은 한국어로만 해주세요! 찾은 Naver 문서와 다른 문서에서 답변이 없는 내용은 절대 출력하지 마세요 \n 질문: {message}\n 문서: '
     naver_docs, naver_time_taken = get_naver_answers( message )
 
     start_time = time.time()
     content += "\n Naver 문서: " + naver_docs
 
+    print( ">>> Length: ", len(naver_docs) )
+
     completion = gpt_client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[