Update main.py

main.py

@@ -1,27 +1,23 @@
 from fastapi import FastAPI
+from fastapi.responses import JSONResponse
 import firebase_admin
 from firebase_admin import credentials, firestore
 from transformers import pipeline
 from pydantic import BaseModel
-from typing import List, Optional
-from huggingface_hub import login
-from fastapi.responses import JSONResponse
-
-
 import os
-
-if not os.path.exists("firebase_config.json"):
-    raise FileNotFoundError("Không tìm thấy file firebase_config.json")
-
+from huggingface_hub import login
 
 # Load Firebase
 cred = credentials.Certificate("firebase_config.json")
 firebase_admin.initialize_app(cred)
 db = firestore.client()
 
-# Load AI Model
-login(os.getenv("HF_TOKEN"))
+# Đăng nhập vào Hugging Face (nếu cần)
+HF_TOKEN = os.getenv("HF_TOKEN")
+if HF_TOKEN:
+    login(HF_TOKEN)
 
+# Load AI Model
 ai_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
 
 app = FastAPI()
@@ -57,11 +53,6 @@ async def add_focus_history(request: FocusHistoryRequest):
     user_ref.set({"focus_history": focus_history}, merge=True)
     return {"message": "Thêm lịch sử focus thành công"}
 
-@app.route('/')
-def home():
-    return jsonify({"message": "Welcome to the Recommendation API!"})
-   
-
 # API lấy dữ liệu người dùng
 @app.get("/get_user_data")
 async def get_user_data(user_id: str):
@@ -80,6 +71,7 @@ async def ai_personal_advice(request: AIRequest):
     bios = 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"""
     Thông tin người dùng:
     - Bios: {bios}
@@ -88,5 +80,11 @@ 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.
     """
+    
     response = ai_model(prompt, max_length=300)
     return {"advice": response[0]["generated_text"]}
+
+# Trang chủ
+@app.get("/")
+async def home():
+    return JSONResponse(content={"message": "Welcome to the Recommendation API!"})