Spaces:

MissSqui
/

Static_H

Running

App Files Files Community

MissSqui commited on Jun 9, 2025

Commit

f9e92fa

verified ·

1 Parent(s): 887ddaa

Update abc12

Browse files

Files changed (1) hide show

abc12 +1092 -801

abc12 CHANGED Viewed

@@ -1,804 +1,1095 @@
-<!DOCTYPE html>
-<html lang="en">
-<head>
-  <meta charset="UTF-8" />
-  <meta name="viewport" content="width=device-width, initial-scale=1.0"/>
-  <title>Chat Interface</title>
-  <!-- Bootstrap CSS & Icons -->
-  <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.2/dist/css/bootstrap.min.css" rel="stylesheet">
-  <link href="https://cdn.jsdelivr.net/npm/bootstrap-icons@1.10.5/font/bootstrap-icons.css" rel="stylesheet">
-  <!-- Tailwind (only if you still need it) -->
-  <script src="https://cdn.tailwindcss.com"></script>
-  <style>
-    body {
-      background: linear-gradient(to bottom right, #e0f2ff, #ede9fe);
-      padding-bottom: 4rem;
-    }
-    .loading-dots {
-      display: inline-block;
-      text-align: left;
-    }
-    .loading-dots span {
-      display: inline-block;
-      width: 8px;
-      height: 8px;
-      margin: 0 2px;
-      background-color: #555;
-      border-radius: 50%;
-      animation: blink 1.4s infinite both;
-    }
-    .loading-dots span:nth-child(2) {
-      animation-delay: 0.2s;
-    }
-    .loading-dots span:nth-child(3) {
-      animation-delay: 0.4s;
-    }
-    @keyframes blink {
-      0%, 80%, 100% { opacity: 0; }
-      40% { opacity: 1; }
-    }
-  </style>
-</head>
-<body class="font-sans">
-  <!-- HEADER -->
-  <div class="bg-gradient-to-r from-blue-500 to-violet-600 text-white shadow-md py-4 px-6 flex justify-between items-center">
-    <div class="flex items-center gap-4">
-      <!-- Back button now has id="backBtn" and no inline onclick -->
-      <button id="backBtn"
-              class="text-white hover:text-gray-200 transition"
-              aria-label="Go back">
-        <svg xmlns="http://www.w3.org/2000/svg" class="h-6 w-6" fill="none"
-             viewBox="0 0 24 24" stroke="currentColor">
-          <path stroke-linecap="round" stroke-linejoin="round" stroke-width="2"
-                d="M15 19l-7-7 7-7" />
-        </svg>
-      </button>
-      <h2 class="text-2xl font-bold">📚 Document Chat &amp; Summarization</h2>
-    </div>
-    <div class="relative inline-block text-left">
-      <select id="zone-dropdown"
-              class="min-w-[14rem] appearance-none px-3 py-2 border rounded-lg bg-white text-gray-800 shadow focus:outline-none focus:ring-2 focus:ring-blue-500">
-      </select>
-      <div class="pointer-events-none absolute inset-y-0 right-3 flex items-center">
-        <svg class="w-4 h-4 text-gray-500" fill="none" stroke="currentColor" stroke-width="2"
-             viewBox="0 0 24 24" stroke-linecap="round" stroke-linejoin="round">
-          <polyline points="6 9 12 15 18 9"></polyline>
-        </svg>
-      </div>
-    </div>
-  </div>
-  <!-- DIRECTORY PATH DISPLAY -->
-  <div class="max-w-7xl mx-auto mt-6 px-4 text-gray-700 text-sm font-medium select-none">
-    <span id="effort-path" class="text-gray-800 font-semibold">{{ effort_name }}</span>
-    <span class="mx-2 text-gray-500">→</span>
-    <span id="notebook-path" class="text-gray-800 font-semibold">{{ notebook_name }}</span>
-    <span class="mx-2 text-gray-500">→</span>
-    <span id="file-path" class="text-blue-700 font-semibold">No File Selected</span>
-  </div>
-  <!-- MAIN LAYOUT -->
-  <div class="flex max-w-7xl mx-auto mt-6 gap-6 px-4">
-    <!-- LEFT: Sidebar -->
-    <div class="w-1/4 bg-white rounded-2xl shadow-md p-4 h-[34rem] overflow-y-auto">
-      <button class="upload-button w-full bg-blue-600 hover:bg-blue-700 text-white py-2 px-4 rounded mb-4"
-              onclick="openModal()">Upload File(s)</button>
-      <h3 class="text-lg font-semibold mb-2 text-gray-700">📂 Available Files</h3>
-      <ul id="file-list" class="space-y-3 text-gray-700 text-sm">
-        <!-- Files injected here -->
-      </ul>
-    </div>
-    <!-- RIGHT: Chat Panel -->
-    <div class="w-3/4 bg-white rounded-2xl shadow-md p-6 flex flex-col">
-      <div id="chat-box" class="h-80 overflow-y-scroll border rounded-lg p-4 bg-gray-50 space-y-3 mb-4">
-        <!-- Chat messages will appear here -->
-      </div>
-      <div class="flex flex-col sm:flex-row items-stretch sm:items-center gap-2 mt-auto">
-        <input id="persona" type="text" placeholder="Enter persona..." value="You are a helpful assistant"
-               class="px-4 py-2 border rounded-lg focus:outline-none focus:ring-2 focus:ring-purple-500" />
-        <input id="message" type="text" placeholder="Type your message..."
-               class="flex-grow px-4 py-2 border rounded-lg focus:outline-none focus:ring-2 focus:ring-blue-500" />
-        <button onclick="sendMessage()"
-                class="bg-blue-600 hover:bg-blue-700 text-white px-4 py-2 rounded-lg transition">
-          Send
-        </button>
-        <select id="summary-type"
-                class="px-3 py-2 border rounded-lg bg-white focus:outline-none focus:ring-2 focus:ring-green-500">
-          <option value="brief">Brief</option>
-          <option value="detailed">Detailed</option>
-        </select>
-        <button onclick="summarize()"
-                class="bg-green-600 hover:bg-green-700 text-white px-4 py-2 rounded-lg transition">
-          Summarize
-        </button>
-      </div>
-    </div>
-  </div>
-  <!-- UPLOAD MODAL -->
-  <div id="uploadModal" class="modal fixed z-50 inset-0 hidden bg-black bg-opacity-50 flex justify-center items-center">
-    <div class="modal-content bg-white p-6 rounded-xl w-[90%] max-w-md relative">
-      <span class="close absolute top-3 right-4 text-xl cursor-pointer" onclick="closeModal()">&times;</span>
-      <h2 class="text-xl font-bold text-gray-800">NotebookLM</h2>
-      <p class="text-sm text-gray-600 mt-1">Add sources (PDF only)</p>
-      <div id="dropZone"
-           class="drop-zone mt-4 border-2 border-dashed border-gray-300 p-6 rounded-lg text-gray-500 text-sm text-center cursor-pointer hover:border-blue-500 hover:text-blue-500"
-           onclick="triggerFileInput()">
-        Drag &amp; drop or <u>choose files</u><br>
-        <small>Supported: PDF</small>
-      </div>
-      <input type="file" id="fileInput" accept=".pdf" multiple class="hidden" />
-      <div id="selectedFilesList" class="mt-4 text-gray-700 text-sm space-y-1 max-h-40 overflow-y-auto">
-        <!-- File names shown here -->
-      </div>
-      <div class="flex justify-center mt-4">
-        <button class="submit-btn bg-blue-600 text-white px-4 py-2 rounded" onclick="submitFile()">Upload</button>
-      </div>
-    </div>
-  </div>
-  <!-- FEEDBACK MODAL (Bootstrap) -->
-  <div class="modal fade" id="feedbackModal" tabindex="-1" aria-labelledby="feedbackModalLabel" aria-hidden="true">
-    <div class="modal-dialog" role="document">
-      <div class="modal-content" style="background-color: white; color: black;">
-        <div class="modal-header">
-          <h5 class="modal-title" id="feedbackModalLabel">Rate Your Experience</h5>
-          <!-- Proper Bootstrap “close” button -->
-          <button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
-        </div>
-        <div class="modal-body">
-          <label for="feedbackLevel">Rating (optional):</label>
-          <select id="feedbackLevel" class="form-control">
-            <option value="">-- Select --</option>
-            <option value="1">Very Bad</option>
-            <option value="2">Bad</option>
-            <option value="3">Neutral</option>
-            <option value="4">Good</option>
-            <option value="5">Very Good</option>
-          </select>
-          <label for="comment" class="mt-2">Comments (optional):</label>
-          <textarea id="comment" class="form-control" rows="3" placeholder="Write your feedback here..."></textarea>
-        </div>
-        <div class="modal-footer">
-          <button id="submitFeedback" class="btn btn-primary">Submit</button>
-          <button type="button" class="btn btn-secondary" data-bs-dismiss="modal">Skip</button>
-        </div>
-      </div>
-    </div>
-  </div>
-  <!-- DELETE FILE CONFIRMATION MODAL (unchanged) -->
-  <div class="modal fade" id="deleteFileModal" tabindex="-1" aria-labelledby="deleteFileModalLabel" aria-hidden="true">
-    <div class="modal-dialog">
-      <div class="modal-content">
-        <div class="modal-header">
-          <h5 class="modal-title" id="deleteFileModalLabel">Confirm Delete</h5>
-          <button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
-        </div>
-        <div class="modal-body">
-          Are you sure you want to delete <strong id="deleteFileName"></strong>?
-        </div>
-        <div class="modal-footer">
-          <button type="button" class="btn btn-secondary" data-bs-dismiss="modal">Cancel</button>
-          <button type="button" class="btn btn-danger" onclick="confirmDeleteFile()">Delete</button>
-        </div>
-      </div>
-    </div>
-  </div>
-  <!-- SCRIPTS -->
-  <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.2/dist/js/bootstrap.bundle.min.js"></script>
-  <script>
-    document.addEventListener("DOMContentLoaded", function () {
-      // Cache references
-      const backButton = document.getElementById("backBtn");
-      const feedbackModalEl = document.getElementById("feedbackModal");
-      const bootstrapFeedbackModal = new bootstrap.Modal(feedbackModalEl);
-      // Open feedback modal when “Back” is clicked
-      backButton.addEventListener("click", function (event) {
-        event.preventDefault();
-        bootstrapFeedbackModal.show();
-      });
-      // When user clicks "Submit" inside feedback modal, navigate back
-      document.getElementById("submitFeedback").addEventListener("click", function () {
-        // You can grab rating/comment if needed:
-        const rating = document.getElementById("feedbackLevel").value || "No rating";
-        const comment = document.getElementById("comment").value || "";
-        console.log("Feedback submitted:", { rating, comment });
-        // Hide the modal first
-        bootstrapFeedbackModal.hide();
-        // Then navigate back to the notebook page:
-        window.location.href = `/notebook/?effort_id=${effortId}&user_id=${userId}`;
-      });
-    });
-  </script>
-  <script>
-    // Your existing chat/upload/delete code starts here:
-    const userId = "{{ user_id }}";
-    const effortId = "{{ effort_id }}";
-    const notebookId = "{{ notebook_id }}";
-    const effortName = "{{ effort_name }}";
-    const notebookName = "{{ notebook_name }}";
-    const preloadedFiles = {{ files | tojson }};
-    const preloadedZones = {{ zones | tojson }};
-    const chatHistory = {{ chat_history | tojson }};
-    const chatBox = document.getElementById("chat-box");
-    const zoneDropdown = document.getElementById("zone-dropdown");
-    const fileList = document.getElementById("file-list");
-    const modal = document.getElementById("uploadModal");
-    const fileInput = document.getElementById("fileInput");
-    fileInput.addEventListener("change", () => {
-      const list = document.getElementById("selectedFilesList");
-      list.innerHTML = "";
-      const files = fileInput.files;
-      if (!files.length) {
-        list.textContent = "No files selected.";
-        return;
-      }
-      for (let file of files) {
-        const item = document.createElement("div");
-        item.className = "flex items-center gap-2";
-        item.innerHTML = `📄 <span class="truncate">${file.name}</span>`;
-        list.appendChild(item);
-      }
-    });
-    let currentZoneId = null;
-    let currentFileUri = "";
-    let selectedFileElement = null;
-    function appendMessage(sender, message, color, alignRight = false) {
-      const wrapper = document.createElement("div");
-      wrapper.className = alignRight ? "flex justify-end" : "flex justify-start";
-      const bubble = document.createElement("div");
-      bubble.className = `max-w-xl px-4 py-2 rounded-lg ${color} text-white`;
-      bubble.innerHTML = `<strong>${sender}:</strong> ${message}`;
-      wrapper.appendChild(bubble);
-      chatBox.appendChild(wrapper);
-      chatBox.scrollTop = chatBox.scrollHeight;
-    }
-    function loadZones() {
-      zoneDropdown.innerHTML = "";
-      const defaultOpt = document.createElement("option");
-      defaultOpt.disabled = true;
-      defaultOpt.selected = true;
-      defaultOpt.textContent = "Select a Configuration";
-      zoneDropdown.appendChild(defaultOpt);
-      try {
-        preloadedZones.forEach(zone => {
-          const opt = document.createElement("option");
-          opt.value = zone.id;
-          opt.textContent = zone.name;
-          zoneDropdown.appendChild(opt);
-        });
-        const divider = document.createElement("option");
-        divider.disabled = true;
-        divider.textContent = "──────────";
-        zoneDropdown.appendChild(divider);
-      } catch (err) {
-        console.error("Error loading zones:", err);
-      }
-      const manage = document.createElement("option");
-      manage.value = "manage";
-      manage.textContent = "➕ Manage Configurations";
-      zoneDropdown.appendChild(manage);
-      zoneDropdown.addEventListener("change", (e) => {
-        if (e.target.value === "manage") {
-          const form = document.createElement("form");
-          form.method = "POST";
-          form.action = "/chat_feature/manage_zones";
-          const userInput = document.createElement("input");
-          userInput.type = "hidden";
-          userInput.name = "user_id";
-          userInput.value = userId;
-          form.appendChild(userInput);
-          const effortInput = document.createElement("input");
-          effortInput.type = "hidden";
-          effortInput.name = "effort_id";
-          effortInput.value = effortId;
-          form.appendChild(effortInput);
-          const notebookInput = document.createElement("input");
-          notebookInput.type = "hidden";
-          notebookInput.name = "notebook_id";
-          notebookInput.value = notebookId;
-          form.appendChild(notebookInput);
-          document.body.appendChild(form);
-          form.submit();
-        } else {
-          currentZoneId = parseInt(e.target.value);
-        }
-      });
-    }
-    function loadFiles() {
-      const files = preloadedFiles || [];
-      fileList.innerHTML = "";
-      files.forEach(file => {
-        const li = document.createElement("li");
-        li.className =
-          "flex items-center gap-3 p-4 rounded-2xl border border-gray-200 bg-white hover:bg-blue-50 hover:shadow-lg transition shadow group";
-        const fileIcon = document.createElement("div");
-        fileIcon.className = "flex-shrink-0 text-blue-600 text-lg";
-        fileIcon.textContent = "📄";
-        const fileName = document.createElement("div");
-        fileName.className = "flex-1 min-w-0 cursor-pointer";
-        fileName.innerHTML = `
-          <p class="font-medium truncate group-hover:whitespace-normal group-hover:break-all" title="${file.name}">
-            ${file.name}
-          </p>
-        `;
-        fileName.onclick = () => {
-          currentFileUri = file.uri;
-          appendMessage("System", `Selected file: ${file.name}`, "bg-purple-600");
-          document.getElementById("notebook-path").textContent = notebookName;
-          document.getElementById("effort-path").textContent = effortName;
-          document.getElementById("file-path").textContent = file.name;
-          if (selectedFileElement) {
-            selectedFileElement.classList.remove(
-              "border-blue-500", "bg-blue-100", "ring", "ring-blue-300", "shadow-lg", "transform", "scale-105"
-            );
-          }
-          li.classList.add("border-blue-500", "bg-blue-100", "ring", "ring-blue-300", "shadow-lg", "transform", "scale-105");
-          selectedFileElement = li;
-        };
-        const deleteButton = document.createElement("button");
-        deleteButton.className = "btn btn-sm btn-outline-danger";
-        deleteButton.setAttribute("data-bs-toggle", "modal");
-        deleteButton.setAttribute("data-bs-target", "#deleteFileModal");
-        deleteButton.innerHTML = `<i class="bi bi-trash"></i>`;
-        deleteButton.onclick = (e) => {
-          e.stopPropagation();
-          setDeleteFile(file.uri, file.name);
-        };
-        li.appendChild(fileIcon);
-        li.appendChild(fileName);
-        li.appendChild(deleteButton);
-        fileList.appendChild(li);
-      });
-    }
-    async function sendMessage() {
-      const message = document.getElementById("message").value;
-      const persona = document.getElementById("persona").value;
-      if (!message.trim() || !currentFileUri) {
-        alert("Please select a file and type a message.");
-        return;
-      }
-      if (!currentZoneId) {
-        alert("Please select a zone before sending a message.");
-        return;
-      }
-      appendMessage("You", message, "bg-blue-500", true); // Right aligned
-      const loadingId = "loading-msg";
-      const loadingWrapper = document.createElement("div");
-      loadingWrapper.id = loadingId;
-      loadingWrapper.className = "flex justify-start";
-      const loadingBubble = document.createElement("div");
-      loadingBubble.className = "max-w-xl px-4 py-2 rounded-lg bg-gray-300 text-gray-800";
-      loadingBubble.innerHTML = `<strong>Assistant:</strong> <span class="loading-dots"><span></span><span></span><span></span></span>`;
-      loadingWrapper.appendChild(loadingBubble);
-      chatBox.appendChild(loadingWrapper);
-      chatBox.scrollTop = chatBox.scrollHeight;
-      const res = await fetch("/chat_feature/send_message", {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: JSON.stringify({
-          message,
-          persona,
-          file_uri: currentFileUri,
-          experiment_zone_id: currentZoneId,
-          effortId: effortId,
-          notebookId: notebookId
-        })
-      });
-      const data = await res.json();
-      document.getElementById(loadingId)?.remove();
-      appendMessage("Assistant", data.message || "No response", "bg-gray-700");
-      document.getElementById("message").value = "";
-    }
-    async function summarize() {
-      const summaryType = document.getElementById("summary-type").value;
-      if (!currentFileUri) {
-        alert("Please select a file and summarize.");
-        return;
-      }
-      if (!currentZoneId) {
-        alert("Please select a zone before summarizing.");
-        return;
-      }
-      appendMessage("You", `Summarize (${summaryType})`, "bg-green-500", true); // Right aligned
-      const loadingId = "loading-msg";
-      const loadingWrapper = document.createElement("div");
-      loadingWrapper.id = loadingId;
-      loadingWrapper.className = "flex justify-start";
-      const loadingBubble = document.createElement("div");
-      loadingBubble.className = "max-w-xl px-4 py-2 rounded-lg bg-gray-300 text-gray-800";
-      loadingBubble.innerHTML = `<strong>Assistant:</strong> <span class="loading-dots"><span></span><span></span><span></span></span>`;
-      loadingWrapper.appendChild(loadingBubble);
-      chatBox.appendChild(loadingWrapper);
-      chatBox.scrollTop = chatBox.scrollHeight;
-      const res = await fetch("/chat_feature/summarize", {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: JSON.stringify({
-          summary_type: summaryType,
-          file_uri: currentFileUri,
-          experiment_zone_id: currentZoneId,
-          effortId: effortId,
-          notebookId: notebookId
-        })
-      });
-      const data = await res.json();
-      document.getElementById(loadingId)?.remove();
-      appendMessage("Summary", data.message || "No summary", "bg-gray-700");
-    }
-    function openModal() { modal.classList.remove("hidden"); }
-    function closeModal() {
-      modal.classList.add("hidden");
-      fileInput.value = "";
-    }
-    function triggerFileInput() { fileInput.click(); }
-    async function submitFile() {
-      const files = fileInput.files;
-      if (!files.length) return alert("Choose at least one file.");
-      const formData = new FormData();
-      for (let file of files) {
-        if (file.type !== "application/pdf") {
-          alert(`Invalid file type: ${file.name}`);
-          return;
-        }
-        formData.append("files", file);
-      }
-      formData.append("effort_id", effortId);
-      try {
-        const res = await fetch("/chat_feature/upload_file", {
-          method: "POST",
-          body: formData
-        });
-        const contentType = res.headers.get("content-type") || "";
-        let responseData = contentType.includes("application/json") ? await res.json() : await res.text();
-        if (!res.ok) {
-          const errorMessage = typeof responseData === "string"
-            ? responseData
-            : (responseData.error || JSON.stringify(responseData));
-          alert(`Upload failed: ${errorMessage}`);
-          return;
-        }
-        alert("Upload successful!");
-        location.reload();  // Refresh to get updated file list from server
-      } catch (err) {
-        alert("Error: " + err.message);
-      }
-    }
-    window.onclick = function (event) {
-      if (event.target == modal) closeModal();
-    };
-    let deleteFileUriToConfirm = "";
-    let deleteFileName = "";
-    function setDeleteFile(fileUri, fileName) {
-      deleteFileUriToConfirm = fileUri;
-      deleteFileName = fileName;
-      document.getElementById("deleteFileName").textContent = fileName;
-    }
-    async function confirmDeleteFile() {
-      try {
-        const res = await fetch("/chat_feature/delete_file", {
-          method: "POST",
-          headers: { "Content-Type": "application/json" },
-          body: JSON.stringify({
-            file_uri: deleteFileUriToConfirm,
-            effort_id: effortId,
-            filename: deleteFileName
-          })
-        });
-        if (!res.ok) {
-          const text = await res.text();
-          alert("Failed to delete: " + text);
-        } else {
-          const modal = bootstrap.Modal.getInstance(document.getElementById("deleteFileModal"));
-          modal.hide();
-          alert("File deleted.");
-          location.reload();
-        }
-      } catch (err) {
-        alert("Error deleting file: " + err.message);
-      }
-    }
-    const dropZone = document.querySelector(".drop-zone");
-    dropZone.addEventListener("dragover", (e) => {
-      e.preventDefault();
-      dropZone.classList.add("border-blue-500", "text-blue-500");
-    });
-    dropZone.addEventListener("dragleave", () => {
-      dropZone.classList.remove("border-blue-500", "text-blue-500");
-    });
-    dropZone.addEventListener("drop", (e) => {
-      e.preventDefault();
-      dropZone.classList.remove("border-blue-500", "text-blue-500");
-      const files = e.dataTransfer.files;
-      fileInput.files = files;
-      const list = document.getElementById("selectedFilesList");
-      list.innerHTML = "";
-      for (const file of files) {
-        const p = document.createElement("p");
-        p.textContent = file.name;
-        list.appendChild(p);
-      }
-    });
-    window.onload = function () {
-      loadFiles();
-      loadZones();
-      let historyFilename = "";
-      if (chatHistory && chatHistory.length > 0) {
-        chatHistory.forEach(entry => {
-          if (entry.filename && (entry.filename !== historyFilename || historyFilename === "")) {
-            historyFilename = entry.filename;
-            appendMessage("System", `Selected file: ${historyFilename}`, "bg-purple-600");
-          }
-          const userMessage = entry.question;
-          appendMessage("You", userMessage, "bg-blue-500", true);
-          const personaMessage = entry.persona;
-          appendMessage("Persona", personaMessage, "bg-blue-500", true);
-          const botMessage = entry.response;
-          appendMessage("Assistant", botMessage, "bg-gray-700", false);
-        });
-      }
-    };
-  </script>
-</body>
-</html>
------------------------------------------------------------
-from flask import Flask, render_template, request, jsonify, Blueprint, redirect, url_for, make_response
-import requests
-from authentication import login_required
-#app = Flask(__name__)
-chat_feature_bp = Blueprint("chat_feature", __name__, template_folder="templates")
-API_URL = "http://localhost:8047/chat/chat_with_document"
-FILE_URI = "C:/Users/Jitmanyu/Desktop/Backend/file_store\\Factors_Influencing_the_Likelihood_of_Cu 1.pdf"
-EXPERIMENT_ZONE_ID = 1
-NOTEBOOK_ID = "1"
-EFFORT_ID = 1
-@chat_feature_bp.route("/")
-@login_required
-def home():
-    notebook_id = request.args.get("notebook_id")
-    effort_id = request.args.get("effort_id")
-    user_id = request.args.get("user_id")
-    if not notebook_id or not effort_id:
-        return "Notebook ID and effort ID are required", 400
-    files_in_effort = []
-    ezones = []
-    notebook_name = None
-    effort_name = None
-    notebook_chat = []
-    try:
-        # Fetch user details
-        response = requests.get(
-            "http://localhost:8047/user/get_user_details/",
-            params={"User_ID": user_id},
-            headers={"accept": "application/json"}
-        )
-        response.raise_for_status()
-        data = response.json()
-        # Parse files and ezones
-        all_files_by_effort = data.get("files", {})
-        effort_files = all_files_by_effort.get(str(effort_id), {})
-        if effort_files:
-            for name, path in effort_files.items():
-                files_in_effort.append({"name": name, "uri": path})
-        ezones = data.get("ezones", {}).get(str(effort_id), [])
-        # Parse notebook name and effort name
-        notebook_name = next(
-            (notebook['name'] for notebook in data['notebooks'].get(str(effort_id), []) if notebook['id'] == int(notebook_id)),
-            None
-        )
-        effort_name = next(
-            (effort["name"] for effort in data.get("efforts", []) if str(effort["id"]) == str(effort_id)),
-            None
-        )
-        # Fetch notebook details using new API
-        notebook_response = requests.get(
-            f"http://localhost:8047/user/get_notebook_details/{notebook_id}",
-            headers={"accept": "application/json"}
-        )
-        notebook_response.raise_for_status()
-        notebook_data = notebook_response.json()
-        # Extract chat from notebook details
-        notebook_chat = notebook_data.get("chat", {}).get("message", [])
-    except requests.RequestException as e:
-        print("Error during API calls:", e)
-        return "Failed to load data", 500
-    return render_template(
-        "chat.html",
-        user_id=user_id,
-        notebook_id=notebook_id,
-        effort_id=effort_id,
-        effort_name=effort_name,
-        notebook_name=notebook_name,
-        files=files_in_effort,
-        zones=ezones,
-        chat_history=notebook_chat  # Pass chat messages to template
-    )
-FASTAPI_URL = "http://localhost:8047"
-@chat_feature_bp.route('/upload_file', methods=['POST'])
-@login_required
-def upload_file():
-    effort_id = request.form.get("effort_id")
-    uploaded_files = request.files.getlist('files')
-    if not uploaded_files:
-        return jsonify({'error': 'No files uploaded'}), 400
-    data = {
-        "effort_id": effort_id,
-    }
-    results = []
-    for uploaded_file in uploaded_files:
-        files = {'file': (uploaded_file.filename, uploaded_file.stream, uploaded_file.mimetype)}
-        response = requests.post('http://localhost:8047/doc/upload/', files=files, data=data)
-        if response.ok:
-            results.append({'filename': uploaded_file.filename, 'status': 'success'})
-        else:
-            results.append({'filename': uploaded_file.filename, 'status': 'error', 'details': response.text})
-    return jsonify({'results': results}), 200
-@chat_feature_bp.route('/delete_file', methods=['POST'])
-@login_required
-def delete_file():
-    data = request.json
-    payload = {
-        "effort_id": data["effort_id"],
-        "filename": data["filename"]
-    }
-    response = requests.delete("http://localhost:8047/doc/delete/", data=payload)
-    if response.status_code != 200:
-        return make_response(jsonify({
-            "message": f"Delete failed: {response.text}"
-        }), response.status_code)
-    return jsonify(response.json())
-@chat_feature_bp.route("/manage_zones", methods=["POST"])
-@login_required
-def manage_zones():
-    effort_id = request.form.get("effort_id")  # Get from POST form data
-    notebook_id = request.form.get("notebook_id")  # Get from POST form data
-    user_id = request.form.get("user_id")
-    return redirect(url_for("zone_page.home", user_id = user_id, effort_id=effort_id, notebook_id=notebook_id))
-@chat_feature_bp.route("/send_message", methods=["POST"])
-@login_required
-def send_message():
-    data = request.json
-    payload = {
-        "file_uri": data["file_uri"],
-        "experiment_zone_id": data["experiment_zone_id"],
-        "query_prompt": data["message"],
-        "persona": data["persona"],
-        "RAG": "Y",
-        "summary_prompt": "",  # unused in chat
-        "notebook_id": data["notebookId"],
-        "effort_id": data["effortId"]
-    }
-    response = requests.post(API_URL, json=payload)
-    print("API call is being made")
-    return jsonify(response.json())
-@chat_feature_bp.route("/summarize", methods=["POST"])
-@login_required
-def summarize():
-    data = request.json
-    payload = {
-        "file_uri": data["file_uri"],
-        "experiment_zone_id": data["experiment_zone_id"],
-        "query_prompt": "",
-        "persona": "You are a helpful assistant",
-        "RAG": "N",
-        "summary_prompt": data["summary_type"],  # unused in summarization
-        "notebook_id": data["notebookId"],
-        "effort_id": data["effortId"]
-    }
-    response = requests.post(API_URL, json=payload)
-    print("API call is being made")
-    return jsonify(response.json())

+import os
+import psycopg2
+import psycopg2.extras
+import json
+import re
+from logging import getLogger
+from fastapi import HTTPException
+from dotenv import load_dotenv
+from pydantic import BaseModel
+from typing import List
+from logging_log.logger import logger
+# Load environment variables from the .env file
+load_dotenv()
+#logger = getLogger(__name__)
+# Pydantic model
+class UserCreate(BaseModel):
+    user_id: str
+    password: str
+    ad_groups: List[str]
+    is_admin: bool = False
+# Database connection function
+def get_db_connection():
+    try:
+        # Create the connection
+        conn = psycopg2.connect(
+            host=os.getenv("HOSTNAME"),
+            database=os.getenv("DATABASE"),
+            user=os.getenv("DB_USERNAME"),
+            password=os.getenv("DB_PASSWORD"),
+            port=os.getenv("PORT")
+        )
+        logger.info("Database connection established.")
+        return conn
+    except Exception as e:
+        logger.error(f"Error connecting to database: {str(e)}")
+        raise e
+# Validate user ID
+def validate_user_id(user_id: str):
+    if not re.match(r'^[a-zA-Z0-9]{5,}$', user_id):
+        raise HTTPException(status_code=400, detail="Invalid user ID format")
+    if not user_id:
+        raise HTTPException(status_code=400, detail="User ID cannot be empty")
+# Validate notebook ID
+def validate_notebook_id(notebook_id: str):
+    if not notebook_id.isdigit():
+        raise HTTPException(status_code=406, detail="Invalid notebook ID format")
+# Create a new user in the DB
+def create_user(user_id: str, password: str, ad_groups: List[str], is_admin: bool):
+    insert_query = """
+            INSERT INTO users (user_id, password, ad_groups, is_admin)
+            VALUES (%s, %s, %s, %s)
+        """
+    try:
+        with get_db_connection() as conn:
+            with conn.cursor() as curr:
+                curr.execute(insert_query, (user_id, password, ad_groups, is_admin))
+            conn.commit()
+            logger.info(f"User {user_id} created")
+    except Exception as e:
+        conn.rollback()
+        logger.error(f"Error creating new user {user_id}")
+# Update an existing user in the DB
+def update_user(user_id: str, password: str, ad_groups: List[str], is_admin: bool):
+    update_query = """
+        UPDATE users
+        SET password = %s,
+            ad_groups = %s,
+            is_admin = %s
+        WHERE user_id = %s
+    """
+    try:
+        with get_db_connection() as conn:
+            with conn.cursor() as curr:
+                curr.execute(update_query, (password, ad_groups, is_admin, user_id))
+            conn.commit()
+            logger.info(f"User {user_id} updated")
+    except Exception as e:
+        conn.rollback()
+        logger.error(f"Error updating user {user_id}: {str(e)}")
+        raise
+# delete a user in the DB
+def delete_user(user_id: str):
+    delete_query = """
+        DELETE FROM users WHERE user_id = %s;
+    """
+    try:
+        with get_db_connection() as conn:
+            with conn.cursor() as curr:
+                curr.execute(delete_query, (user_id, ))
+            conn.commit()
+            logger.info(f"User {user_id} deleted")
+    except Exception as e:
+        conn.rollback()
+        logger.error(f"Error deleting user {user_id}: {str(e)}")
+        raise
+def get_user_credentials(user_id: str):
+    select_query = """
+        SELECT * FROM users WHERE user_id = %s;
+    """
+    try:
+        with get_db_connection() as conn:
+            with conn.cursor() as curr:
+                curr.execute(select_query, (user_id,))
+                user = curr.fetchone()
+                if user:
+                    logger.info(f"User {user_id} fetched")
+                    return {
+                        "user_id": user[0],
+                        "password": user[1],
+                        "ad_groups": user[2],
+                        "is_admin": user[3]
+                    }
+                else:
+                    logger.warning(f"User {user_id} not found")
+                    return {"error": "User not found"}
+    except Exception as e:
+        logger.error(f"Error fetching user {user_id}: {str(e)}")
+        raise
+# Fetch user details
+def get_user_details(User_ID):
+    validate_user_id(User_ID)
+    conn = get_db_connection()
+    cursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
+    try:
+        cursor.execute(
+            "SELECT ad_groups, is_admin FROM users WHERE user_id=%s",
+            (User_ID, )
+        )
+        user_row = cursor.fetchone()
+        if not user_row:
+            raise HTTPException(status_code=404, detail="User not found")
+        if user_row:
+            ad_groups, is_admin = user_row
+        cursor.execute("SELECT id, name, files, permission FROM efforts")
+        efforts_data = cursor.fetchall()
+        permissions = {
+            "permissions": [],
+            "efforts": [],
+            "notebooks": {},
+            "ezones": {},
+            "files": {}
+        }
+        """
+        if ad_group in user_permissions.get("can_create_effort", []):
+            permissions["permissions"].append("can_create_effort")
+        if ad_group in user_permissions.get("can_view_effort", []):
+            permissions["permissions"].append("can_view_effort")
+        if ad_group in user_permissions.get("can_revoke_effort", []):
+            permissions["permissions"].append("can_revoke_effort")
+        if ad_group in user_permissions.get("can_modify_effort", []):
+            permissions["permissions"].append("can_modify_effort")
+        """
+        if is_admin:
+            permissions["permissions"].append("can_create_efforts")
+            permissions["permissions"].append("can_view_efforts")
+            permissions["permissions"].append("can_revoke_efforts")
+            permissions["permissions"].append("can_modify_efforts")
+            for effort in efforts_data:
+                effort_id = effort["id"]
+                effort_name = effort["name"]
+                files = effort["files"] if effort["files"] else []
+                permission_json = effort["permission"]
+                effort_permissions = []
+                if "can_view_effort" not in effort_permissions:
+                    effort_permissions.append("can_view_effort")
+                    permissions["efforts"].append({"name": effort_name, "id": effort_id})
+                    permissions["files"][effort_id] = files
+                    cursor.execute("SELECT notebook_id, notebook_name FROM notebook_details WHERE effort_id=%s", (effort_id,))
+                    notebook_details = cursor.fetchall()
+                    permissions["notebooks"][effort_id] = [{"name": note["notebook_name"], "id": note["notebook_id"]} for note in notebook_details]
+                    cursor.execute("SELECT id, name FROM zones WHERE effort_id=%s", (effort_id,))
+                    zones = cursor.fetchall()
+                    permissions["ezones"][effort_id] = [{"name": zone["name"], "id": zone["id"]} for zone in zones]
+                if "can_create_zones" not in effort_permissions:
+                    effort_permissions.append("can_create_zones")
+                if "can_modify_zones" not in effort_permissions:
+                    effort_permissions.append("can_modify_zones")
+                if "can_delete_zones" not in effort_permissions:
+                    effort_permissions.append("can_delete_zones")
+                if "can_create_notebooks" not in effort_permissions:
+                    effort_permissions.append("can_create_notebooks")
+                if "can_delete_notebooks" not in effort_permissions:
+                    effort_permissions.append("can_delete_notebooks")
+                if effort_permissions:
+                    permissions["permissions"].append({effort_id: effort_permissions})
+        else:
+            permissions["permissions"].append("can_view_efforts")
+            for effort in efforts_data:
+                effort_id = effort["id"]
+                effort_name = effort["name"]
+                files = effort["files"] if effort["files"] else []
+                permission_json = effort["permission"]
+                effort_permissions = []
+                """
+                if ad_group in permission_json.get("can_create_zones", []):
+                    effort_permissions.append("can_create_zones")
+                if ad_group in permission_json.get("can_modify_zones", []):
+                    effort_permissions.append("can_modify_zones")
+                if ad_group in permission_json.get("can_delete_zones", []):
+                    effort_permissions.append("can_delete_zones")
+                if ad_group in permission_json.get("can_create_notebooks", []):
+                    effort_permissions.append("can_create_notebooks")
+                if ad_group in permission_json.get("can_delete_notebooks", []):
+                    effort_permissions.append("can_delete_notebooks")
+                """
+                for ad_group in ad_groups:
+                    if ad_group in permission_json.get("can_view_effort", []):
+                        if "can_view_effort" not in effort_permissions:
+                            effort_permissions.append("can_view_effort")
+                            permissions["efforts"].append({"name": effort_name, "id": effort_id})
+                            permissions["files"][effort_id] = files
+                            cursor.execute("SELECT notebook_id, notebook_name FROM notebook_details WHERE effort_id=%s", (effort_id,))
+                            notebook_details = cursor.fetchall()
+                            permissions["notebooks"][effort_id] = [{"name": note["notebook_name"], "id": note["notebook_id"]} for note in notebook_details]
+                            cursor.execute("SELECT id, name FROM zones WHERE effort_id=%s", (effort_id,))
+                            zones = cursor.fetchall()
+                            permissions["ezones"][effort_id] = [{"name": zone["name"], "id": zone["id"]} for zone in zones]
+                    if ad_group in permission_json.get("can_create_and_modify_zones", []):
+                        if "can_create_zones" not in effort_permissions:
+                            effort_permissions.append("can_create_zones")
+                        if "can_modify_zones" not in effort_permissions:
+                            effort_permissions.append("can_modify_zones")
+                        if "can_delete_zones" not in effort_permissions:
+                            effort_permissions.append("can_delete_zones")
+                    if ad_group in permission_json.get("can_create_and_modify_notebooks", []):
+                        if "can_create_notebooks" not in effort_permissions:
+                            effort_permissions.append("can_create_notebooks")
+                        if "can_delete_notebooks" not in effort_permissions:
+                            effort_permissions.append("can_delete_notebooks")
+                if effort_permissions:
+                    permissions["permissions"].append({effort_id: effort_permissions})
+        return {
+            "permissions": permissions["permissions"],
+            "efforts": permissions["efforts"],
+            "notebooks": permissions["notebooks"],
+            "ezones": permissions["ezones"],
+            "files": permissions["files"]
+        }
+    except Exception as e:
+        logger.error(f"Error fetching user details: {str(e)}")
+        raise e
+    finally:
+        cursor.close()
+        conn.close()
+# Fetch notebook details
+def get_notebook_details(notebook_id):
+    validate_notebook_id(notebook_id)
+    conn = get_db_connection()
+    cursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
+    try:
+        cursor.execute(
+            "SELECT notebook_id, notebook_name, chat_history, created_at, effort_id FROM notebook_details WHERE notebook_id=%s",
+            (notebook_id,)
+        )
+        notebook = cursor.fetchone()
+        if not notebook:
+            raise HTTPException(status_code=404, detail="Notebook not found")
+        cursor.execute("SELECT name FROM efforts WHERE id=%s", (notebook["effort_id"],))
+        effort = cursor.fetchone()
+        return {
+            "id": notebook["notebook_id"],
+            "name": notebook["notebook_name"],
+            "chat": notebook["chat_history"],
+            "Created at": notebook["created_at"],
+            "effort_name": effort["name"] if effort else None
+        }
+    except Exception as e:
+        logger.error(f"Error fetching notebook details: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error fetching notebook details: {str(e)}")
+    finally:
+        cursor.close()
+        conn.close()
+# Effort Management
+# Create effort
+def create_effort(name, permission):
+    conn = None
+    cursor = None
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute(
+            "INSERT INTO efforts (name, permission) VALUES (%s, %s)",
+            (name, psycopg2.extras.Json(permission))
+        )
+        conn.commit()
+        logger.info(f"Effort '{name}' created.")
+    except Exception as e:
+        if conn:
+            conn.rollback()
+        logger.error(f"Error creating effort '{name}': {str(e)}")
+        raise e
+    finally:
+        if cursor:
+            cursor.close()
+        if conn:
+            conn.close()
+# Delete effort
+def delete_effort(effort_id):
+    conn = None
+    cursor = None
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute("DELETE FROM efforts WHERE id = %s", (effort_id,))
+        conn.commit()
+        if cursor.rowcount == 0:
+            raise ValueError(f"Effort with ID {effort_id} not found")
+        logger.info(f"Effort with ID {effort_id} deleted successfully.")
+    except Exception as e:
+        if conn:
+            conn.rollback()
+        logger.error(f"Error deleting effort with ID {effort_id}: {str(e)}")
+        raise e
+    finally:
+        if cursor:
+            cursor.close()
+        if conn:
+            conn.close()
+# Modify effort
+def modify_effort(effort_id, name, permission):
+    conn = None
+    cursor = None
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute(
+            "UPDATE efforts SET name = %s, permission = %s WHERE id = %s",
+            (name, psycopg2.extras.Json(permission), effort_id)
+        )
+        conn.commit()
+        if cursor.rowcount == 0:
+            raise ValueError(f"Effort with ID {effort_id} not found")
+        logger.info(f"Effort with ID {effort_id} updated successfully.")
+    except Exception as e:
+        if conn:
+            conn.rollback()
+        logger.error(f"Error modifying effort with ID {effort_id}: {str(e)}")
+        raise e
+    finally:
+        if cursor:
+            cursor.close()
+        if conn:
+            conn.close()
+# List effort details
+def list_effort_details(effort_id):
+    conn = None
+    cursor = None
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
+        cursor.execute("SELECT * FROM efforts WHERE id = %s", (effort_id,))
+        effort_details = cursor.fetchone()
+        if effort_details:
+            logger.info(f"Effort details fetched for ID {effort_id}")
+            return dict(effort_details)
+        else:
+            raise ValueError(f"Effort with ID {effort_id} not found")
+    except Exception as e:
+        logger.error(f"Error fetching effort details for ID {effort_id}: {str(e)}")
+        raise e
+    finally:
+        if cursor:
+            cursor.close()
+        if conn:
+            conn.close()
+# Zone Management
+# Delete experiment zone
+def delete_experiment_zone(zone_id: int):
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute("DELETE FROM zones WHERE id = %s", (zone_id,))
+        conn.commit()
+        if cursor.rowcount == 0:
+            raise ValueError(f"Zone with ID {zone_id} not found")
+        logger.info(f"Experiment zone with ID {zone_id} deleted successfully.")
+    except Exception as e:
+        logger.error(f"Error while deleting experiment zone: {str(e)}")
+        raise e
+    finally:
+        cursor.close()
+        conn.close()
+# Fetch zone details
+def get_zone_details(zone_id: int):
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
+        cursor.execute("SELECT id, name, config, created_at FROM zones WHERE id = %s", (zone_id,))
+        zone = cursor.fetchone()
+        if not zone:
+            logger.warning(f"No experiment zone found with ID {zone_id}.")
+            raise ValueError(f"Zone with ID {zone_id} not found")
+        logger.info(f"Details for experiment zone ID {zone_id} fetched successfully.")
+        return zone
+    except Exception as e:
+        logger.error(f"Error while fetching zone details: {str(e)}")
+        raise e
+    finally:
+        cursor.close()
+        conn.close()
+# Create experiment zone
+def create_experiment_zone(effort_id: int, name: str, vector_db: str, llm_model: str, chunking_strategy: str, embedding_model: str, indexer: str, top_k: int, overlap: int, chunk_size: int):
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        if not name:
+            raise HTTPException(status_code=400, detail="Zone name cannot be empty")
+        config = {
+            "llm_model": llm_model,
+            "embedding_model": embedding_model,
+            "vector_db": vector_db,
+            "chunking_strategy": chunking_strategy,
+            "indexer": indexer,
+            "top_k": top_k,
+            "overlap": overlap,
+            "chunk_size": chunk_size
+        }
+        cursor.execute(
+            "INSERT INTO zones (effort_id, name, config) VALUES (%s, %s, %s) RETURNING id",
+            (effort_id, name, json.dumps(config))
+        )
+        zone_id = cursor.fetchone()[0]
+        conn.commit()
+        logger.info(f"Experiment zone '{name}' created with ID {zone_id}.")
+        return {"zone_id": zone_id}
+    except Exception as e:
+        logger.error(f"Error while creating experiment zone: {str(e)}")
+        raise e
+    finally:
+        cursor.close()
+        conn.close()
+# Modify experiment zone
+def modify_experiment_zone(zone_id: int, name: str, vector_db: str, llm_model: str, chunking_strategy: str, embedding_model: str, indexer: str, top_k: int, overlap: int, chunk_size: int):
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        if not name:
+            raise HTTPException(status_code=400, detail="Zone name cannot be empty")
+        cursor.execute("SELECT effort_id FROM zones WHERE id = %s", (zone_id,))
+        if cursor.rowcount == 0:
+            raise ValueError(f"Experiment zone with ID {zone_id} not found")
+        effort_id = cursor.fetchone()[0]
+        config = {
+            "llm_model": llm_model,
+            "embedding_model": embedding_model,
+            "vector_db": vector_db,
+            "chunking_strategy": chunking_strategy,
+            "indexer": indexer,
+            "top_k": top_k,
+            "overlap": overlap,
+            "chunk_size": chunk_size
+        }
+        config_json = json.dumps(config)
+        cursor.execute(
+            "INSERT INTO zones (effort_id, name, config) VALUES (%s, %s, %s) RETURNING id",
+            (effort_id, name, config_json)
+        )
+        new_zone_id = cursor.fetchone()[0]
+        cursor.execute("""
+            DELETE FROM zones
+            WHERE id = %s
+        """, (zone_id,))
+        cursor.execute("""
+            DELETE FROM chunk_embeddings
+            WHERE zone_id = %s
+        """, (zone_id,))
+        cursor.execute("""
+            DELETE FROM file_chunks
+            WHERE zone_id = %s
+        """, (zone_id,))
+        conn.commit()
+        logger.info(f"Experiment zone with ID {zone_id} modified successfully. New zone ID is {new_zone_id}")
+        return new_zone_id
+    except Exception as e:
+        logger.error(f"Error while modifying experiment zone: {str(e)}")
+        raise e
+    finally:
+        cursor.close()
+        conn.close()