Update main.py

main.py

@@ -112,13 +112,17 @@ async def ai_personal_advice(request: AIRequest):
             raise HTTPException(status_code=500, detail="Firebase not initialized.")
 
         user_doc = db.collection("user_profiles").document(request.user_id).get()
+
+        # Check if the document exists
         if not user_doc.exists:
             logging.warning(f"User profile not found for user_id: {request.user_id}")
-            return {"advice": "Không tìm thấy thông tin người dùng."}
+            bios = "Không có thông tin người dùng."
+            focus_history = []
+        else:
+            data = user_doc.to_dict() or {}
+            bios = request.bios if request.bios else data.get("bios", "Chưa có bios.")
+            focus_history = data.get("focus_history", [])
 
-        data = user_doc.to_dict() or {}
-        bios = request.bios if request.bios else data.get("bios", "Chưa có bios.")
-        focus_history = data.get("focus_history", [])
         focus_text = "\n".join([f"- {f['time_start']}: {f['total_time']} phút" for f in focus_history])
 
         prompt = f"""
@@ -129,10 +133,16 @@ async def ai_personal_advice(request: AIRequest):
 
         Hãy tư vấn cách cải thiện hiệu suất làm việc dựa trên thông tin trên.
         """
+
+        # Truncate prompt if it's too long
         input_ids = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=1024)
         response = ai_model.generate(**input_ids, max_new_tokens=500)
         advice = tokenizer.decode(response[0], skip_special_tokens=True)
 
+        # Đảm bảo câu trả lời là một câu hoàn chỉnh
+        if not advice.endswith("."):
+            advice += "."
+
         return {"advice": advice}
 
     except Exception as e:
@@ -141,6 +151,7 @@ async def ai_personal_advice(request: AIRequest):
         logging.error(traceback.format_exc())
         raise HTTPException(status_code=500, detail=error_message)
 
+
 # Trang chủ
 @app.get("/")
 async def home():