Upload 13 files

.gitattributes

@@ -39,3 +39,5 @@ public/JUIT_icon.png filter=lfs diff=lfs merge=lfs -text
 public/raman.png filter=lfs diff=lfs merge=lfs -text
 backend/chroma_store/chroma.sqlite3 filter=lfs diff=lfs merge=lfs -text
 backend/ragg_pdf.pdf filter=lfs diff=lfs merge=lfs -text
+chroma_store/chroma.sqlite3 filter=lfs diff=lfs merge=lfs -text
+static/JUIT_icon.png filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["gunicorn", "-b", "0.0.0.0:7860", "app:app"]

app.py

@@ -0,0 +1,84 @@
+from dotenv import load_dotenv
+load_dotenv()
+
+import os
+from flask import Flask, request, jsonify, render_template
+from flask_cors import CORS
+
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_community.vectorstores import Chroma
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_core.prompts import PromptTemplate
+from langchain.chains import RetrievalQA
+
+# === Step 1: Load API Key ===
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
+if not GOOGLE_API_KEY:
+    raise ValueError("GOOGLE_API_KEY not found in environment variables.")
+
+# === Step 2: Initialize LLM (Gemini) ===
+llm = ChatGoogleGenerativeAI(
+    model="gemini-2.0-flash-lite",
+    google_api_key=GOOGLE_API_KEY,
+    convert_system_message_to_human=True
+)
+
+# === Step 3: Load Chroma Vector Store ===
+embedding_model = HuggingFaceEmbeddings(model_name="BAAI/bge-large-en-v1.5")
+vectordb = Chroma(
+    persist_directory="./chroma_store",
+    embedding_function=embedding_model,
+    collection_name="pdf_search_chroma"
+)
+retriever = vectordb.as_retriever(search_kwargs={"k": 6})
+
+# === Step 4: Custom Prompt Template ===
+prompt_template = PromptTemplate.from_template("""
+You are an intelligent assistant for students asking about their university.
+If answer is not defined or not clearly understood, ask for clarification.
+Answer clearly and helpfully based on the retrieved context. Do not make up information or suggestions.
+
+Context:
+{context}
+
+Question:
+{question}
+
+Answer:
+""")
+
+# === Step 5: Create Retrieval-QA Chain ===
+qa_chain = RetrievalQA.from_chain_type(
+    llm=llm,
+    chain_type="stuff",
+    retriever=retriever,
+    chain_type_kwargs={"prompt": prompt_template}
+)
+
+# === Step 6: Flask Setup ===
+app = Flask(__name__, static_folder="static", template_folder="templates")
+CORS(app)
+
+# === Step 7: Serve Frontend ===
+@app.route("/")
+def index():
+    return render_template("index.html")
+
+# === Step 8: API Endpoint ===
+@app.route('/api/chat', methods=['POST'])
+def chat():
+    data = request.json
+    query = data.get('message', '').strip()
+
+    if not query:
+        return jsonify({"error": "No message provided."}), 400
+
+    try:
+        response = qa_chain.run(query)
+        return jsonify({"response": response})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+# === Step 9: Run the App ===
+if __name__ == '__main__':
+    app.run(debug=False)

chroma_store/chroma.sqlite3

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

chroma_store/e83a4737-70af-4828-8889-50c4503e6ab0/data_level0.bin

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

chroma_store/e83a4737-70af-4828-8889-50c4503e6ab0/header.bin

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

chroma_store/e83a4737-70af-4828-8889-50c4503e6ab0/index_metadata.pickle

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

chroma_store/e83a4737-70af-4828-8889-50c4503e6ab0/length.bin

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

chroma_store/e83a4737-70af-4828-8889-50c4503e6ab0/link_lists.bin

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

requirements.txt

@@ -0,0 +1,27 @@
+chromadb==1.0.4
+faiss-cpu==1.7.4
+fastapi==0.115.9
+Flask==3.1.0
+flask-cors==5.0.1
+fpdf2==2.7.7
+google-ai-generativelanguage==0.4.0
+google-api-core==2.24.2
+google-auth==2.38.0
+google-generativeai==0.4.1
+gunicorn==23.0.0
+huggingface-hub==0.30.2
+importlib_metadata==8.4.0
+importlib_resources==6.5.2
+langchain==0.1.13
+langchain-community==0.0.29
+langchain-google-genai==0.0.11
+numpy==1.26.4
+pypdf==5.4.0
+PyPDF2==3.0.1
+python-dotenv==1.0.1
+sentence-transformers==4.0.2
+db-sqlite3
+transformers==4.51.3
+torch==2.6.0
+uvicorn==0.34.1
+gunicorn

static/script.js

@@ -0,0 +1,124 @@
+const chatContainer = document.getElementById("chat-container");
+const chatInput = document.getElementById("chat-input");
+const sendButton = document.getElementById("send-button");
+const themeToggle = document.getElementById("theme-toggle");
+
+let isTyping = false;
+
+const initialMessage = {
+  id: "1",
+  text: "Hello! I'm the JUIT University Assistant. How can I help you today?",
+  isUser: false,
+};
+
+let messages = [initialMessage];
+
+function renderMessages() {
+  chatContainer.innerHTML = "";
+  messages.forEach((msg) => {
+    const bubble = document.createElement("div");
+    bubble.className = `flex items-start gap-2 ${msg.isUser ? "flex-row-reverse" : "flex-row"}`;
+
+    const avatar = document.createElement("div");
+    avatar.className =
+      "h-6 w-6 rounded-full bg-gray-200 dark:bg-gray-700 flex items-center justify-center text-sm";
+    avatar.innerText = msg.isUser ? "👤" : "🎓";
+
+    const msgDiv = document.createElement("div");
+    msgDiv.className = `max-w-[75%] break-words rounded-md px-3 py-1 text-sm ${
+      msg.isUser
+        ? "bg-blue-600 text-white rounded-tr-none"
+        : "bg-gray-200 dark:bg-gray-700 text-black dark:text-white rounded-tl-none"
+    }`;
+
+    msgDiv.innerText = msg.text;
+    bubble.appendChild(avatar);
+    bubble.appendChild(msgDiv);
+    chatContainer.appendChild(bubble);
+  });
+
+  if (isTyping) {
+    const typing = document.createElement("div");
+    typing.className = "flex items-start gap-2";
+    typing.innerHTML = `
+      <div class="h-6 w-6 rounded-full bg-gray-200 dark:bg-gray-700 flex items-center justify-center text-sm">🎓</div>
+      <div class="max-w-[75%] rounded-md rounded-tl-none bg-gray-200 dark:bg-gray-700 px-3 py-1 text-sm animate-pulse">Typing...</div>
+    `;
+    chatContainer.appendChild(typing);
+  }
+
+  chatContainer.scrollTop = chatContainer.scrollHeight;
+}
+
+async function simulateBotResponse(userInput) {
+  const response = await fetch("http://localhost:8080/api/chat", {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify({ message: userInput }),
+  });
+
+  if (!response.ok) {
+    throw new Error("API error");
+  }
+
+  const data = await response.json();
+  if (data.error) {
+    throw new Error(data.error);
+  }
+
+  return data;
+}
+
+async function handleSendMessage() {
+  const text = chatInput.value.trim();
+  if (!text) return;
+
+  const userMessage = {
+    id: Date.now().toString(),
+    text,
+    isUser: true,
+  };
+
+  messages.push(userMessage);
+  chatInput.value = "";
+  isTyping = true;
+  renderMessages();
+
+  try {
+    const res = await simulateBotResponse(text);
+    const botMessage = {
+      id: (Date.now() + 1).toString(),
+      text: res.response,
+      isUser: false,
+    };
+    messages.push(botMessage);
+  } catch (error) {
+    messages.push({
+      id: Date.now().toString(),
+      text: "Error: " + error.message,
+      isUser: false,
+    });
+  }
+
+  isTyping = false;
+  renderMessages();
+}
+
+sendButton.addEventListener("click", handleSendMessage);
+
+chatInput.addEventListener("keydown", (e) => {
+  if (e.key === "Enter" && !e.shiftKey) {
+    e.preventDefault();
+    handleSendMessage();
+  }
+});
+
+themeToggle.addEventListener("click", () => {
+  document.body.classList.toggle("dark");
+  const isDark = document.body.classList.contains("dark");
+  themeToggle.textContent = isDark ? "☀️ Light" : "🌙 Dark";
+});
+
+renderMessages();

templates/index.html

@@ -0,0 +1,217 @@
+<!DOCTYPE html>
+<html lang="en">
+
+<head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>University Assistant</title>
+    <script src="https://cdn.tailwindcss.com"></script>
+    <script>
+        tailwind.config = {
+            darkMode: 'class',
+        };
+    </script>
+    <style>
+        html,
+        body {
+            margin: 0;
+            padding: 0;
+            height: 100%;
+            font-family: 'Inter', sans-serif;
+        }
+
+        @keyframes wave {
+            0%,100% {
+                transform: translateY(0);
+            }
+
+            50% {
+                transform: translateY(-5px);
+            }
+        } 
+
+        /* @keyframes wave {
+            0%,100% {
+                transform: translateY(0);
+            }
+
+            50% {
+                transform: translateY(-5px);
+            }
+        } */
+
+        .dot-typing span {
+            display: inline-block;
+            animation: wave 1.4s infinite ease-in-out;
+            font-size: 1.5rem;
+            margin: 0 2px;
+            color: black;
+            /* Or white in dark mode */
+        }
+
+        .dot-typing span:nth-child(1) {
+            animation-delay: 0s;
+        }
+
+        .dot-typing span:nth-child(2) {
+            animation-delay: 0.2s;
+        }
+
+        .dot-typing span:nth-child(3) {
+            animation-delay: 0.4s;
+        }
+    </style>
+</head>
+
+<body class="bg-zinc-100 dark:bg-slate-950 text-black dark:text-white flex items-center justify-center min-h-screen">
+    <div
+        class="flex w-full w-full flex-col overflow-hidden border border-slate-800 dark:border-gray-700 shadow-lg rounded-lg h-[90vh] h-full">
+
+        <div class="flex flex-col flex-grow overflow-hidden">
+            <div id="chat-container" class="flex-grow overflow-y-auto p-3 space-y-3 bg-zinc-100 dark:bg-slate-950">
+            </div>
+            <div class="border-t p-2 border-slate-800 dark:border-slate-800">
+                <div class="flex gap-2">
+                    <input id="chat-input"
+                        class="flex-grow rounded-md border border-slate-800 dark:border-slate-800 dark:bg-slate-950 bg-zinc-100 dark:bg-slate-800 px-3 py-1 text-sm text-black dark:text-white focus:outline-none"
+                        type="text" placeholder="Type your question here..." />
+                    <button id="send-button"
+                        class="rounded-md bg-blue-600 px-3 py-1 text-white hover:bg-blue-700">➤</button>
+                </div>
+            </div>
+        </div>
+    </div>
+
+    <script>
+        window.addEventListener("message", (event) => {
+            if (event.origin !== "https://your-domain.com") return; // security
+            const {
+                type,
+                theme
+            } = event.data;
+
+            if (type === "set-theme" && (theme === "light" || theme === "dark")) {
+                document.documentElement.setAttribute("data-theme", theme);
+                // Or, toggle class if your app uses class-based dark mode:
+                document.documentElement.classList.toggle("dark", theme === "dark");
+            }
+        });
+
+        const chatContainer = document.getElementById("chat-container");
+        const chatInput = document.getElementById("chat-input");
+        const sendButton = document.getElementById("send-button");
+
+        const initialMessage = {
+            id: "1",
+            text: "Hello! I'm the JUIT University Assistant. How can I help you today?",
+            isUser: false,
+        };
+
+        let messages = [initialMessage];
+        let isTyping = false;
+
+        function renderMessages() {
+            chatContainer.innerHTML = "";
+            messages.forEach((msg) => {
+                const bubble = document.createElement("div");
+                bubble.className = `flex items-start gap-2 ${msg.isUser ? "flex-row-reverse" : "flex-row"}`;
+                const avatar = document.createElement("img");
+                avatar.className = "h-9 w-9 rounded-full object-cover";
+                avatar.src = msg.isUser ? "../static/profile.png" :
+                    "../static/JUIT_icon.png"; // adjust path if needed
+                avatar.alt = msg.isUser ? "User" : "Bot";
+                const msgDiv = document.createElement("div");
+                msgDiv.className = `max-w-[75%] break-words rounded-md px-3 py-1 text-sm ${
+            msg.isUser
+                ? "bg-blue-600 text-white rounded-tr-none"
+                : "bg-gray-200 text-black dark:bg-gray-700 dark:text-white rounded-tl-none"
+        }`;
+                msgDiv.innerText = msg.text;
+                bubble.appendChild(avatar);
+                bubble.appendChild(msgDiv);
+                chatContainer.appendChild(bubble);
+            });
+            if (isTyping) {
+                const typing = document.createElement("div");
+                typing.className = "flex items-start gap-2";
+                typing.innerHTML = `
+                    <img src="../static/JUIT_icon.png" alt="Bot" class="h-9 w-9 rounded-full object-cover" />
+                    <div class="max-w-[75%] rounded-md rounded-tl-none bg-gray-200 dark:bg-gray-700 px-3 py-1 text-sm">
+                    <div class="dot-typing text-black dark:text-white">
+                        <span>.</span><span>.</span><span>.</span>
+                    </div>
+                    </div>
+                `;
+                chatContainer.appendChild(typing);
+            }
+
+            chatContainer.scrollTop = chatContainer.scrollHeight;
+        }
+
+        async function simulateBotResponse(userInput) {
+            const response = await fetch("http://localhost:8080/api/chat", {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                },
+                body: JSON.stringify({
+                    message: userInput
+                }),
+            });
+
+            if (!response.ok) throw new Error("API error");
+
+            const data = await response.json();
+            if (data.error) throw new Error(data.error);
+
+            return data;
+        }
+
+        async function handleSendMessage() {
+            const text = chatInput.value.trim();
+            if (!text) return;
+
+            const userMessage = {
+                id: Date.now().toString(),
+                text,
+                isUser: true,
+            };
+
+            messages.push(userMessage);
+            chatInput.value = "";
+            isTyping = true;
+            renderMessages();
+
+            try {
+                const res = await simulateBotResponse(text);
+                const botMessage = {
+                    id: (Date.now() + 1).toString(),
+                    text: res.response,
+                    isUser: false,
+                };
+                messages.push(botMessage);
+            } catch (error) {
+                messages.push({
+                    id: Date.now().toString(),
+                    text: "Error: " + error.message,
+                    isUser: false,
+                });
+            }
+
+            isTyping = false;
+            renderMessages();
+        }
+
+        sendButton.addEventListener("click", handleSendMessage);
+        chatInput.addEventListener("keydown", (e) => {
+            if (e.key === "Enter" && !e.shiftKey) {
+                e.preventDefault();
+                handleSendMessage();
+            }
+        });
+
+        renderMessages();
+    </script>
+</body>
+
+</html>