a

Dockerfile

@@ -1,60 +1,23 @@
-FROM ghcr.io/huggingface/chat-ui:latest AS base
+# Use the official Python 3.9 image from the Docker Hub
+FROM python:3.9-slim
 
-FROM ghcr.io/huggingface/text-generation-inference:latest AS final
+# Set the working directory in the container
+WORKDIR /app
 
-ARG MODEL_NAME
-ENV MODEL_NAME=${MODEL_NAME}
+# Copy the requirements.txt file into the container at /app
+COPY requirements.txt /app/
 
-ENV TZ=Europe/Paris \
-    PORT=3000
+# Install the dependencies from requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
-# mongo installation
-RUN curl -fsSL https://www.mongodb.org/static/pgp/server-7.0.asc | \
-    gpg -o /usr/share/keyrings/mongodb-server-7.0.gpg \
-   --dearmor
+# Copy the rest of the application code into the container
+COPY . /app/
 
-RUN echo "deb [ arch=amd64,arm64 signed-by=/usr/share/keyrings/mongodb-server-7.0.gpg ] https://repo.mongodb.org/apt/ubuntu jammy/mongodb-org/7.0 multiverse" | tee /etc/apt/sources.list.d/mongodb-org-7.0.list
+# Expose the port your app will run on (assuming Flask/Django app listens on port 5000)
+EXPOSE 5000
 
-RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
-    mongodb-org && \
-    rm -rf /var/lib/apt/lists/*
+# Command to run the app with Gunicorn (for production use)
+CMD ["gunicorn", "app:app", "--bind", "0.0.0.0:$PORT"]
 
-# node installation
-RUN curl -fsSL https://deb.nodesource.com/setup_20.x | /bin/bash -
-
-RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
-    nodejs && \
-    rm -rf /var/lib/apt/lists/*
-
-# image setup
-RUN useradd -m -u 1000 user
-
-RUN mkdir /app
-RUN chown -R 1000:1000 /app
-RUN mkdir /data
-RUN chown -R 1000:1000 /data
-
-# Switch to the "user" user
-USER user
-
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH
-
-RUN npm config set prefix /home/user/.local
-RUN npm install -g dotenv-cli
-
-
-# copy chat-ui from base image
-COPY --from=base --chown=1000 /app/node_modules /app/node_modules
-COPY --from=base --chown=1000 /app/package.json /app/package.json
-COPY --from=base --chown=1000 /app/build /app/build
-
-COPY --from=base --chown=1000 /app/.env /app/.env
-COPY --chown=1000 .env.local /app/.env.local
-
-COPY --chown=1000 entrypoint.sh /app/entrypoint.sh
-
-RUN chmod +x /app/entrypoint.sh
-
-# entrypoint
-ENTRYPOINT [ "/app/entrypoint.sh" ]
+# For development (uncomment the following line if you're in development mode):
+# CMD ["python", "app.py"]

Procfile

@@ -0,0 +1 @@
+web: gunicorn app:app --bind 0.0.0.0:$PORT

README.md

@@ -1,13 +1,4 @@
----
-title: DrugRecommendation
-emoji: 🚀
-colorFrom: indigo
-colorTo: blue
-sdk: docker
-pinned: false
-app_port: 3000
-suggested_hardware: a10g-small
-short_description: This is a system support to recommend drugs based on symptom
----
+RUN APP:
+    pip install -r requirements.txt
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+    python app.py

app.py

@@ -0,0 +1,394 @@
+from flask import Flask, request, jsonify, render_template,make_response,redirect, url_for, session
+from flask_pymongo import PyMongo
+from bson.objectid import ObjectId
+import bcrypt
+from flask_cors import CORS
+from sentence_transformers import SentenceTransformer, util
+import faiss
+import numpy as np
+import json
+import os
+import pickle
+import pandas as pd
+import google.generativeai as genai
+from datetime import datetime
+import uuid
+from gemini_handler import GeminiHandler, GenerationConfig, Strategy, KeyRotationStrategy
+from dotenv import load_dotenv
+# Load biến môi trường từ file .env
+load_dotenv()
+GEMINI_API_KEYS = os.getenv("GEMINI_API_KEYS").split(",")
+
+# Khởi tạo Flask app
+app = Flask(__name__)
+
+app.secret_key = 'AIzaSyCvlZ63Nkt5NpjdmxYPAsG8Qskex6usCFw'
+app.config['MONGO_URI'] = 'mongodb+srv://admin:[email protected]/drug_recom'
+mongo = PyMongo(app)
+
+
+
+
+CORS(app)
+
+genai.configure(api_key="AIzaSyCvlZ63Nkt5NpjdmxYPAsG8Qskex6usCFw")
+
+# Tải FAISS index
+faiss_index = faiss.read_index("source/faiss_index_vn.bin")
+
+# Tải embeddings
+with open("source/sentence_embeddings_vn.pkl", "rb") as f:
+    disease_embeddings = pickle.load(f)
+
+# Khởi tạo mô hình Sentence Transformer
+model_em = SentenceTransformer('hiieu/halong_embedding')
+# Đọc dữ liệu bệnh
+merged_df = pd.read_csv('source/merged_df_vn.csv')
+
+
+def get_disease_and_generate_prompt(symptoms_input, faiss_index, model_em, merged_df, top_k=5):
+    # 1. Mã hóa triệu chứng đầu vào
+    input_embedding = model_em.encode([symptoms_input], convert_to_tensor=True)
+
+    # 2. Tìm kiếm top_k trong FAISS Index
+    distances, indices = faiss_index.search(np.array(input_embedding.cpu().numpy()), k=top_k)
+
+    # Lấy danh sách các bệnh và điểm tương ứng
+    top_diseases = [(merged_df.iloc[idx], score) for idx, score in zip(indices[0], distances[0])]
+
+    # 3. Mã hóa thông tin của top_k bệnh để đánh giá lại
+    candidate_embeddings = model_em.encode(
+        [disease['Information'] for disease, _ in top_diseases],
+        convert_to_tensor=True
+    )
+
+    # 4. Tính độ tương đồng cosine giữa triệu chứng đầu vào và các ứng viên
+    scores = util.cos_sim(input_embedding, candidate_embeddings).squeeze()
+
+    # 5. Sắp xếp lại danh sách ứng viên dựa trên điểm similarity
+    ranked_indices = scores.argsort(descending=True)
+    best_match = top_diseases[ranked_indices[0].item()][0]  # Ứng viên tốt nhất sau re-ranking
+
+     # 6. Chuyển thông tin bệnh tốt nhất thành danh sách
+    result_list = [
+        f"Patient Symptoms: {symptoms_input}",
+        f"similarity: {scores}",
+        # f"disease: {best_match['Disease']}",
+        f"symptoms: {best_match['Symptoms']}",
+        f"medications: {best_match['Medication']}",
+        f"diets: {best_match['Diet']}",
+        f"workouts: {best_match['workout']}",
+        f"precautions: {best_match.get('Precaution_1', '')}, {best_match.get('Precaution_2', '')}, {best_match.get('Precaution_3', '')}, {best_match.get('Precaution_4', '')}",
+    ]
+    #print(result_list)
+    return result_list
+
+
+
+def parse_contexts(raw_contexts):
+    """Convert raw context strings into structured dictionaries."""
+    structured_contexts = []
+    temp_context = {}
+    for context in raw_contexts:
+        # Check for each expected piece of information and assign it to the dictionary
+        if context.startswith("Patient Symptoms:"):
+            temp_context["patient_symptoms"] = context.replace("Patient Symptoms:", "").strip()
+        elif context.startswith("disease"):
+            temp_context["disease"] = context.replace("disease", "").strip()
+        elif context.startswith("symptoms:"):
+            temp_context["symptoms"] = context.replace("symptoms:", "").strip().split(", ")
+        elif context.startswith("medications:"):
+            temp_context["medications"] = context.replace("medications:", "").strip().split(", ")
+        elif context.startswith("diets:"):
+            temp_context["diets"] = context.replace("diets:", "").strip().split(", ")
+        elif context.startswith("workouts:"):
+            temp_context["workouts"] = context.replace("workouts:", "").strip().split(", ")
+        elif context.startswith("precautions:"):
+            temp_context["precautions"] = context.replace("precautions:", "").strip().split(", ")
+
+        # When all necessary fields are collected, add to the list and reset temp_context
+        if len(temp_context) == 7:
+            structured_contexts.append(temp_context)
+            temp_context = {}
+
+    return structured_contexts
+
+prompt_template = (
+    "### Hệ thống:"
+    "Bạn đang nhận được một yêu cầu tư vấn y tế. Dưới đây là thông tin cần thiết để bạn đưa ra câu trả lời chính xác, dễ hiểu và hữu ích cho người dùng."
+    "Hãy tập trung vào việc cung cấp tên thuốc phù hợp và hướng dẫn rõ ràng để giúp người dùng áp dụng dễ dàng."
+    "### Hướng dẫn:"
+    "{instruction}\n\n"
+
+    "### Thông tin y tế:\n" 
+    "{input}\n\n"
+
+    "### Câu trả lời:\n" 
+    "{output}" 
+    )
+
+def get_prompt(question, raw_contexts):
+    if not raw_contexts:
+        raise ValueError("Danh sách thông tin y tế không được để trống.")
+
+    # Xử lý dữ liệu đầu vào thành dạng dễ đọc
+    contexts = parse_contexts(raw_contexts)
+    
+    context = "".join([
+        f"\n<b>📌 Trường hợp {i+1}:</b>\n"
+        #f"- <b>Bệnh:</b> {x.get('disease', 'Chưa xác định')}\n"
+        f"- <b>Triệu chứng:</b> {', '.join(map(str, x.get('symptoms', [])))}\n"
+        f"- <b>Thuốc đề xuất:</b> <i>{', '.join(map(str, x.get('medications', [])))}</i>\n"
+        f"- <b>Chế độ ăn uống:</b> {', '.join(map(str, x.get('diets', [])))}\n"
+        f"- <b>Bài tập hỗ trợ:</b> {', '.join(map(str, x.get('workouts', [])))}\n"
+        f"- <b>Lưu ý quan trọng:</b> {', '.join(map(str, x.get('precautions', [])))}\n"
+        for i, x in enumerate(contexts)
+    ])
+
+    instruction = (
+        "💊 Bạn là một dược sĩ có kinh nghiệm lâu năm. Hãy cung cấp câu trả lời <b>đầy đủ, chính xác</b>, "
+        "dễ hiểu và tập trung vào <b>tư vấn thuốc</b> cho người dùng.\n"
+        "🔹 Trình bày câu trả lời theo danh sách số thứ tự (1️⃣, 2️⃣, 3️⃣...) để dễ đọc.\n"
+        "🔹 Định dạng rõ ràng: <b>in đậm</b> những điểm quan trọng, <i>in nghiêng</i> tên thuốc, <b>in đậm</b> các lưu ý quan trọng, "
+        "sử dụng dấu gạch đầu dòng (-) để liệt kê thông tin." 
+    )
+
+    input_text = (
+        "🩺 Dựa trên thông tin y tế sau đây, hãy trả lời câu hỏi của người dùng:\n"
+        f"{context}\n"
+        "❓ <b>Câu hỏi:</b> " + question + "\n"
+        "📌 <b>Yêu cầu:</b> Hãy trả lời theo bố cục số thứ tự, dễ đọc, ngắn gọn nhưng đầy đủ, giúp người dùng dễ áp dụng.\n"
+        "📋 <b>Định dạng:</b>\n"
+        "- <b>In đậm</b> cho thông tin quan trọng\n"
+        "- <i>In nghiêng</i> cho tên thuốc\n"
+        "- <b>In đậm</b> cho các lưu ý đặc biệt\n"
+        "- Dấu gạch đầu dòng (-) để trình bày rõ ràng khi liệt kê"
+    )
+
+    prompt = prompt_template.format(
+        instruction=instruction,
+        input=input_text,
+        output=''  # AI sẽ tự điền câu trả lời
+    )
+
+    return prompt
+
+@app.route("/send_message", methods=["POST"])
+def send_message():
+    try:
+        data = request.json
+        print("📩 Received Data:", data)  # Debug log
+
+        # Kiểm tra request JSON hợp lệ
+        if not data or "message" not in data:
+            return jsonify({"error": "Dữ liệu không hợp lệ!"}), 400
+
+        message = data.get("message", "").strip()
+        conversation_id = data.get("conversation_id", "")
+
+        if not message:
+            return jsonify({"error": "Câu hỏi không hợp lệ!"}), 400
+
+        # Nếu không có conversation_id, tạo cuộc trò chuyện mới
+        if not conversation_id:
+            conversation_id = str(uuid.uuid4())
+            session["conversation_id"] = conversation_id
+            mongo.db.chat_history.insert_one({
+                "conversation_id": conversation_id,
+                "name": "Chưa có tên",
+                "messages": [],
+                "created_at": datetime.now().strftime("%d/%m/%Y %H:%M:%S")
+            })
+
+        # Gọi hàm lấy dữ liệu context
+        context_data = get_disease_and_generate_prompt(message, faiss_index, model_em, merged_df, top_k=2)
+        prompt = get_prompt(message, context_data)
+
+        # Get API keys từ environment
+        api_keys = os.getenv('GEMINI_API_KEYS')
+        if not api_keys:
+            raise ValueError("GEMINI_API_KEYS environment variable is not set")
+
+
+        # Round-robin strategy
+        handler = GeminiHandler(
+            config_path="config.yaml",
+            content_strategy=Strategy.ROUND_ROBIN,
+            key_strategy=KeyRotationStrategy.SMART_COOLDOWN
+        )
+
+                # Generate content
+        response_new = handler.generate_content(
+            prompt=prompt,
+            model_name="gemini-2.0-flash-thinking-exp-1219",
+            return_stats=True  # Get key usage stats
+        )
+
+                # Process response
+        # if response_new['success']:
+        #     print(response_new['text'])
+        # else:
+        #     print(f"Generation failed: {response_new['error']}")
+
+        # # Gửi prompt đến model Gemini
+        # model = genai.GenerativeModel("gemini-1.5-flash", generation_config=generation_config)
+        # chat_session = model.start_chat(history=[])
+        # response = chat_session.send_message(prompt)
+
+        # Lấy nội dung phản hồi từ bot
+        try:
+            text_response = response_new['text']
+        except Exception as e:
+            print(f"⚠️ Error getting response: {e}")
+            return jsonify({"error": "Lỗi khi lấy phản hồi từ mô hình!"}), 500
+
+        # Cấu trúc tin nhắn
+        timestamp = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
+        user_message = {"text": message, "timestamp": timestamp, "sender": "user"}
+        bot_message = {"text": text_response, "timestamp": timestamp, "sender": "bot"}
+
+        # Cập nhật cuộc trò chuyện
+        mongo.db.chat_history.update_one(
+            {"conversation_id": conversation_id},
+            {"$push": {"messages": {"$each": [user_message, bot_message]}}},
+            upsert=True
+        )
+
+        # Trả về phản hồi JSON
+        return jsonify({
+            "conversation_id": conversation_id,
+            "status": "sent",
+            "bot_reply": text_response,
+            "timestamp": timestamp
+        })
+
+    except Exception as e:
+        print(f"🚨 Error in /query: {str(e)}")
+        return jsonify({"error": f"Lỗi xử lý: {str(e)}"}), 500
+
+
+@app.route("/start_conversation", methods=["POST"])
+def start_conversation():
+    try:
+        data = request.json
+        message = data.get("message", "").strip()
+
+        if not message:
+            return jsonify({"error": "Tin nhắn không hợp lệ!"}), 400
+
+        conversation_id = str(uuid.uuid4())  # Tạo conversation_id mới
+        session["conversation_id"] = conversation_id
+
+        mongo.db.chat_history.insert_one({
+            "conversation_id": conversation_id,
+            "name": message,
+            "messages": [],
+            "created_at": datetime.now().strftime("%d/%m/%Y %H:%M:%S")
+        })
+
+        return jsonify({"conversation_id": conversation_id})
+    
+    except Exception as e:
+        print(f"🚨 Error in /start_conversation: {str(e)}")
+        return jsonify({"error": f"Lỗi xử lý: {str(e)}"}), 500
+
+
+@app.route("/all_history", methods=["GET"])
+def get_all_history():
+    """Lấy danh sách tất cả cuộc trò chuyện"""
+    chats = list(mongo.db.chat_history.find({}, {"_id": 0, "messages": 0}))
+    return jsonify(chats)
+
+
+@app.route("/conversation/<conversation_id>", methods=["GET"])
+def get_conversation(conversation_id):
+    """Lấy nội dung cuộc trò chuyện theo ID"""
+    chat = mongo.db.chat_history.find_one({"conversation_id": conversation_id}, {"_id": 0})
+    if not chat:
+        return jsonify({"error": "Cuộc trò chuyện không tồn tại"}), 404
+    return jsonify(chat)
+
+@app.route('/get_conversations', methods=['GET'])
+def get_conversations():
+    # Lấy tất cả cuộc trò chuyện từ MongoDB
+    conversations = list(mongo.db.chat_history.find({}, {"_id": 0}))
+
+    return jsonify([{
+        "conversation_id": conv.get("conversation_id", "N/A"),
+        "name": conv.get("name", "Chưa có tên"),
+        "bot_messages": [msg["text"] for msg in conv.get("messages", []) if msg.get("sender") == "bot"],  # Chỉ lấy text của bot
+        "created_at": conv.get("created_at", "Không rõ ngày")
+    } for conv in conversations])
+
+@app.route('/get_messages', methods=['GET'])
+def get_messages():
+    conversation_id = request.args.get("conversation_id")
+    
+    if not conversation_id:
+        return jsonify({"error": "Thiếu conversation_id"}), 400
+
+    # Truy vấn cuộc trò chuyện theo conversation_id
+    conversation = mongo.db.chat_history.find_one(
+        {"conversation_id": conversation_id}, 
+        {"_id": 0, "messages": 1,"name":1, "timestamp":1}
+    )
+
+    if conversation:
+        return jsonify(conversation)  # Trả về danh sách tin nhắn
+    else:
+        return jsonify({"error": "Không tìm thấy cuộc trò chuyện"}), 404
+
+
+
+@app.route("/")
+def index():
+    return render_template('index.html')
+
+@app.route('/register', methods=['GET', 'POST'])
+def register():
+    if request.method == 'POST':
+        fullname = request.form['fullname']
+        username = request.form['username']
+        password = request.form['password']
+        hash_password = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
+        
+        # Insert user into the 'user' collection
+        mongo.db.user.insert_one({'fullname': fullname,'username': username, 'password': hash_password})
+        return redirect(url_for('login'))
+    return render_template('register.html')
+
+@app.route('/login', methods=['GET', 'POST'])
+def login():
+    if request.method == 'POST':
+        username = request.form['username']
+        password = request.form['password']
+        
+        # Find user in the 'user' collection
+        user = mongo.db.user.find_one({'username': username})
+
+        # Kiểm tra nếu người dùng tồn tại và mật khẩu đúng
+        if user and bcrypt.checkpw(password.encode('utf-8'), user['password']):
+            # Lưu fullname của người dùng vào session
+            session['user'] = user.get('fullname', 'Unknown User')  # Đảm bảo lấy giá trị 'fullname' nếu có
+            return redirect(url_for('dashboard'))
+        else:
+            return 'Invalid username or password'
+    return render_template('login.html')
+    
+@app.route("/dashboard")
+def dashboard():
+    if 'user' in session:
+        current_time = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
+        username = session["user"]   
+        return render_template('dashboard.html',username=username,current_time=current_time)
+    return redirect(url_for('login'))
+
+@app.route('/logout')
+def logout():
+    session.pop('user', None)
+    return redirect(url_for('login'))
+
+
+
+if __name__ == "__main__":
+    app.run(debug=True)

apps.py

@@ -0,0 +1,372 @@
+from flask import Flask, request, jsonify, render_template,make_response,redirect, url_for, session
+from flask_pymongo import PyMongo
+from bson.objectid import ObjectId
+import bcrypt
+from flask_cors import CORS
+from sentence_transformers import SentenceTransformer, util
+import faiss
+import numpy as np
+import json
+import os
+import pickle
+import pandas as pd
+import google.generativeai as genai
+from datetime import datetime
+import uuid
+
+# Khởi tạo Flask app
+app = Flask(__name__)
+
+app.secret_key = 'AIzaSyCvlZ63Nkt5NpjdmxYPAsG8Qskex6usCFw'
+app.config['MONGO_URI'] = 'mongodb+srv://hoangsontruonghcm:[email protected]/drug_recom'
+mongo = PyMongo(app)
+
+
+
+
+CORS(app)
+
+genai.configure(api_key="AIzaSyCvlZ63Nkt5NpjdmxYPAsG8Qskex6usCFw")
+
+# Tải FAISS index
+faiss_index = faiss.read_index("source/faiss_index_vn.bin")
+
+# Tải embeddings
+with open("source/sentence_embeddings_vn.pkl", "rb") as f:
+    disease_embeddings = pickle.load(f)
+
+# Khởi tạo mô hình Sentence Transformer
+model_em = SentenceTransformer('hiieu/halong_embedding')
+# Đọc dữ liệu bệnh
+merged_df = pd.read_csv('source/merged_df_vn.csv')
+
+
+def get_disease_and_generate_prompt(symptoms_input, faiss_index, model_em, merged_df, top_k=5):
+    # 1. Mã hóa triệu chứng đầu vào
+    input_embedding = model_em.encode([symptoms_input], convert_to_tensor=True)
+
+    # 2. Tìm kiếm top_k trong FAISS Index
+    distances, indices = faiss_index.search(np.array(input_embedding.cpu().numpy()), k=top_k)
+
+    # Lấy danh sách các bệnh và điểm tương ứng
+    top_diseases = [(merged_df.iloc[idx], score) for idx, score in zip(indices[0], distances[0])]
+
+    # 3. Mã hóa thông tin của top_k bệnh để đánh giá lại
+    candidate_embeddings = model_em.encode(
+        [disease['Information'] for disease, _ in top_diseases],
+        convert_to_tensor=True
+    )
+
+    # 4. Tính độ tương đồng cosine giữa triệu chứng đầu vào và các ứng viên
+    scores = util.cos_sim(input_embedding, candidate_embeddings).squeeze()
+
+    # 5. Sắp xếp lại danh sách ứng viên dựa trên điểm similarity
+    ranked_indices = scores.argsort(descending=True)
+    best_match = top_diseases[ranked_indices[0].item()][0]  # Ứng viên tốt nhất sau re-ranking
+
+     # 6. Chuyển thông tin bệnh tốt nhất thành danh sách
+    result_list = [
+        f"Patient Symptoms: {symptoms_input}",
+        f"similarity: {scores}",
+        f"disease: {best_match['Disease']}",
+        f"symptoms: {best_match['Symptoms']}",
+        f"medications: {best_match['Medication']}",
+        f"diets: {best_match['Diet']}",
+        f"workouts: {best_match['workout']}",
+        f"precautions: {best_match.get('Precaution_1', '')}, {best_match.get('Precaution_2', '')}, {best_match.get('Precaution_3', '')}, {best_match.get('Precaution_4', '')}",
+    ]
+
+    return result_list
+
+
+
+def parse_contexts(raw_contexts):
+    """Convert raw context strings into structured dictionaries."""
+    structured_contexts = []
+    temp_context = {}
+    for context in raw_contexts:
+        # Check for each expected piece of information and assign it to the dictionary
+        if context.startswith("Patient Symptoms:"):
+            temp_context["patient_symptoms"] = context.replace("Patient Symptoms:", "").strip()
+        elif context.startswith("disease"):
+            temp_context["disease"] = context.replace("disease", "").strip()
+        elif context.startswith("symptoms:"):
+            temp_context["symptoms"] = context.replace("symptoms:", "").strip().split(", ")
+        elif context.startswith("medications:"):
+            temp_context["medications"] = context.replace("medications:", "").strip().split(", ")
+        elif context.startswith("diets:"):
+            temp_context["diets"] = context.replace("diets:", "").strip().split(", ")
+        elif context.startswith("workouts:"):
+            temp_context["workouts"] = context.replace("workouts:", "").strip().split(", ")
+        elif context.startswith("precautions:"):
+            temp_context["precautions"] = context.replace("precautions:", "").strip().split(", ")
+
+        # When all necessary fields are collected, add to the list and reset temp_context
+        if len(temp_context) == 7:
+            structured_contexts.append(temp_context)
+            temp_context = {}
+
+    return structured_contexts
+
+prompt_template = (
+    "### Hệ thống:"
+    "Bạn đang nhận được một yêu cầu tư vấn y tế. Dưới đây là thông tin cần thiết để bạn đưa ra câu trả lời chính xác, dễ hiểu và hữu ích cho người dùng."
+    "Hãy tập trung vào việc cung cấp tên thuốc phù hợp và hướng dẫn rõ ràng để giúp người dùng áp dụng dễ dàng."
+    "### Hướng dẫn:"
+    "{instruction}\n\n"
+
+    "### Thông tin y tế:\n" 
+    "{input}\n\n"
+
+    "### Câu trả lời:\n" 
+    "{output}" 
+    )
+
+def get_prompt(question, raw_contexts):
+    if not raw_contexts:
+        raise ValueError("Danh sách thông tin y tế không được để trống.")
+
+    # Xử lý dữ liệu đầu vào thành dạng dễ đọc
+    contexts = parse_contexts(raw_contexts)
+    
+    context = "".join([
+        f"\n<b>📌 Trường hợp {i+1}:</b>\n"
+        f"- <b>Bệnh:</b> {x.get('disease', 'Chưa xác định')}\n"
+        f"- <b>Triệu chứng:</b> {', '.join(map(str, x.get('symptoms', [])))}\n"
+        f"- <b>Thuốc đề xuất:</b> <i>{', '.join(map(str, x.get('medications', [])))}</i>\n"
+        f"- <b>Chế độ ăn uống:</b> {', '.join(map(str, x.get('diets', [])))}\n"
+        f"- <b>Bài tập hỗ trợ:</b> {', '.join(map(str, x.get('workouts', [])))}\n"
+        f"- <b>Lưu ý quan trọng:</b> {', '.join(map(str, x.get('precautions', [])))}\n"
+        for i, x in enumerate(contexts)
+    ])
+
+    instruction = (
+        "💊 Bạn là một dược sĩ có kinh nghiệm lâu năm. Hãy cung cấp câu trả lời <b>đầy đủ, chính xác</b>, "
+        "dễ hiểu và tập trung vào <b>tư vấn thuốc</b> cho người dùng.\n"
+        "🔹 Trình bày câu trả lời theo danh sách số thứ tự (1️⃣, 2️⃣, 3️⃣...) để dễ đọc.\n"
+        "🔹 Định dạng rõ ràng: <b>in đậm</b> những điểm quan trọng, <i>in nghiêng</i> tên thuốc, <b>in đậm</b> các lưu ý quan trọng, "
+        "sử dụng dấu gạch đầu dòng (-) để liệt kê thông tin." 
+    )
+
+    input_text = (
+        "🩺 Dựa trên thông tin y tế sau đây, hãy trả lời câu hỏi của người dùng:\n"
+        f"{context}\n"
+        "❓ <b>Câu hỏi:</b> " + question + "\n"
+        "📌 <b>Yêu cầu:</b> Hãy trả lời theo bố cục số thứ tự, dễ đọc, ngắn gọn nhưng đầy đủ, giúp người dùng dễ áp dụng.\n"
+        "📋 <b>Định dạng:</b>\n"
+        "- <b>In đậm</b> cho thông tin quan trọng\n"
+        "- <i>In nghiêng</i> cho tên thuốc\n"
+        "- <b>In đậm</b> cho các lưu ý đặc biệt\n"
+        "- Dấu gạch đầu dòng (-) để trình bày rõ ràng khi liệt kê"
+    )
+
+    prompt = prompt_template.format(
+        instruction=instruction,
+        input=input_text,
+        output=''  # AI sẽ tự điền câu trả lời
+    )
+
+    return prompt
+
+@app.route("/send_message", methods=["POST"])
+def send_message():
+    try:
+        data = request.json
+        print("📩 Received Data:", data)  # Debug log
+
+        # Kiểm tra request JSON hợp lệ
+        if not data or "message" not in data:
+            return jsonify({"error": "Dữ liệu không hợp lệ!"}), 400
+
+        message = data.get("message", "").strip()
+        conversation_id = data.get("conversation_id", "")
+
+        if not message:
+            return jsonify({"error": "Câu hỏi không hợp lệ!"}), 400
+
+        # Nếu không có conversation_id, tạo cuộc trò chuyện mới
+        if not conversation_id:
+            conversation_id = str(uuid.uuid4())
+            session["conversation_id"] = conversation_id
+            mongo.db.chat_history.insert_one({
+                "conversation_id": conversation_id,
+                "name": "Chưa có tên",
+                "messages": [],
+                "created_at": datetime.now().strftime("%d/%m/%Y %H:%M:%S")
+            })
+
+        # Gọi hàm lấy dữ liệu context
+        context_data = get_disease_and_generate_prompt(message, faiss_index, model_em, merged_df, top_k=5)
+        prompt = get_prompt(message, context_data)
+
+        # Cấu hình model Gemini
+        generation_config = {
+            "temperature": 0.2,
+            "top_p": 0.6,
+            "top_k": 40,
+            "max_output_tokens": 3000,
+            "response_mime_type": "text/plain",
+        }
+
+        # Gửi prompt đến model Gemini
+        model = genai.GenerativeModel("gemini-1.5-flash", generation_config=generation_config)
+        chat_session = model.start_chat(history=[])
+        response = chat_session.send_message(prompt)
+
+        # Lấy nội dung phản hồi từ bot
+        try:
+            text_response = response.candidates[0].content.parts[0].text
+        except Exception as e:
+            print(f"⚠️ Error getting response: {e}")
+            return jsonify({"error": "Lỗi khi lấy phản hồi từ mô hình!"}), 500
+
+        # Cấu trúc tin nhắn
+        timestamp = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
+        user_message = {"text": message, "timestamp": timestamp, "sender": "user"}
+        bot_message = {"text": text_response, "timestamp": timestamp, "sender": "bot"}
+
+        # Cập nhật cuộc trò chuyện
+        mongo.db.chat_history.update_one(
+            {"conversation_id": conversation_id},
+            {"$push": {"messages": {"$each": [user_message, bot_message]}}},
+            upsert=True
+        )
+
+        # Trả về phản hồi JSON
+        return jsonify({
+            "conversation_id": conversation_id,
+            "status": "sent",
+            "bot_reply": text_response,
+            "timestamp": timestamp
+        })
+
+    except Exception as e:
+        print(f"🚨 Error in /query: {str(e)}")
+        return jsonify({"error": f"Lỗi xử lý: {str(e)}"}), 500
+
+
+@app.route("/start_conversation", methods=["POST"])
+def start_conversation():
+    try:
+        data = request.json
+        message = data.get("message", "").strip()
+
+        if not message:
+            return jsonify({"error": "Tin nhắn không hợp lệ!"}), 400
+
+        conversation_id = str(uuid.uuid4())  # Tạo conversation_id mới
+        session["conversation_id"] = conversation_id
+
+        mongo.db.chat_history.insert_one({
+            "conversation_id": conversation_id,
+            "name": message,
+            "messages": [],
+            "created_at": datetime.now().strftime("%d/%m/%Y %H:%M:%S")
+        })
+
+        return jsonify({"conversation_id": conversation_id})
+    
+    except Exception as e:
+        print(f"🚨 Error in /start_conversation: {str(e)}")
+        return jsonify({"error": f"Lỗi xử lý: {str(e)}"}), 500
+
+
+@app.route("/all_history", methods=["GET"])
+def get_all_history():
+    """Lấy danh sách tất cả cuộc trò chuyện"""
+    chats = list(mongo.db.chat_history.find({}, {"_id": 0, "messages": 0}))
+    return jsonify(chats)
+
+
+@app.route("/conversation/<conversation_id>", methods=["GET"])
+def get_conversation(conversation_id):
+    """Lấy nội dung cuộc trò chuyện theo ID"""
+    chat = mongo.db.chat_history.find_one({"conversation_id": conversation_id}, {"_id": 0})
+    if not chat:
+        return jsonify({"error": "Cuộc trò chuyện không tồn tại"}), 404
+    return jsonify(chat)
+
+@app.route('/get_conversations', methods=['GET'])
+def get_conversations():
+    # Lấy tất cả cuộc trò chuyện từ MongoDB
+    conversations = list(mongo.db.chat_history.find({}, {"_id": 0}))
+
+    return jsonify([{
+        "conversation_id": conv.get("conversation_id", "N/A"),
+        "name": conv.get("name", "Chưa có tên"),
+        "bot_messages": [msg["text"] for msg in conv.get("messages", []) if msg.get("sender") == "bot"],  # Chỉ lấy text của bot
+        "created_at": conv.get("created_at", "Không rõ ngày")
+    } for conv in conversations])
+
+@app.route('/get_messages', methods=['GET'])
+def get_messages():
+    conversation_id = request.args.get("conversation_id")
+    
+    if not conversation_id:
+        return jsonify({"error": "Thiếu conversation_id"}), 400
+
+    # Truy vấn cuộc trò chuyện theo conversation_id
+    conversation = mongo.db.chat_history.find_one(
+        {"conversation_id": conversation_id}, 
+        {"_id": 0, "messages": 1,"name":1, "timestamp":1}
+    )
+
+    if conversation:
+        return jsonify(conversation)  # Trả về danh sách tin nhắn
+    else:
+        return jsonify({"error": "Không tìm thấy cuộc trò chuyện"}), 404
+
+
+
+@app.route("/")
+def index():
+    return render_template('index.html')
+
+@app.route('/register', methods=['GET', 'POST'])
+def register():
+    if request.method == 'POST':
+        fullname = request.form['fullname']
+        username = request.form['username']
+        password = request.form['password']
+        hash_password = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
+        
+        # Insert user into the 'user' collection
+        mongo.db.user.insert_one({'fullname': fullname,'username': username, 'password': hash_password})
+        return redirect(url_for('login'))
+    return render_template('register.html')
+
+@app.route('/login', methods=['GET', 'POST'])
+def login():
+    if request.method == 'POST':
+        username = request.form['username']
+        password = request.form['password']
+        
+        # Find user in the 'user' collection
+        user = mongo.db.user.find_one({'username': username})
+
+        # Kiểm tra nếu người dùng tồn tại và mật khẩu đúng
+        if user and bcrypt.checkpw(password.encode('utf-8'), user['password']):
+            # Lưu fullname của người dùng vào session
+            session['user'] = user.get('fullname', 'Unknown User')  # Đảm bảo lấy giá trị 'fullname' nếu có
+            return redirect(url_for('dashboard'))
+        else:
+            return 'Invalid username or password'
+    return render_template('login.html')
+    
+@app.route("/dashboard")
+def dashboard():
+    if 'user' in session:
+        current_time = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
+        username = session["user"]   
+        return render_template('dashboard.html',username=username,current_time=current_time)
+    return redirect(url_for('login'))
+
+@app.route('/logout')
+def logout():
+    session.pop('user', None)
+    return redirect(url_for('login'))
+
+
+
+if __name__ == "__main__":
+    app.run(debug=True)

config.yaml

@@ -0,0 +1,34 @@
+gemini:
+  # Required: API Keys
+  api_keys:
+    - "AIzaSyBZy7wnuLbRpngTyuplRz1FNjpP8uxttVw"
+    - "AIzaSyCvlZ63Nkt5NpjdmxYPAsG8Qskex6usCFw"
+    - "AIzaSyB4BlhbVDHupKtExi59btmX5Y5Nkm0eN7g"
+
+  # Optional: Generation Settings
+  generation:
+    temperature: 0.7
+    top_p: 1.0
+    top_k: 40
+    max_output_tokens: 8192
+    stop_sequences: []
+    response_mime_type: "text/plain"
+
+  # Optional: Rate Limiting
+  rate_limits:
+    requests_per_minute: 60
+    reset_window: 60  # seconds
+
+  # Optional: Strategies
+  strategies:
+    content: "fallback"  # round_robin, fallback, retry
+    key_rotation: "smart_cooldown"  # smart_cooldown, sequential, round_robin, least_used
+
+  # Optional: Retry Settings
+  retry:
+    max_attempts: 3
+    delay: 30  # seconds
+
+  # Optional: Model Settings
+  default_model: "gemini-2.0-flash-exp"
+  system_instruction: null  # Custom system prompt

gemini_handler.py

@@ -0,0 +1,559 @@
+from abc import ABC, abstractmethod
+import google.generativeai as genai
+import time
+import os
+import yaml
+from typing import List, Dict, Any, Optional, Tuple, Union
+from enum import Enum
+from dataclasses import dataclass
+from itertools import cycle
+from pathlib import Path
+
+@dataclass
+class GenerationConfig:
+    """Configuration for model generation parameters."""
+    temperature: float = 1.0
+    top_p: float = 1.0
+    top_k: int = 40
+    max_output_tokens: int = 8192
+    stop_sequences: Optional[List[str]] = None
+    response_mime_type: str = "text/plain"
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert config to dictionary, excluding None values."""
+        return {k: v for k, v in self.__dict__.items() if v is not None}
+
+
+@dataclass
+class ModelResponse:
+    """Represents a standardized response from any model."""
+    success: bool
+    model: str
+    text: str = ""
+    error: str = ""
+    time: float = 0.0
+    attempts: int = 1
+    api_key_index: int = 0
+
+
+class Strategy(Enum):
+    """Available content generation strategies."""
+    ROUND_ROBIN = "round_robin"
+    FALLBACK = "fallback"
+    RETRY = "retry"
+
+
+class KeyRotationStrategy(Enum):
+    """Available key rotation strategies."""
+    SEQUENTIAL = "sequential"
+    ROUND_ROBIN = "round_robin"
+    LEAST_USED = "least_used"
+    SMART_COOLDOWN = "smart_cooldown"
+
+
+@dataclass
+class KeyStats:
+    """Track usage statistics for each API key."""
+    uses: int = 0
+    last_used: float = 0
+    failures: int = 0
+    rate_limited_until: float = 0
+
+
+class ConfigLoader:
+    """Handles loading configuration from various sources."""
+    
+    @staticmethod
+    def load_api_keys(config_path: Optional[Union[str, Path]] = None) -> List[str]:
+        """
+        Load API keys from multiple sources in priority order:
+        1. YAML config file if provided
+        2. Environment variables (GEMINI_API_KEYS as comma-separated string)
+        3. Single GEMINI_API_KEY environment variable
+        """
+        # Try loading from YAML config
+        if config_path:
+            try:
+                with open(config_path, 'r') as f:
+                    config = yaml.safe_load(f)
+                    if config and 'gemini' in config and 'api_keys' in config['gemini']:
+                        keys = config['gemini']['api_keys']
+                        if isinstance(keys, list) and all(isinstance(k, str) for k in keys):
+                            return keys
+            except Exception as e:
+                print(f"Warning: Failed to load config from {config_path}: {e}")
+
+        # Try loading from GEMINI_API_KEYS environment variable
+        api_keys_str = os.getenv('GEMINI_API_KEYS')
+        if api_keys_str:
+            keys = [k.strip() for k in api_keys_str.split(',') if k.strip()]
+            if keys:
+                return keys
+
+        # Try loading single API key
+        single_key = os.getenv('GEMINI_API_KEY')
+        if single_key:
+            return [single_key]
+
+        raise ValueError(
+            "No API keys found. Please provide keys via config file, "
+            "GEMINI_API_KEYS environment variable (comma-separated), "
+            "or GEMINI_API_KEY environment variable."
+        )
+
+
+class ModelConfig:
+    """Configuration for model settings."""
+    def __init__(self):
+        self.models = [
+            "gemini-2.0-flash-exp",
+            "gemini-1.5-pro",
+            "learnlm-1.5-pro-experimental",
+            "gemini-exp-1206",
+            "gemini-exp-1121",
+            "gemini-exp-1114",
+            "gemini-2.0-flash-thinking-exp-1219",
+            "gemini-1.5-flash"
+        ]
+        self.max_retries = 3
+        self.retry_delay = 30
+        self.default_model = "gemini-2.0-flash-exp"
+
+
+class KeyRotationManager:
+    """Enhanced key rotation manager with multiple strategies."""
+    def __init__(
+        self,
+        api_keys: List[str],
+        strategy: KeyRotationStrategy = KeyRotationStrategy.ROUND_ROBIN,
+        rate_limit: int = 60,
+        reset_window: int = 60
+    ):
+        if not api_keys:
+            raise ValueError("At least one API key must be provided")
+        
+        self.api_keys = api_keys
+        self.strategy = strategy
+        self.rate_limit = rate_limit
+        self.reset_window = reset_window
+        
+        # Initialize tracking
+        self.key_stats = {i: KeyStats() for i in range(len(api_keys))}
+        self._key_cycle = cycle(range(len(api_keys)))
+        self.current_index = 0
+
+    def _is_key_available(self, key_index: int) -> bool:
+        """Check if a key is available based on rate limits and cooldown."""
+        stats = self.key_stats[key_index]
+        current_time = time.time()
+        
+        if current_time < stats.rate_limited_until:
+            return False
+            
+        if current_time - stats.last_used > self.reset_window:
+            stats.uses = 0
+            
+        return stats.uses < self.rate_limit
+
+    def _get_sequential_key(self) -> Tuple[str, int]:
+        """Get next key using sequential strategy."""
+        start_index = self.current_index
+        
+        while True:
+            if self._is_key_available(self.current_index):
+                key_index = self.current_index
+                self.current_index = (self.current_index + 1) % len(self.api_keys)
+                return self.api_keys[key_index], key_index
+                
+            self.current_index = (self.current_index + 1) % len(self.api_keys)
+            if self.current_index == start_index:
+                self._handle_all_keys_busy()
+
+    def _get_round_robin_key(self) -> Tuple[str, int]:
+        """Get next key using round-robin strategy."""
+        start_index = next(self._key_cycle)
+        current_index = start_index
+        
+        while True:
+            if self._is_key_available(current_index):
+                return self.api_keys[current_index], current_index
+                
+            current_index = next(self._key_cycle)
+            if current_index == start_index:
+                self._handle_all_keys_busy()
+
+    def _get_least_used_key(self) -> Tuple[str, int]:
+        """Get key with lowest usage count."""
+        while True:
+            available_keys = [
+                (idx, stats) for idx, stats in self.key_stats.items()
+                if self._is_key_available(idx)
+            ]
+            
+            if available_keys:
+                key_index, _ = min(available_keys, key=lambda x: x[1].uses)
+                return self.api_keys[key_index], key_index
+                
+            self._handle_all_keys_busy()
+
+    def _get_smart_cooldown_key(self) -> Tuple[str, int]:
+        """Get key using smart cooldown strategy."""
+        while True:
+            current_time = time.time()
+            available_keys = [
+                (idx, stats) for idx, stats in self.key_stats.items()
+                if current_time >= stats.rate_limited_until and self._is_key_available(idx)
+            ]
+            
+            if available_keys:
+                key_index, _ = min(
+                    available_keys,
+                    key=lambda x: (x[1].failures, -(current_time - x[1].last_used))
+                )
+                return self.api_keys[key_index], key_index
+                
+            self._handle_all_keys_busy()
+
+    def _handle_all_keys_busy(self) -> None:
+        """Handle situation when all keys are busy."""
+        current_time = time.time()
+        any_reset = False
+        
+        for idx, stats in self.key_stats.items():
+            if current_time - stats.last_used > self.reset_window:
+                stats.uses = 0
+                any_reset = True
+                
+        if not any_reset:
+            time.sleep(1)
+
+    def get_next_key(self) -> Tuple[str, int]:
+        """Get next available API key based on selected strategy."""
+        strategy_methods = {
+            KeyRotationStrategy.SEQUENTIAL: self._get_sequential_key,
+            KeyRotationStrategy.ROUND_ROBIN: self._get_round_robin_key,
+            KeyRotationStrategy.LEAST_USED: self._get_least_used_key,
+            KeyRotationStrategy.SMART_COOLDOWN: self._get_smart_cooldown_key
+        }
+        
+        method = strategy_methods.get(self.strategy)
+        if not method:
+            raise ValueError(f"Unknown strategy: {self.strategy}")
+            
+        api_key, key_index = method()
+        
+        stats = self.key_stats[key_index]
+        stats.uses += 1
+        stats.last_used = time.time()
+        
+        return api_key, key_index
+
+    def mark_success(self, key_index: int) -> None:
+        """Mark successful API call."""
+        if 0 <= key_index < len(self.api_keys):
+            self.key_stats[key_index].failures = 0
+
+    def mark_rate_limited(self, key_index: int) -> None:
+        """Mark API key as rate limited."""
+        if 0 <= key_index < len(self.api_keys):
+            stats = self.key_stats[key_index]
+            stats.failures += 1
+            stats.rate_limited_until = time.time() + self.reset_window
+            stats.uses = self.rate_limit
+
+
+class ResponseHandler:
+    """Handles and processes model responses."""
+    @staticmethod
+    def process_response(
+        response: Any,
+        model_name: str,
+        start_time: float,
+        key_index: int
+    ) -> ModelResponse:
+        """Process and validate model response."""
+        try:
+            if hasattr(response, 'candidates') and response.candidates:
+                finish_reason = response.candidates[0].finish_reason
+                if finish_reason == 4:  # Copyright material
+                    return ModelResponse(
+                        success=False,
+                        model=model_name,
+                        error='Copyright material detected in response',
+                        time=time.time() - start_time,
+                        api_key_index=key_index
+                    )
+            
+            return ModelResponse(
+                success=True,
+                model=model_name,
+                text=response.text,
+                time=time.time() - start_time,
+                api_key_index=key_index
+            )
+        except Exception as e:
+            if "The `response.text` quick accessor requires the response to contain a valid `Part`" in str(e):
+                return ModelResponse(
+                    success=False,
+                    model=model_name,
+                    error='No valid response parts available',
+                    time=time.time() - start_time,
+                    api_key_index=key_index
+                )
+            raise
+
+
+class ContentStrategy(ABC):
+    """Abstract base class for content generation strategies."""
+    def __init__(
+        self,
+        config: ModelConfig,
+        key_manager: KeyRotationManager,
+        system_instruction: Optional[str] = None,
+        generation_config: Optional[GenerationConfig] = None
+    ):
+        self.config = config
+        self.key_manager = key_manager
+        self.system_instruction = system_instruction
+        self.generation_config = generation_config or GenerationConfig()
+
+    @abstractmethod
+    def generate(self, prompt: str, model_name: str) -> ModelResponse:
+        """Generate content using the specific strategy."""
+        pass
+
+    def _try_generate(self, model_name: str, prompt: str, start_time: float) -> ModelResponse:
+        """Helper method for generating content with key rotation."""
+        api_key, key_index = self.key_manager.get_next_key()
+        try:
+            genai.configure(api_key=api_key)
+            model = genai.GenerativeModel(
+                model_name=model_name,
+                generation_config=self.generation_config.to_dict(),
+                system_instruction=self.system_instruction
+            )
+            response = model.generate_content(prompt)
+            
+            result = ResponseHandler.process_response(response, model_name, start_time, key_index)
+            if result.success:
+                self.key_manager.mark_success(key_index)
+            return result
+            
+        except Exception as e:
+            if "429" in str(e):
+                self.key_manager.mark_rate_limited(key_index)
+            return ModelResponse(
+                success=False,
+                model=model_name,
+                error=str(e),
+                time=time.time() - start_time,
+                api_key_index=key_index
+            )
+
+
+class RoundRobinStrategy(ContentStrategy):
+    """Round robin implementation of content generation."""
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self._current_index = 0
+
+    def _get_next_model(self) -> str:
+        """Get next model in round-robin fashion."""
+        model = self.config.models[self._current_index]
+        self._current_index = (self._current_index + 1) % len(self.config.models)
+        return model
+
+    def generate(self, prompt: str, _: str) -> ModelResponse:
+        start_time = time.time()
+        
+        for _ in range(len(self.config.models)):
+            model_name = self._get_next_model()
+            result = self._try_generate(model_name, prompt, start_time)
+            if result.success or 'Copyright' in result.error:
+                return result
+
+        return ModelResponse(
+            success=False,
+            model='all_models_failed',
+            error='All models failed (rate limited or copyright issues)',
+            time=time.time() - start_time
+        )
+
+
+class FallbackStrategy(ContentStrategy):
+    """Fallback implementation of content generation."""
+    def generate(self, prompt: str, start_model: str) -> ModelResponse:
+        start_time = time.time()
+        
+        try:
+            start_index = self.config.models.index(start_model)
+        except ValueError:
+            return ModelResponse(
+                success=False,
+                model=start_model,
+                error=f"Model {start_model} not found in available models",
+                time=time.time() - start_time
+            )
+
+        for model_name in self.config.models[start_index:]:
+            result = self._try_generate(model_name, prompt, start_time)
+            if result.success or 'Copyright' in result.error:
+                return result
+
+        return ModelResponse(
+            success=False,
+            model='all_models_failed',
+            error='All models failed (rate limited or copyright issues)',
+            time=time.time() - start_time
+        )
+
+
+class RetryStrategy(ContentStrategy):
+    """Retry implementation of content generation."""
+    def generate(self, prompt: str, model_name: str) -> ModelResponse:
+        start_time = time.time()
+        
+        for attempt in range(self.config.max_retries):
+            result = self._try_generate(model_name, prompt, start_time)
+            result.attempts = attempt + 1
+            
+            if result.success or 'Copyright' in result.error:
+                return result
+                
+            if attempt < self.config.max_retries - 1:
+                print(f"Error encountered. Waiting {self.config.retry_delay}s... "
+                      f"(Attempt {attempt + 1}/{self.config.max_retries})")
+                time.sleep(self.config.retry_delay)
+        
+        return ModelResponse(
+            success=False,
+            model=model_name,
+            error='Max retries exceeded',
+            time=time.time() - start_time,
+            attempts=self.config.max_retries
+        )
+
+
+class GeminiHandler:
+    """Main handler class for Gemini API interactions."""
+    def __init__(
+        self,
+        api_keys: Optional[List[str]] = None,
+        config_path: Optional[Union[str, Path]] = None,
+        content_strategy: Strategy = Strategy.ROUND_ROBIN,
+        key_strategy: KeyRotationStrategy = KeyRotationStrategy.ROUND_ROBIN,
+        system_instruction: Optional[str] = None,
+        generation_config: Optional[GenerationConfig] = None
+    ):
+        """
+        Initialize GeminiHandler with flexible configuration options.
+        
+        Args:
+            api_keys: Optional list of API keys
+            config_path: Optional path to YAML config file
+            content_strategy: Strategy for content generation
+            key_strategy: Strategy for key rotation
+            system_instruction: Optional system instruction
+            generation_config: Optional generation configuration
+        """
+        # Load API keys from provided list or config sources
+        self.api_keys = api_keys or ConfigLoader.load_api_keys(config_path)
+        
+        self.config = ModelConfig()
+        self.key_manager = KeyRotationManager(
+            api_keys=self.api_keys,
+            strategy=key_strategy,
+            rate_limit=60,
+            reset_window=60
+        )
+        self.system_instruction = system_instruction
+        self.generation_config = generation_config
+        self._strategy = self._create_strategy(content_strategy)
+
+    def _create_strategy(self, strategy: Strategy) -> ContentStrategy:
+        """Factory method to create appropriate strategy."""
+        strategies = {
+            Strategy.ROUND_ROBIN: RoundRobinStrategy,
+            Strategy.FALLBACK: FallbackStrategy,
+            Strategy.RETRY: RetryStrategy
+        }
+        
+        strategy_class = strategies.get(strategy)
+        if not strategy_class:
+            raise ValueError(f"Unknown strategy: {strategy}")
+            
+        return strategy_class(
+            config=self.config,
+            key_manager=self.key_manager,
+            system_instruction=self.system_instruction,
+            generation_config=self.generation_config
+        )
+
+    def generate_content(
+        self,
+        prompt: str,
+        model_name: Optional[str] = None,
+        return_stats: bool = False
+    ) -> Dict[str, Any]:
+        """
+        Generate content using the selected strategies.
+        
+        Args:
+            prompt: The input prompt for content generation
+            model_name: Optional specific model to use (default: None)
+            return_stats: Whether to include key usage statistics (default: False)
+            
+        Returns:
+            Dictionary containing generation results and optionally key statistics
+        """
+        if not model_name:
+            model_name = self.config.default_model
+            
+        response = self._strategy.generate(prompt, model_name)
+        result = response.__dict__
+        
+        if return_stats:
+            result["key_stats"] = {
+                idx: {
+                    "uses": stats.uses,
+                    "last_used": stats.last_used,
+                    "failures": stats.failures,
+                    "rate_limited_until": stats.rate_limited_until
+                }
+                for idx, stats in self.key_manager.key_stats.items()
+            }
+            
+        return result
+
+    def get_key_stats(self, key_index: Optional[int] = None) -> Dict[int, Dict[str, Any]]:
+        """
+        Get current key usage statistics.
+        
+        Args:
+            key_index: Optional specific key index to get stats for
+            
+        Returns:
+            Dictionary of key statistics
+        """
+        if key_index is not None:
+            if 0 <= key_index < len(self.key_manager.api_keys):
+                stats = self.key_manager.key_stats[key_index]
+                return {
+                    key_index: {
+                        "uses": stats.uses,
+                        "last_used": stats.last_used,
+                        "failures": stats.failures,
+                        "rate_limited_until": stats.rate_limited_until
+                    }
+                }
+            raise ValueError(f"Invalid key index: {key_index}")
+        
+        return {
+            idx: {
+                "uses": stats.uses,
+                "last_used": stats.last_used,
+                "failures": stats.failures,
+                "rate_limited_until": stats.rate_limited_until
+            }
+            for idx, stats in self.key_manager.key_stats.items()
+        }

requirements.txt

@@ -0,0 +1,12 @@
+Flask
+flask_pymongo
+flask_cors
+bcrypt
+sentence-transformers
+faiss-cpu
+numpy
+pandas
+google-generativeai
+python-dotenv
+pymongo
+gunicorn

source/faiss_index_drug.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ea0955d2743718823cbf92570fc5ba7004bc0d6ba9457ed88df8df65becc2c66
+size 568365

source/faiss_index_vn.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:40ea00c1b969a484fa7c4f0eae61d569f8f663c1140b5fd6bbb950eb08184d31
+size 1136685

source/sentence_embeddings.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a561406ed04d8a7227231108dfd66ba8a7cf5a9da4a5a9d6e598de7438ab67c4
+size 568483

source/sentence_embeddings_vn.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:38fad112ffbfd8eb359e55b5e308ffb9feb5e19ee8391779db8619ab6a3a87e4
+size 1136803

static/style.css

@@ -0,0 +1,267 @@
+:root {
+    --primary: #202123;
+    --overhover: hsla(240, 9%, 59%, 0.1);
+  }
+  
+  * {
+    margin: 0;
+    padding: 0;
+    font-family: courier;
+  }
+  
+  nav {
+    height: 50px;
+    background: var(--primary);
+    color: white;
+    display: none;
+    justify-content: space-between;
+    align-items: center;
+    position: fixed;
+    width: 100%;
+  }
+  
+  nav div {
+    margin: 0 20px;
+    display: inline-block;
+    overflow: hidden;
+    white-space: nowrap;
+    text-overflow: ellipsis;
+  }
+  
+  .ham-menu {
+    background: var(--primary);
+    color: white;
+    border: none;
+    cursor: pointer;
+  }
+  
+  .ham-menu span {
+    display: block;
+    width: 25px;
+    height: 3px;
+    margin-bottom: 5px;
+    position: relative;
+    background: black;
+    border-radius: 3px;
+    z-index: 1;
+    transform-origin: 4px 0px;
+    transition: transform 0.5s cubic-bezier(0.77, 0.2, 0.05, 1),
+      background 0.5s cubic-bezier(0.77, 0.2, 0.05, 1), opacity 0.55s ease;
+  }
+  
+  .ham-menu:hover span {
+    background: white;
+  }
+  
+  #main {
+    display: flex;
+  }
+  
+  /*Side Nav*/
+  #sidenav {
+    height: 100vh;
+    width: 20rem;
+    background-color: var(--primary);
+    font-size: 1rem;
+    color: white;
+    position: fixed;
+    left: 0;
+    top: 0;
+    margin: 0;
+    z-index: 11;
+  }
+  
+  #sidenav .sidenav-content {
+    padding: 0.5rem;
+  }
+  
+  #close {
+    position: absolute;
+    right: -40px;
+    background: var(--primary);
+    color: white;
+    border: 1px solid grey;
+    border-radius: 5px;
+    cursor: pointer;
+    display: none;
+    width: 40px;
+    height: 40px;
+  }
+  #close span {
+    width: 40px;
+    height: 40px;
+    position: relative;
+  }
+  #close span:before,
+  #close span:after {
+    position: absolute;
+    content: " ";
+    top: -16px;
+    left: -1px;
+    height: 33px;
+    width: 3px;
+    background-color: black;
+  }
+  #close span:before {
+    transform: rotate(45deg);
+  }
+  #close span:after {
+    transform: rotate(-45deg);
+  }
+  
+  #close:hover span:after,
+  #close:hover span:before {
+    background: white;
+  }
+  
+  #new-chat-btn {
+    background: var(--primary);
+    color: white;
+    border: 1px solid grey;
+    border-radius: 5px;
+    width: 100%;
+    cursor: pointer;
+    text-align: start;
+    padding: 0.7rem;
+    font-size: inherit;
+  }
+  
+  #new-chat-btn:hover {
+    background: var(--overhover);
+  }
+  
+  .new-chat {
+    cursor: pointer;
+  }
+  
+  .saved-chats {
+    margin: 0.5rem 0;
+  }
+  
+  .saved-chats p {
+    padding: 0.8rem;
+    margin: 3px 0;
+    border-radius: 5px;
+    cursor: pointer;
+  }
+  
+  .saved-chats p:hover {
+    background: var(--overhover);
+  }
+  
+  .saved-chats .selected {
+    padding: 0.8rem;
+    background: var(--overhover);
+    margin: 3px 0;
+    border-radius: 5px;
+  }
+  
+  .config {
+    margin: 0.5rem 0;
+    position: absolute;
+    bottom: 0;
+    width: inherit;
+    background: var(--primary);
+  }
+  
+  .config p {
+    padding: 0.7rem;
+    cursor: pointer;
+    margin: 5px 0;
+    border-radius: 5px;
+  }
+  
+  .config p:hover {
+    background: var(--overhover);
+  }
+  
+  .config hr {
+    width: 19rem;
+  }
+  
+  /*Main Content Body*/
+  #content-body {
+    height: 100vh;
+    background-color: white;
+    text-align: center;
+    margin-left: 20rem;
+    flex: 1;
+  }
+  
+  #messages {
+    padding-bottom: 100px;
+  }
+  
+  .message-div {
+    display: flex;
+    margin: 10px auto;
+    max-width: 800px;
+    justify-content: start;
+  }
+  
+  .user-message {
+    background: ;
+    padding: 10px;
+  }
+  
+  .gpt-message {
+    background: rgba(247, 247, 248);
+    padding: 10px;
+  }
+  
+  .message-profile-pic {
+    margin-right: 20px;
+  }
+  
+  .message-content {
+    text-align: start;
+    margin-top: 5px;
+  }
+  
+  .message-content p {
+    margin-bottom: 20px;
+  }
+  
+  #chat-section {
+    width: -webkit-fill-available;
+    height: 130px;
+    background-image: linear-gradient(to bottom, transparent 10%, white 90%);
+    position: fixed;
+    bottom: 0;
+  }
+  
+  #chat-section div {
+    max-width: 800px;
+    margin: 2rem auto;
+    padding: 0 20px;
+  }
+  
+  #chat-section input {
+    padding: 0.9rem;
+    border-radius: 5px;
+    border: 0.1px solid grey;
+    width: 90%;
+    font-size: 20px;
+    box-shadow: 0 0 7px 0px grey;
+  }
+  
+  #chat-section input:focus {
+    outline: none;
+  }
+  
+  @media screen and (max-width: 800px) {
+    nav {
+      display: flex;
+    }
+    #close {
+      display: block;
+    }
+    #sidenav {
+      position: fixed;
+      left: -400px;
+    }
+    #content-body {
+      margin: 50px 0 0 0;
+    }
+  }
+  

templates/index.html

@@ -0,0 +1,356 @@
+
+<!DOCTYPE html>
+<html>
+
+<head>
+  <!-- Basic -->
+  <meta charset="utf-8" />
+  <meta http-equiv="X-UA-Compatible" content="IE=edge" />
+  <!-- Mobile Metas -->
+  <meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no" />
+  <!-- Site Metas -->
+  <meta name="keywords" content="" />
+  <meta name="description" content="" />
+  <meta name="author" content="" />
+  <base href="https://themewagon.github.io/orthoc/">
+  <link rel="shortcut icon" href="images/favicon.png" type="">
+
+  <title> Orthoc </title>
+
+  <!-- bootstrap core css -->
+  <link rel="stylesheet" type="text/css" href="css/bootstrap.css" />
+
+  <!-- fonts style -->
+  <link href="https://fonts.googleapis.com/css2?family=Roboto:wght@400;500;700;900&display=swap" rel="stylesheet">
+
+  <!--owl slider stylesheet -->
+  <link rel="stylesheet" type="text/css" href="https://cdnjs.cloudflare.com/ajax/libs/OwlCarousel2/2.3.4/assets/owl.carousel.min.css" />
+
+  <!-- font awesome style -->
+  <link href="css/font-awesome.min.css" rel="stylesheet" />
+
+  <!-- Custom styles for this template -->
+  <link href="css/style.css" rel="stylesheet" />
+  <!-- responsive style -->
+  <link href="css/responsive.css" rel="stylesheet" />
+  <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.7.2/css/all.min.css" integrity="sha512-Evv84Mr4kqVGRNSgIGL/F/aIDqQb7xQ2vcrdIwxfjThSH8CSR7PBEakCr51Ck+w+/U6swU2Im1vVX0SVk9ABhg==" crossorigin="anonymous" referrerpolicy="no-referrer" />
+
+</head>
+
+<body>
+
+  <div class="hero_area">
+
+    <div class="hero_bg_box">
+      <img src="images/hero-bg.png" alt="">
+    </div>
+
+    <!-- header section strats -->
+    <header class="header_section">
+      <div class="container">
+        <nav class="navbar navbar-expand-lg custom_nav-container ">
+          <a class="navbar-brand" href="index.html">
+            <span>
+              Orthoc
+            </span>
+          </a>
+
+          <button class="navbar-toggler" type="button" data-toggle="collapse" data-target="#navbarSupportedContent" aria-controls="navbarSupportedContent" aria-expanded="false" aria-label="Toggle navigation">
+            <span class=""> </span>
+          </button>
+
+          <div class="collapse navbar-collapse" id="navbarSupportedContent">
+            <ul class="navbar-nav">
+              <li class="nav-item active">
+                <a class="nav-link" href="http://127.0.0.1:5000/">Trang chủ <span class="sr-only">(current)</span></a>
+              </li>
+              <li class="nav-item">
+                <a class="nav-link" href="http://127.0.0.1:5000/register"> Đăng ký </a>
+              </li>
+              <li class="nav-item">
+                <a class="nav-link" href="http://127.0.0.1:5000/login">Đăng nhập</a>
+              </li>
+              <form class="form-inline">
+                <button class="btn  my-2 my-sm-0 nav_search-btn" type="submit">
+                  <i class="fa fa-search" aria-hidden="true"></i>
+                </button>
+              </form>
+            </ul>
+          </div>
+        </nav>
+      </div>
+    </header>
+    <!-- end header section -->
+    <!-- slider section -->
+    <section class="slider_section ">
+      <div id="customCarousel1" class="carousel slide" data-ride="carousel">
+        <div class="carousel-inner">
+          <div class="carousel-item active">
+            <div class="container ">
+              <div class="row">
+                <div class="col-md-7">
+                  <div class="detail-box">
+                    <h1>
+                     Hệ Thống Gợi Ý Thuốc Dựa Trên Triệu Chứng
+                    </h1>
+                    <p>
+                      Hệ thống được phát triển hướng tới việc cải thiện khả năng cá nhân hóa trong kê đơn thuốc, vốn là một thách thức lớn đối với các phương pháp truyền thống dựa trên triệu chứng bề mặt. Thay vì chỉ dừng lại ở việc phân tích các triệu chứng đơn lẻ, hệ thống sẽ tích hợp thông tin từ các nguồn dữ liệu đa chiều để đưa ra những khuyến nghị thuốc tối ưu, phù hợp với từng trường hợp cụ thể.
+                    </p>
+                    <div class="btn-box">
+                      <a href="http://127.0.0.1:5000/login" class="btn1">
+                       Đăng nhập
+                      </a>
+                    </div>
+                  </div>
+                </div>
+              </div>
+            </div>
+          </div>
+          <!-- <div class="carousel-item ">
+            <div class="container ">
+              <div class="row">
+                <div class="col-md-7">
+                  <div class="detail-box">
+                    <h1>
+                      We Provide Best Healthcare
+                    </h1>
+                    <p>
+                      Explicabo esse amet tempora quibusdam laudantium, laborum eaque magnam fugiat hic? Esse dicta aliquid error repudiandae earum suscipit fugiat molestias, veniam, vel architecto veritatis delectus repellat modi impedit sequi.
+                    </p>
+                    <div class="btn-box">
+                      <a href="" class="btn1">
+                        Read More
+                      </a>
+                    </div>
+                  </div>
+                </div>
+              </div>
+            </div>
+          </div>
+          <div class="carousel-item">
+            <div class="container ">
+              <div class="row">
+                <div class="col-md-7">
+                  <div class="detail-box">
+                    <h1>
+                      We Provide Best Healthcare
+                    </h1>
+                    <p>
+                      Explicabo esse amet tempora quibusdam laudantium, laborum eaque magnam fugiat hic? Esse dicta aliquid error repudiandae earum suscipit fugiat molestias, veniam, vel architecto veritatis delectus repellat modi impedit sequi.
+                    </p>
+                    <div class="btn-box">
+                      <a href="" class="btn1">
+                        Read More
+                      </a>
+                    </div>
+                  </div>
+                </div>
+              </div>
+            </div>
+          </div> -->
+        </div>
+        <!-- <ol class="carousel-indicators">
+          <li data-target="#customCarousel1" data-slide-to="0" class="active"></li>
+          <li data-target="#customCarousel1" data-slide-to="1"></li>
+          <li data-target="#customCarousel1" data-slide-to="2"></li>
+        </ol> -->
+      </div>
+
+    </section>
+    <!-- end slider section -->
+  </div>
+
+
+  <!-- department section -->
+
+  <section class="department_section layout_padding">
+    <div class="department_container">
+      <div class="container ">
+        <div class="heading_container heading_center">
+          <h2>
+            CHUYÊN ĐỀ NGHIÊN CỨU 2
+          </h2>
+          <p>
+            Mô hình gợi ý thuốc dựa trên triệu chứng
+          </p>
+        </div>
+        <div class="row">
+          <div class="col-md-3">
+            <div class="box ">
+              <div class="img-box">
+                <img src="images/s1.png" alt="">
+              </div>
+              <div class="detail-box">
+                <h5>
+                  Mô tả đề tài
+                </h5>
+                <p>
+                  Mô hình gợi ý thuốc dựa trên triệu chứng giúp hỗ trợ chẩn đoán và tối ưu hóa điều trị bằng trí tuệ nhân tạo.
+                </p>
+              </div>
+            </div>
+          </div>
+          <div class="col-md-3">
+            <div class="box ">
+              <div class="img-box">
+                <img src="images/s2.png" alt="">
+              </div>
+              <div class="detail-box">
+                <h5>
+                  Mục tiêu thực hiện
+                </h5>
+                <p>
+                 Tạo ra mô hình gợi ý thuốc dựa trên triệu chứng bằng trí tuệ nhân tạo, nhằm hỗ trợ bác sĩ và bệnh nhân.
+                </p>
+              </div>
+            </div>
+          </div>
+          <div class="col-md-3">
+            <div class="box ">
+              <div class="img-box">
+                <img src="images/s3.png" alt="">
+              </div>
+              <div class="detail-box">
+                <h5>
+                  Đối tượng nghiên cứu
+                </h5>
+                <p>
+                  Mô hình gợi ý thuốc dựa trên triệu chứng, sử dụng trí tuệ nhân tạo trong hỗ trợ chẩn đoán và điều trị.
+                </p>
+              </div>
+            </div>
+          </div>
+          <div class="col-md-3">
+            <div class="box ">
+              <div class="img-box">
+                <img src="images/s4.png" alt="">
+              </div>
+              <div class="detail-box">
+                <h5>
+                 Phạm vi nghiên cứu
+                </h5>
+                <p>
+                  Tập trung vào việc thu thập và xử lý dữ liệu triệu chứng - thuốc từ nguồn y khoa đáng tin.
+                </p>
+              </div>
+            </div>
+          </div>
+        </div>
+        <!-- <div class="btn-box">
+          <a href="">
+            View All
+          </a>
+        </div> -->
+      </div>
+    </div>
+  </section>
+
+  <!-- end department section -->
+
+  <!-- footer section -->
+  <footer class="footer_section">
+    <div class="container">
+      <div class="row">
+        <div class="col-md-6 col-lg-4 footer_col">
+          <div class="footer_contact">
+            <h4>
+              Thông tin chuyên đề
+            </h4>
+            <div class="contact_link_box">
+              <a href="">
+                <i class="fa fa-user" aria-hidden="true"></i>
+                <span>
+                  TS Trần Thanh Phước
+                </span>
+              </a>
+              <a href="">
+                <i class="fa fa-users" aria-hidden="true"></i>
+                <span>
+                 Nguyễn Trần Quỳnh Như - 241805005
+                </span>
+              </a>
+              <a href="">
+                <i class="fa fa-users" aria-hidden="true"></i>
+                <span>
+                  Trần Thành Trung - 241805005
+                </span>
+              </a>
+              <a href="">
+                <i class="fa-solid fa-school" aria-hidden="true"></i>
+                <span>
+                 Lớp:  24180501
+                </span>
+              </a>
+              <a href="">
+                <i class="fa-solid fa-calendar-days" aria-hidden="true"></i>
+                <span>
+                 Khóa: 2024-2026
+                </span>
+              </a>
+            </div>
+          </div>
+         
+        </div>
+        <div class="col-md-6 col-lg-4 footer_col">
+          <div class="footer_detail">
+            <h4>
+              Giới thiệu
+            </h4>
+            <p>
+              Hệ thống hướng tới cải thiện khả năng cá nhân hóa trong kê đơn thuốc bằng cách tích hợp dữ liệu đa chiều, thay vì chỉ dựa trên triệu chứng bề mặt, nhằm đưa ra khuyến nghị tối ưu cho từng trường hợp.
+            </p>
+          </div>
+        </div>
+        <div class="col-md-6 col-lg-3 mx-auto footer_col">
+          <div class="footer_link_box">
+            <h4>
+              Links
+            </h4>
+            <div class="footer_links">
+              <a class="active" href="http://127.0.0.1:5000/">
+                Trang chủ
+              </a>
+              <a class="" href="http://127.0.0.1:5000/register">
+                Đăng ký
+              </a>
+              <a class="" href="http://127.0.0.1:5000/login">
+                Đăng nhập
+              </a>
+            </div>
+          </div>
+        </div>
+       
+      </div>
+      <div class="footer-info">
+        <p>
+          &copy; <span id="displayYear"></span> All Rights Reserved By
+          <a href="">QuynhNhu<br><br></a>
+            &copy; <span id="displayYear"></span> Distributed By
+            <a href="">QuynhNhu</a>
+        </p>
+       
+      </div>  
+    </div>
+  </footer>
+  <!-- footer section -->
+
+  <!-- jQery -->
+  <script type="text/javascript" src="js/jquery-3.4.1.min.js"></script>
+  <!-- popper js -->
+  <script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/umd/popper.min.js" integrity="sha384-Q6E9RHvbIyZFJoft+2mJbHaEWldlvI9IOYy5n3zV9zzTtmI3UksdQRVvoxMfooAo" crossorigin="anonymous">
+  </script>
+  <!-- bootstrap js -->
+  <script type="text/javascript" src="js/bootstrap.js"></script>
+  <!-- owl slider -->
+  <script type="text/javascript" src="https://cdnjs.cloudflare.com/ajax/libs/OwlCarousel2/2.3.4/owl.carousel.min.js">
+  </script>
+  <!-- custom js -->
+  <script type="text/javascript" src="js/custom.js"></script>
+  <!-- Google Map -->
+  <script src="https://maps.googleapis.com/maps/api/js?key=AIzaSyCh39n5U-4IoWpsVGUHWdqB6puEkhRLdmI&callback=myMap">
+  </script>
+  <!-- End Google Map -->
+
+</body>
+
+</html>

templates/login.html

@@ -0,0 +1,101 @@
+
+<!doctype html>
+<html lang="en">
+
+<head>
+
+    <meta charset="utf-8" />
+    <title>Log in | Chaton - Responsive Bootstrap 5 Chat App</title>
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <meta content="Chaton - Responsive Chat App Template in HTML. A fully featured HTML chat messenger template in Bootstrap 5" name="description" />
+    <meta name="keywords" content="Chaton chat template, chat, web chat template, chat status, chat template, communication, discussion, group chat, message, messenger template, status" />
+    <meta content="Themesdesign" name="author" />
+    <!-- App favicon -->
+    <link rel="shortcut icon" href="https://themesdesign.in/chaton/layouts/assets/images/favicon.ico" id="tabIcon">
+
+    <!-- Bootstrap Css -->
+    <link href="https://themesdesign.in/chaton/layouts/assets/css/bootstrap.min.css" id="bootstrap-style" rel="stylesheet" type="text/css" />
+    <!-- Icons Css -->
+    <link href="https://themesdesign.in/chaton/layouts/assets/css/icons.min.css" rel="stylesheet" type="text/css" />
+    <!-- App Css-->
+    <link href="https://themesdesign.in/chaton/layouts/assets/css/app.min.css" id="app-style" rel="stylesheet" type="text/css" />
+
+</head>
+
+<body>
+<div class="auth-bg">
+    <div class="container p-0">
+        <div class="row justify-content-center g-0">
+            <div class="col-xl-9 col-lg-8">
+                <div class="authentication-page-content shadow-lg">
+                    <div class="d-flex flex-column h-100 px-4 pt-4">
+                        <div class="row justify-content-center">
+                            <div class="col-sm-8 col-lg-6 col-xl-6">
+
+                                <div class="py-md-5 py-4">
+
+                                    <div class="text-center mb-5">
+                                        <h3>Xin chào!</h3>
+                                        <p class="text-muted">Chào mừng đến với trang đăng nhập</p>
+                                    </div>
+
+                                    <form action="http://127.0.0.1:5000/login" method="POST">
+                                        <div class="mb-3">
+                                            <label for="username" class="form-label">Tên đăng nhập</label>
+                                            <input type="text" name="username" class="form-control" id="username" placeholder="Enter username">
+                                        </div>
+
+                                        <div class="mb-3">
+                                            <label for="userpassword" class="form-label">Mật khẩu</label>
+                                            <div class="position-relative auth-pass-inputgroup mb-3">
+                                                <input type="password" name="password" class="form-control pe-5" placeholder="Enter Password" id="password-input">
+                                                <button class="btn btn-link position-absolute end-0 top-0 text-decoration-none text-muted" type="button" id="password-addon"><i class="ri-eye-fill align-middle"></i></button>
+                                            </div>
+                                        </div>
+
+
+                                        <div class="text-center mt-4">
+                                            <button class="btn btn-primary w-100" type="submit">Đăng nhập</button>
+                                        </div>
+                                       
+                                    </form><!-- end form -->
+                                </div>
+                            </div><!-- end col -->
+                        </div><!-- end row -->
+
+                        <div class="row">
+                            <div class="col-xl-12">
+                                <div class="text-center text-muted p-4">
+                                    <p class="mb-0">&copy;
+                                        <script>document.write(new Date().getFullYear())</script> Chaton. Crafted with <i class="mdi mdi-heart text-danger"></i> by Themesdesign
+                                    </p>
+                                </div>
+                            </div><!-- end col -->
+                        </div><!-- end row -->
+
+                    </div>
+                </div>
+            </div>
+            <!-- end col -->
+        </div>
+        <!-- end row -->
+    </div>
+    <!-- end container-fluid -->
+</div>
+<!-- end auth bg -->
+
+<!-- JAVASCRIPT -->
+<script src="https://themesdesign.in/chaton/layouts/assets/libs/bootstrap/js/bootstrap.bundle.min.js"></script>
+<script src="https://themesdesign.in/chaton/layouts/assets/libs/simplebar/simplebar.min.js"></script>
+<script src="https://themesdesign.in/chaton/layouts/assets/libs/node-waves/waves.min.js"></script>
+<!-- password addon -->
+<script src="https://themesdesign.in/chaton/layouts/assets/js/pages/password-addon.init.js"></script>
+
+<!-- theme-style init -->
+<script src="https://themesdesign.in/chaton/layouts/assets/js/pages/theme-style.init.js"></script>
+
+<script src="https://themesdesign.in/chaton/layouts/assets/js/app.js"></script>
+
+</body>
+
+</html>

templates/register.html

@@ -0,0 +1,106 @@
+
+<!doctype html>
+<html lang="en">
+
+<head>
+
+    <meta charset="utf-8" />
+    <title>Log in | Chaton - Responsive Bootstrap 5 Chat App</title>
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <meta content="Chaton - Responsive Chat App Template in HTML. A fully featured HTML chat messenger template in Bootstrap 5" name="description" />
+    <meta name="keywords" content="Chaton chat template, chat, web chat template, chat status, chat template, communication, discussion, group chat, message, messenger template, status" />
+    <meta content="Themesdesign" name="author" />
+    <!-- App favicon -->
+    <link rel="shortcut icon" href="https://themesdesign.in/chaton/layouts/assets/images/favicon.ico" id="tabIcon">
+
+    <!-- Bootstrap Css -->
+    <link href="https://themesdesign.in/chaton/layouts/assets/css/bootstrap.min.css" id="bootstrap-style" rel="stylesheet" type="text/css" />
+    <!-- Icons Css -->
+    <link href="https://themesdesign.in/chaton/layouts/assets/css/icons.min.css" rel="stylesheet" type="text/css" />
+    <!-- App Css-->
+    <link href="https://themesdesign.in/chaton/layouts/assets/css/app.min.css" id="app-style" rel="stylesheet" type="text/css" />
+
+</head>
+
+<body>
+<div class="auth-bg">
+    <div class="container p-0">
+        <div class="row justify-content-center g-0">
+            <div class="col-xl-9 col-lg-8">
+                <div class="authentication-page-content shadow-lg">
+                    <div class="d-flex flex-column h-100 px-4 pt-4">
+                        <div class="row justify-content-center">
+                            <div class="col-sm-8 col-lg-6 col-xl-6">
+
+                                <div class="py-md-5 py-4">
+
+                                    <div class="text-center mb-5">
+                                        <h3>Xin chào!</h3>
+                                        <p class="text-muted">Hãy đăng ký ngay</p>
+                                    </div>
+
+                                    <form action="http://127.0.0.1:5000/register" method="POST">
+
+                                        <div class="mb-3">
+                                            <label for="fullname" class="form-label">Họ và tên</label>
+                                            <input type="text" name="fullname" class="form-control" id="username" placeholder="Enter fullname">
+                                        </div>
+                                        
+                                        <div class="mb-3">
+                                            <label for="username" class="form-label">Tên đăng nhập</label>
+                                            <input type="text" name="username" class="form-control" id="username" placeholder="Enter username">
+                                        </div>
+
+                                        <div class="mb-3">
+                                            <label for="userpassword" class="form-label">Mật khẩu</label>
+                                            <div class="position-relative auth-pass-inputgroup mb-3">
+                                                <input type="password" name="password" class="form-control pe-5" placeholder="Enter Password" id="password-input">
+                                                <button class="btn btn-link position-absolute end-0 top-0 text-decoration-none text-muted" type="button" id="password-addon"><i class="ri-eye-fill align-middle"></i></button>
+                                            </div>
+                                        </div>
+
+                                        <div class="text-center mt-4">
+                                            <button class="btn btn-primary w-100" type="submit">Đăng ký</button>
+                                        </div>
+                                       
+                                    </form><!-- end form -->
+                                </div>
+                            </div><!-- end col -->
+                        </div><!-- end row -->
+
+                        <div class="row">
+                            <div class="col-xl-12">
+                                <div class="text-center text-muted p-4">
+                                    <p class="mb-0">&copy;
+                                        <script>document.write(new Date().getFullYear())</script> Chaton. Crafted with <i class="mdi mdi-heart text-danger"></i> by Themesdesign
+                                    </p>
+                                </div>
+                            </div><!-- end col -->
+                        </div><!-- end row -->
+
+                    </div>
+                </div>
+            </div>
+            <!-- end col -->
+        </div>
+        <!-- end row -->
+    </div>
+    <!-- end container-fluid -->
+</div>
+<!-- end auth bg -->
+
+<!-- JAVASCRIPT -->
+<script src="https://themesdesign.in/chaton/layouts/assets/libs/bootstrap/js/bootstrap.bundle.min.js"></script>
+<script src="https://themesdesign.in/chaton/layouts/assets/libs/simplebar/simplebar.min.js"></script>
+<script src="https://themesdesign.in/chaton/layouts/assets/libs/node-waves/waves.min.js"></script>
+<!-- password addon -->
+<script src="https://themesdesign.in/chaton/layouts/assets/js/pages/password-addon.init.js"></script>
+
+<!-- theme-style init -->
+<script src="https://themesdesign.in/chaton/layouts/assets/js/pages/theme-style.init.js"></script>
+
+<script src="https://themesdesign.in/chaton/layouts/assets/js/app.js"></script>
+
+</body>
+
+</html>