first commit

.gitattributes

@@ -0,0 +1,5 @@
+*.png filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text
+*.jpeg filter=lfs diff=lfs merge=lfs -text
+*.gif filter=lfs diff=lfs merge=lfs -text
+*.webp filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,8 @@
+.env
+uploads/
+vectorstore/
+__pycache__/
+*.pyc
+.venv/
+
+

app.py

@@ -0,0 +1,255 @@
+import os
+import pickle
+import shutil
+from flask import Flask, request, jsonify, render_template, redirect, url_for
+from flask_login import LoginManager, login_user, logout_user, login_required, current_user
+from dotenv import load_dotenv
+from models import db, User
+from rag.chunker import load_and_chunk
+from rag.embeddings import store_embeddings
+from rag.retriever import retrieve_chunks
+from rag.generator import generate_answer
+from config import SECRET_KEY, SQLALCHEMY_DATABASE_URI
+
+# ── Init ─────────────────────────────────────────────
+load_dotenv()
+app = Flask(__name__)
+
+app.config["SECRET_KEY"] = SECRET_KEY
+app.config["SQLALCHEMY_DATABASE_URI"] = SQLALCHEMY_DATABASE_URI
+app.config["UPLOAD_FOLDER"] = "uploads"
+
+META_PATH = os.path.join("vectorstore", "metadata.pkl")
+
+os.makedirs("uploads", exist_ok=True)
+os.makedirs("vectorstore", exist_ok=True)
+
+# ── Database & Login Manager ──────────────────────────
+db.init_app(app)
+login_manager = LoginManager()
+login_manager.init_app(app)
+login_manager.login_view = "login"
+
+chat_history = {}
+
+@login_manager.user_loader
+def load_user(user_id):
+    return User.query.get(int(user_id))
+
+# ── Create DB ─────────────────────────────────────────
+with app.app_context():
+    db.create_all()
+
+# ── Helper ────────────────────────────────────────────
+def allowed_file(filename):
+    return "." in filename and filename.rsplit(".", 1)[1].lower() in {"pdf"}
+
+def get_user_upload_folder(username):
+    folder = os.path.join("uploads", username)
+    os.makedirs(folder, exist_ok=True)
+    return folder
+
+def get_user_meta_path(username):
+    path = os.path.join("vectorstore", username)
+    os.makedirs(path, exist_ok=True)
+    return os.path.join(path, "metadata.pkl")
+
+# ── Auth Routes ───────────────────────────────────────
+
+@app.route("/")
+@login_required
+def index():
+    return render_template("index.html", username=current_user.username)
+
+@app.route("/register", methods=["GET", "POST"])
+def register():
+    if request.method == "POST":
+        data = request.form
+        username = data.get("username")
+        email = data.get("email")
+        password = data.get("password")
+
+        if User.query.filter_by(username=username).first():
+            return render_template("register.html", error="Username already exists!")
+
+        if User.query.filter_by(email=email).first():
+            return render_template("register.html", error="Email already exists!")
+
+        user = User(username=username, email=email)
+        user.set_password(password)
+        db.session.add(user)
+        db.session.commit()
+
+        return redirect(url_for("login"))
+
+    return render_template("register.html")
+
+@app.route("/login", methods=["GET", "POST"])
+def login():
+    if request.method == "POST":
+        data = request.form
+        username = data.get("username")
+        password = data.get("password")
+
+        user = User.query.filter_by(username=username).first()
+
+        if not user or not user.check_password(password):
+            return render_template("login.html", error="Invalid username or password!")
+
+        login_user(user)
+        return redirect(url_for("index"))
+
+    return render_template("login.html")
+
+@app.route("/logout")
+@login_required
+def logout():
+    logout_user()
+    return redirect(url_for("login"))
+
+# ── App Routes ────────────────────────────────────────
+
+@app.route("/chat")
+@login_required
+def chat():
+    return render_template("chat.html", username=current_user.username)
+
+@app.route("/files", methods=["GET"])
+@login_required
+def get_files():
+    try:
+        folder = get_user_upload_folder(current_user.username)
+        files = [f for f in os.listdir(folder) if f.endswith(".pdf")]
+        return jsonify({"files": files}), 200
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/upload", methods=["GET", "POST"])
+@login_required
+def upload():
+    try:
+        if "pdf" not in request.files:
+            return jsonify({"error": "No file found"}), 400
+
+        file = request.files["pdf"]
+
+        if file.filename == "":
+            return jsonify({"error": "No file selected"}), 400
+
+        if not allowed_file(file.filename):
+            return jsonify({"error": "Only PDF files allowed"}), 400
+
+        folder = get_user_upload_folder(current_user.username)
+        filepath = os.path.join(folder, file.filename)
+        file.save(filepath)
+
+        meta_path = get_user_meta_path(current_user.username)
+        chunks = load_and_chunk(filepath)
+        store_embeddings(chunks, file.filename, meta_path)
+
+        return jsonify({"message": f"{file.filename} uploaded successfully!"}), 200
+
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/ask", methods=["POST"])
+@login_required
+def ask():
+    try:
+        data = request.get_json()
+        question = data.get("question", "").strip()
+        filename = data.get("filename", "").strip()
+
+        if not question:
+            return jsonify({"error": "Question cannot be empty"}), 400
+
+        meta_path = get_user_meta_path(current_user.username)
+        context_chunks = retrieve_chunks(question, filename, meta_path)
+        answer = generate_answer(question, context_chunks)
+
+        username = current_user.username
+        if username not in chat_history:
+            chat_history[username] = []
+
+        chat_history[username].append({
+            "question": question,
+            "answer": answer
+        })
+
+        return jsonify({
+            "answer": answer,
+            "sources": context_chunks
+        }), 200
+
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/history", methods=["GET"])
+@login_required
+def history():
+    try:
+        username = current_user.username
+        return jsonify({"history": chat_history.get(username, [])}), 200
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/clear", methods=["POST"])
+@login_required
+def clear():
+    try:
+        username = current_user.username
+        chat_history[username] = []
+        return jsonify({"message": "Chat history cleared!"}), 200
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/delete", methods=["POST"])
+@login_required
+def delete():
+    try:
+        data = request.get_json()
+        filename = data.get("filename", "")
+
+        if not filename:
+            return jsonify({"error": "Filename not provided"}), 400
+
+        folder = get_user_upload_folder(current_user.username)
+        filepath = os.path.join(folder, filename)
+
+        if not os.path.exists(filepath):
+            return jsonify({"error": "File not found"}), 404
+
+        os.remove(filepath)
+
+        meta_path = get_user_meta_path(current_user.username)
+        if os.path.exists(meta_path):
+            with open(meta_path, "rb") as f:
+                metadata = pickle.load(f)
+            new_metadata = [m for m in metadata if m["filename"] != filename]
+            with open(meta_path, "wb") as f:
+                pickle.dump(new_metadata, f)
+
+        return jsonify({"message": f"{filename} deleted successfully!"}), 200
+
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/clear_vectorstore", methods=["POST"])
+@login_required
+def clear_vectorstore():
+    try:
+        username = current_user.username
+        vectorstore_path = os.path.join("vectorstore", username)
+
+        if os.path.exists(vectorstore_path):
+            shutil.rmtree(vectorstore_path)
+            os.makedirs(vectorstore_path, exist_ok=True)
+
+        return jsonify({"message": "Vector store cleared successfully!"}), 200
+
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+# ── Run ───────────────────────────────────────────────
+if __name__ == "__main__":
+    app.run(debug=True, host="0.0.0.0", port=5000)

config.py

@@ -0,0 +1,20 @@
+import os 
+from dotenv import load_dotenv
+
+load_dotenv()
+
+OLLAMA_BASE_URL = os.getenv("OLLAMA_BASE_URL", "http://localhost:11434")
+MODEL_NAME = os.getenv("MODEL_NAME", "llama3:latest")
+
+EMBEDDING_MODEL = "all-MiniLM-L6-V2"
+
+SECRET_KEY = "your_secret_key_here"
+SQLALCHEMY_DATABASE_URI = "sqlite:///users.db"
+
+CHUNK_SIZE = 500
+CHUNK_OVERLAP = 50
+
+UPLOAD_FOLDER = "uploads"
+CHROMA_DB_PATH = "vectorstore"
+
+TOP_K = 10

models.py

@@ -0,0 +1,18 @@
+from flask_sqlalchemy import SQLAlchemy
+from flask_login import UserMixin
+from werkzeug.security import generate_password_hash, check_password_hash
+
+db = SQLAlchemy()
+
+class User(UserMixin, db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    username = db.Column(db.String(80), unique=True, nullable=False)
+    email = db.Column(db.String(120), unique=True, nullable=False)
+    password = db.Column(db.String(200), nullable=False)
+    upload_folder = db.Column(db.String(200), nullable=True)
+
+    def set_password(self, password):
+        self.password = generate_password_hash(password)
+
+    def check_password(self, password):
+        return check_password_hash(self.password, password)

rag/chunker.py

@@ -0,0 +1,42 @@
+import fitz
+from config import CHUNK_SIZE, CHUNK_OVERLAP
+
+def load_pdf(filepath):
+    doc = fitz.open(filepath)
+    text_pages = []
+
+    for page_num, page in enumerate(doc):
+        text = page.get_text()
+        text_pages.append({
+            "text": text, 
+            "page": page_num + 1
+        })
+
+    doc.close()
+    return text_pages
+
+def split_text(text, page_num):
+    chunks = []
+    start = 0
+
+    while start < len(text):
+        end = start + CHUNK_SIZE
+        chunk = text[start:end]
+
+        if chunk.strip():
+            chunks.append({
+                "text": chunk,
+                "page": page_num
+            })
+        start = end - CHUNK_OVERLAP
+    return chunks
+
+def load_and_chunk(filepath):
+    pages = load_pdf(filepath)
+    all_chunks = []
+
+    for page in pages:
+        chunks = split_text(page["text"], page["page"])
+        all_chunks.extend(chunks)
+
+    return all_chunks

rag/embeddings.py

@@ -0,0 +1,49 @@
+import faiss
+import numpy as np
+import pickle
+import os
+from sentence_transformers import SentenceTransformer
+from config import EMBEDDING_MODEL, CHROMA_DB_PATH
+
+# ── Load Model ───────────────────────────────────────
+embedding_model = SentenceTransformer(EMBEDDING_MODEL)
+
+INDEX_PATH = CHROMA_DB_PATH + "/index.faiss"
+
+def embed_text(text):
+    return embedding_model.encode(text)
+
+# ── Updated to accept meta_path ──────────────────────
+def store_embeddings(chunks, filename, meta_path):
+    embeddings = []
+    metadata = []
+
+    for i, chunk in enumerate(chunks):
+        emb = embed_text(chunk["text"])
+        embeddings.append(emb)
+        metadata.append({
+            "text": chunk["text"],
+            "filename": filename,
+            "page": chunk["page"],
+            "chunk_index": i
+        })
+
+    embeddings_np = np.array(embeddings).astype("float32")
+    dimension = embeddings_np.shape[1]
+
+    index_path = os.path.join(os.path.dirname(meta_path), "index.faiss")
+
+    if os.path.exists(index_path):
+        index = faiss.read_index(index_path)
+        with open(meta_path, "rb") as f:
+            existing_metadata = pickle.load(f)
+    else:
+        index = faiss.IndexFlatL2(dimension)
+        existing_metadata = []
+
+    index.add(embeddings_np)
+    existing_metadata.extend(metadata)
+
+    faiss.write_index(index, index_path)
+    with open(meta_path, "wb") as f:
+        pickle.dump(existing_metadata, f)

rag/generator.py

@@ -0,0 +1,51 @@
+import ollama
+from config import MODEL_NAME
+
+def build_prompt(question, chunks):
+    context = ""
+
+    for i, chunk in enumerate(chunks):
+        context += f"\nchunk {i+1} (From: {chunk['filename']} Page: {chunk['page']}):\n"
+        context += chunk["text"]
+        context += "\n"
+
+
+        prompt = f"""
+        You are a helpful assistant. 
+        Use the following context to answer the question.
+        If you don't know the answer, say "I don't know".
+        Do not make up answers.
+
+        Context:
+        {context}
+
+        Question:
+        {question}
+
+        Answer:
+        """
+
+        return prompt
+    
+
+def call_llm(prompt):
+    response = ollama.chat(
+        model=MODEL_NAME,
+        messages=[
+            {
+                "role": "user",
+                "content": prompt
+            }
+        ]
+    )
+
+    return response["message"]["content"]
+
+def generate_answer(question, chunks):
+    if not chunks:
+        return "no context found in the document."
+    
+    prompt = build_prompt(question, chunks)
+    answer = call_llm(prompt)
+
+    return answer

rag/retriever.py

@@ -0,0 +1,66 @@
+import faiss
+import numpy as np
+import pickle
+import os
+from sentence_transformers import SentenceTransformer
+from config import EMBEDDING_MODEL, TOP_K
+
+embedding_model = SentenceTransformer(EMBEDDING_MODEL)
+
+def embed_query(query):
+    return embedding_model.encode(query)
+
+def retrieve_chunks(query, filename=None, meta_path=None):
+    if meta_path is None or not os.path.exists(meta_path):
+        return []
+
+    index_path = os.path.join(os.path.dirname(meta_path), "index.faiss")
+
+    if not os.path.exists(index_path):
+        return []
+
+    query_embedding = np.array([embed_query(query)]).astype("float32")
+
+    index = faiss.read_index(index_path)
+
+    with open(meta_path, "rb") as f:
+        metadata = pickle.load(f)
+
+    # ── Fix: use min to avoid out of range ──
+    n_results = min(TOP_K, len(metadata))
+
+    if n_results == 0:
+        return []
+
+    distances, indices = index.search(query_embedding, n_results)
+
+    # ── Fix: check distances is not empty ──
+    if len(distances) == 0 or len(distances[0]) == 0:
+        return []
+
+    max_distance = float(distances[0].max()) if distances[0].max() > 0 else 1
+
+    chunks = []
+    for i, idx in enumerate(indices[0]):
+        # ── Fix: skip invalid indices ──
+        if idx == -1 or idx >= len(metadata):
+            continue
+
+        if filename and metadata[idx]["filename"] != filename:
+            continue
+
+        raw_score = float(distances[0][i])
+        confidence = round((1 - (raw_score / max_distance)) * 100, 2)
+
+        chunks.append({
+            "text": metadata[idx]["text"],
+            "filename": metadata[idx]["filename"],
+            "page": metadata[idx]["page"],
+            "score": raw_score,
+            "confidence": confidence
+        })
+
+        if len(chunks) == TOP_K:
+            break
+
+    return chunks 

requirements.txt

@@ -0,0 +1,9 @@
+flask
+python-dotenv
+pymupdf
+faiss-cpu
+sentence-transformers
+ollama
+flask-login
+flask-sqlalchemy
+werkzeug

static/style.css

@@ -0,0 +1,376 @@
+/* ── Global ─────────────────────────────────────────── */
+* {
+    margin: 0;
+    padding: 0;
+    box-sizing: border-box;
+    font-family: Arial, sans-serif;
+}
+
+body {
+    background-color: #f5f5f5;
+    color: #333;
+}
+
+header {
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    padding: 20px;
+    background-color: #2c3e50;
+    color: white;
+}
+
+.header-right {
+    display: flex;
+    align-items: center;
+    gap: 15px;
+    font-size: 14px;
+}
+
+.logout-btn {
+    background-color: #e74c3c;
+    padding: 6px 14px;
+    font-size: 13px;
+}
+
+.logout-btn:hover {
+    background-color: #c0392b;
+}
+
+header h1 {
+    font-size: 28px;
+    margin-bottom: 5px;
+}
+
+header p {
+    font-size: 14px;
+    color: #ccc;
+}
+
+header a {
+    color: #ccc;
+    text-decoration: none;
+    font-size: 14px;
+}
+
+header a:hover {
+    color: white;
+}
+
+/* ── Container ──────────────────────────────────────── */
+.container {
+    max-width: 800px;
+    margin: 30px auto;
+    padding: 20px;
+}
+
+/* ── Upload Box ─────────────────────────────────────── */
+.upload-box {
+    background-color: white;
+    padding: 30px;
+    border-radius: 10px;
+    box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+    text-align: center;
+}
+
+.upload-box h2 {
+    margin-bottom: 20px;
+    color: #2c3e50;
+}
+
+.upload-box input[type="file"] {
+    display: block;
+    margin: 0 auto 15px auto;
+    padding: 10px;
+}
+
+/* ── Buttons ────────────────────────────────────────── */
+button {
+    background-color: #2c3e50;
+    color: white;
+    border: none;
+    padding: 10px 20px;
+    border-radius: 5px;
+    cursor: pointer;
+    font-size: 14px;
+    margin: 5px;
+}
+
+button:hover {
+    background-color: #1a252f;
+}
+
+/* ── Nav Button ─────────────────────────────────────── */
+.nav-btn {
+    text-align: center;
+    margin-top: 20px;
+}
+
+.nav-btn a {
+    text-decoration: none;
+}
+
+/* ── Upload Status ──────────────────────────────────── */
+#uploadStatus {
+    margin-top: 15px;
+    font-size: 14px;
+    font-weight: bold;
+}
+
+/* ── Chat Box ───────────────────────────────────────── */
+.chat-box {
+    background-color: white;
+    border-radius: 10px;
+    box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+    padding: 20px;
+    height: 450px;
+    overflow-y: auto;
+    margin-bottom: 15px;
+}
+
+/* ── Messages ───────────────────────────────────────── */
+.message {
+    padding: 10px 15px;
+    border-radius: 8px;
+    margin-bottom: 10px;
+    line-height: 1.5;
+}
+
+.message.user {
+    background-color: #d6eaf8;
+    text-align: right;
+}
+
+.message.bot {
+    background-color: #eafaf1;
+    text-align: left;
+}
+
+.message.error {
+    background-color: #fadbd8;
+    text-align: left;
+}
+
+/* ── Sources ────────────────────────────────────────── */
+.sources {
+    background-color: #fef9e7;
+    border-left: 4px solid #f39c12;
+    padding: 10px 15px;
+    border-radius: 5px;
+    margin-bottom: 10px;
+    font-size: 13px;
+}
+
+.sources ul {
+    margin-top: 5px;
+    padding-left: 20px;
+}
+
+.sources ul li {
+    margin-top: 3px;
+}
+/* ...existing code... */
+
+.confidence {
+    font-size: 12px;
+    font-weight: bold;
+    margin-left: 10px;
+}
+
+.sources li {
+    display: flex;
+    align-items: center;
+    justify-content: space-between;
+    padding: 5px 0;
+    border-bottom: 1px solid #f0e0b0;
+}
+
+.sources li:last-child {
+    border-bottom: none;
+}
+.input-area {
+    display: flex;
+    gap: 10px;
+}
+
+.input-area input {
+    flex: 1;
+    padding: 10px 15px;
+    border: 1px solid #ccc;
+    border-radius: 5px;
+    font-size: 14px;
+    outline: none;
+}
+
+.input-area input:focus {
+    border-color: #2c3e50;
+}
+
+#loader {
+    text-align: center;
+    font-size: 14px;
+    color: #888;
+    margin-top: 10px;
+}
+
+/* ...existing code... */
+
+/* ── Files Box ──────────────────────────────────────── */
+.files-box {
+    background-color: white;
+    padding: 20px;
+    border-radius: 10px;
+    box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+    margin-top: 20px;
+}
+
+.files-box h2 {
+    margin-bottom: 15px;
+    color: #2c3e50;
+}
+
+.files-box ul {
+    list-style: none;
+    padding: 0;
+}
+
+
+.file-item {
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    padding: 10px 15px;
+    border-bottom: 1px solid #eee;
+    font-size: 14px;
+    color: #333;
+}
+
+.file-item:last-child {
+    border-bottom: none;
+}
+
+.file-item:hover {
+    background-color: #f5f5f5;
+    border-radius: 5px;
+}
+
+.delete-btn {
+    background-color: #e74c3c;
+    color: white;
+    border: none;
+    padding: 5px 10px;
+    border-radius: 5px;
+    cursor: pointer;
+    font-size: 12px;
+}
+
+.delete-btn:hover {
+    background-color: #c0392b;
+}
+
+
+/* ── Clear Box ───────────────────────────────────────── */
+.clear-box {
+    background-color: white;
+    padding: 20px;
+    border-radius: 10px;
+    box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+    margin-top: 20px;
+    text-align: center;
+}
+
+.clear-box h2 {
+    margin-bottom: 10px;
+    color: #2c3e50;
+}
+
+.clear-box p {
+    font-size: 13px;
+    color: #888;
+    margin-bottom: 15px;
+}
+
+/* ── Clear Vector Button ─────────────────────────────── */
+.clear-vector-btn {
+    background-color: #e67e22;
+    color: white;
+    border: none;
+    padding: 10px 20px;
+    border-radius: 5px;
+    cursor: pointer;
+    font-size: 14px;
+}
+
+.clear-vector-btn:hover {
+    background-color: #d35400;
+}
+
+#clearStatus {
+    margin-top: 10px;
+    font-size: 14px;
+    font-weight: bold;
+}
+
+/* ...existing code... */
+
+/* ── Auth Box ────────────────────────────────────────── */
+.auth-box {
+    background-color: white;
+    padding: 30px;
+    border-radius: 10px;
+    box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+    max-width: 400px;
+    margin: 40px auto;
+    text-align: center;
+}
+
+.auth-box h2 {
+    margin-bottom: 20px;
+    color: #2c3e50;
+}
+
+.auth-box input {
+    width: 100%;
+    padding: 10px 15px;
+    margin-bottom: 15px;
+    border: 1px solid #ccc;
+    border-radius: 5px;
+    font-size: 14px;
+    outline: none;
+    box-sizing: border-box;
+}
+
+.auth-box input:focus {
+    border-color: #2c3e50;
+}
+
+.auth-box button {
+    width: 100%;
+    padding: 10px;
+    font-size: 15px;
+}
+
+.auth-box p {
+    margin-top: 15px;
+    font-size: 13px;
+    color: #888;
+}
+
+.auth-box a {
+    color: #2c3e50;
+    font-weight: bold;
+    text-decoration: none;
+}
+
+.auth-box a:hover {
+    text-decoration: underline;
+}
+
+.error-msg {
+    background-color: #fadbd8;
+    color: #e74c3c;
+    padding: 10px;
+    border-radius: 5px;
+    margin-bottom: 15px;
+    font-size: 13px;
+}

templates/chat.html

@@ -0,0 +1,169 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>RAG Chat</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='style.css') }}">
+</head>
+<body>
+
+    <!-- ── Header ── -->
+    <header>
+        <h1>💬 RAG PDF Assistant</h1>
+        <a href="/">← Upload New PDF</a>
+    </header>
+
+    <!-- ── Chat Box ── -->
+    <div class="container">
+
+        <!-- ── Current File Badge ── -->
+        <div id="currentFileBadge" class="file-badge">
+            📄 No file selected
+        </div>
+
+        <div class="chat-box" id="chatBox">
+            <!-- Messages appear here -->
+        </div>
+
+        <!-- ── Input Area ── -->
+        <div class="input-area">
+            <input 
+                type="text" 
+                id="questionInput" 
+                placeholder="Ask a question..." 
+            />
+            <button id="askBtn">Ask</button>
+            <button id="clearBtn">Clear History</button>
+        </div>
+
+        <!-- ── Loading Spinner ── -->
+        <div id="loader" style="display:none;">
+            ⏳ Thinking...
+        </div>
+    </div>
+
+    <script>
+        // ── Store Current Filename ───────────────────────
+        let currentFile = localStorage.getItem("currentFile") || ""
+
+        // ── Show Current File Badge ──────────────────────
+        const badge = document.getElementById("currentFileBadge")
+        if (currentFile) {
+            badge.innerText = `📄 Asking from: ${currentFile}`
+            badge.style.backgroundColor = "#eafaf1"
+            badge.style.color = "green"
+        } else {
+            badge.innerText = "⚠️ No file selected - Go back and upload a PDF"
+            badge.style.backgroundColor = "#fadbd8"
+            badge.style.color = "red"
+        }
+
+        // ── Ask Question ─────────────────────────────────
+        document.getElementById("askBtn").addEventListener("click", async () => {
+            const question = document.getElementById("questionInput").value.trim()
+            const chatBox = document.getElementById("chatBox")
+            const loader = document.getElementById("loader")
+
+            if (!question) {
+                alert("Please enter a question!")
+                return
+            }
+
+            if (!currentFile) {
+                alert("Please upload a PDF first!")
+                return
+            }
+
+            // ── Show user message ──
+            chatBox.innerHTML += `
+                <div class="message user">
+                    <strong>You:</strong> ${question}
+                </div>
+            `
+
+            document.getElementById("questionInput").value = ""
+            loader.style.display = "block"
+            chatBox.scrollTop = chatBox.scrollHeight
+
+            try {
+                const response = await fetch("/ask", {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ 
+                        question: question,
+                        filename: currentFile
+                    })
+                })
+
+                const data = await response.json()
+                loader.style.display = "none"
+
+                if (response.ok) {
+                    // ── Show answer ──
+                    chatBox.innerHTML += `
+                        <div class="message bot">
+                            <strong>Assistant:</strong> ${data.answer}
+                        </div>
+                    `
+
+                    // ── Show sources with confidence score ──
+                    if (data.sources && data.sources.length > 0) {
+                        let sourcesHtml = `<div class="sources"><strong>📌 Sources:</strong><ul>`
+                        data.sources.forEach(src => {
+                            let confidenceColor = src.confidence >= 70 ? "green"
+                                               : src.confidence >= 40 ? "orange"
+                                               : "red"
+                            sourcesHtml += `
+                                <li>
+                                    📄 ${src.filename} - Page ${src.page}
+                                    <span class="confidence" style="color:${confidenceColor}">
+                                        ● ${src.confidence}% match
+                                    </span>
+                                </li>
+                            `
+                        })
+                        sourcesHtml += `</ul></div>`
+                        chatBox.innerHTML += sourcesHtml
+                    }
+
+                } else {
+                    chatBox.innerHTML += `
+                        <div class="message error">
+                            ❌ ${data.error}
+                        </div>
+                    `
+                }
+
+            } catch (error) {
+                loader.style.display = "none"
+                chatBox.innerHTML += `
+                    <div class="message error">
+                        ❌ Something went wrong. Please try again.
+                    </div>
+                `
+            }
+
+            // ── Auto scroll to bottom ──
+            chatBox.scrollTop = chatBox.scrollHeight
+        })
+
+
+        // ── Clear History ─────────────────────────────────
+        document.getElementById("clearBtn").addEventListener("click", async () => {
+            if (!confirm("Are you sure you want to clear chat history?")) return
+            await fetch("/clear", { method: "POST" })
+            document.getElementById("chatBox").innerHTML = ""
+        })
+
+
+        // ── Enter Key Support ─────────────────────────────
+        document.getElementById("questionInput").addEventListener("keypress", (e) => {
+            if (e.key === "Enter") {
+                document.getElementById("askBtn").click()
+            }
+        })
+    </script>
+
+</body>
+</html>

templates/index.html

@@ -0,0 +1,165 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>RAG Application</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='style.css') }}">
+</head>
+<body>
+
+    <!-- ── Header ── -->
+    <header>
+    <h1>📄 RAG PDF Assistant</h1>
+    <div class="header-right">
+        <span>👤 {{ username }}</span>
+        <a href="/logout"><button class="logout-btn">Logout</button></a>
+    </div>
+    </header>
+
+    <div class="container">
+        <!-- ── Upload Section ── -->
+        <div class="upload-box">
+            <h2>Upload PDF</h2>
+            <form id="uploadForm">
+                <input type="file" id="pdfFile" accept=".pdf" required />
+                <button type="submit">Upload & Index</button>
+            </form>
+            <div id="uploadStatus"></div>
+        </div>
+
+        <!-- ── Uploaded Files List ── -->
+        <div class="files-box">
+            <h2>📋 Uploaded Files</h2>
+            <ul id="filesList">
+                <!-- Files appear here -->
+            </ul>
+        </div>
+
+        <div class="clear-box">
+            <h2>🔄 Reset Vector Store</h2>
+            <p>This will delete all stored embeddings</p>
+            <button id="clearVectorBtn" class="clear-vector-btn">
+                🔄 Clear Vector Store
+            </button>
+            <div id="clearStatus"></div>
+        </div>
+
+        <!-- ── Navigate to Chat ── -->
+        <div class="nav-btn">
+            <a href="/chat">
+                <button>Go to Chat →</button>
+            </a>
+        </div>
+    </div>
+
+    <script>
+        // ── Load Files on Page Load ───────────────────────
+        async function loadFiles() {
+            const response = await fetch("/files")
+            const data = await response.json()
+            const filesList = document.getElementById("filesList")
+
+            filesList.innerHTML = ""
+
+            if (data.files.length === 0) {
+                filesList.innerHTML = "<li>No files uploaded yet</li>"
+                return
+            }
+
+            data.files.forEach(file => {
+                filesList.innerHTML += `
+                    <li class="file-item">
+                    <span>📄 ${file}</span>
+                    <button 
+                        class="delete-btn" 
+                        onclick="deleteFile('${file}')">
+                        🗑️ Delete
+                    </button>
+                </li>
+                `
+            })
+        }
+
+        // ── Upload PDF ────────────────────────────────────
+        document.getElementById("uploadForm").addEventListener("submit", async (e) => {
+            e.preventDefault()
+
+            const fileInput = document.getElementById("pdfFile")
+            const statusDiv = document.getElementById("uploadStatus")
+            const formData = new FormData()
+
+            formData.append("pdf", fileInput.files[0])
+            statusDiv.innerText = "Uploading..."
+
+            const response = await fetch("/upload", {
+                method: "POST",
+                body: formData
+            })
+
+            const data = await response.json()
+
+            if (response.ok) {
+                localStorage.setItem("currentFile", fileInput.files[0].name)
+                statusDiv.innerText = "✅ " + data.message
+                statusDiv.style.color = "green"
+                loadFiles()  // Refresh files list
+            } else {
+                statusDiv.innerText = "❌ " + data.error
+                statusDiv.style.color = "red"
+            }
+        })
+
+        document.getElementById("clearVectorBtn").addEventListener("click", async () => {
+        if (!confirm("Are you sure? This will delete ALL embeddings!")) return
+
+        const clearStatus = document.getElementById("clearStatus")
+        clearStatus.innerText = "Clearing..."
+
+        const response = await fetch("/clear_vectorstore", {
+            method: "POST"
+        })
+
+        const data = await response.json()
+
+        if (response.ok) {
+            clearStatus.innerText = "✅ " + data.message
+            clearStatus.style.color = "green"
+            localStorage.removeItem("currentFile")
+            loadFiles()
+        } else {
+            clearStatus.innerText = "❌ " + data.error
+            clearStatus.style.color = "red"
+        }
+    })
+
+        // ── Load Files on Page Load ───────────────────────
+        async function deleteFile(filename) {
+        if (!confirm(`Are you sure you want to delete ${filename}?`)) return
+
+        const response = await fetch("/delete", {
+            method: "POST",
+            headers: { "Content-Type": "application/json" },
+            body: JSON.stringify({ filename: filename })
+        })
+
+        const data = await response.json()
+
+        if (response.ok) {
+            alert("✅ " + data.message)
+
+            // Clear localStorage if deleted file was selected
+            if (localStorage.getItem("currentFile") === filename) {
+                localStorage.removeItem("currentFile")
+            }
+
+            loadFiles() // Refresh list
+        } else {
+            alert("❌ " + data.error)
+        }
+    }
+        loadFiles()
+    </script>
+
+</body>
+</html>

templates/login.html

@@ -0,0 +1,32 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Login</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='style.css') }}">
+</head>
+<body>
+    <header>
+        <h1>🔑 Login</h1>
+    </header>
+
+    <div class="container">
+        <div class="auth-box">
+            <h2>Welcome Back!</h2>
+
+            {% if error %}
+            <div class="error-msg">❌ {{ error }}</div>
+            {% endif %}
+
+            <form method="POST" action="/login">
+                <input type="text" name="username" placeholder="Username" required />
+                <input type="password" name="password" placeholder="Password" required />
+                <button type="submit">Login</button>
+            </form>
+
+            <p>Don't have an account? <a href="/register">Register</a></p>
+        </div>
+    </div>
+</body>
+</html>

templates/register.html

@@ -0,0 +1,33 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Register</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='style.css') }}">
+</head>
+<body>
+    <header>
+        <h1>📝 Register</h1>
+    </header>
+
+    <div class="container">
+        <div class="auth-box">
+            <h2>Create Account</h2>
+
+            {% if error %}
+            <div class="error-msg">❌ {{ error }}</div>
+            {% endif %}
+
+            <form method="POST" action="/register">
+                <input type="text" name="username" placeholder="Username" required />
+                <input type="email" name="email" placeholder="Email" required />
+                <input type="password" name="password" placeholder="Password" required />
+                <button type="submit">Register</button>
+            </form>
+
+            <p>Already have an account? <a href="/login">Login</a></p>
+        </div>
+    </div>
+</body>
+</html>