Update script.js

script.js

@@ -1,27 +1,107 @@
 document.addEventListener("DOMContentLoaded", () => {
     const workerUrl = "https://ctmresearchagent.aiagents.workers.dev";
+    
+    // Chat elements
     const submitButton = document.getElementById("submitButton");
     const researchQueryInput = document.getElementById("researchQuery");
     const resultDisplay = document.getElementById("resultDisplay");
     const historyContainer = document.getElementById("historyContainer");
 
+    // Document management elements
+    const uploadForm = document.getElementById("uploadForm");
+    const fileInput = document.getElementById("fileInput");
+    const uploadStatus = document.getElementById("uploadStatus");
+    const reportsList = document.getElementById("reportsList");
+
+    /**
+     * Handles file upload
+     */
+    async function handleFileUpload(event) {
+        event.preventDefault();
+        const file = fileInput.files[0];
+        if (!file) {
+            uploadStatus.textContent = 'Please select a file to upload.';
+            uploadStatus.style.color = 'red';
+            return;
+        }
+
+        uploadStatus.textContent = `Uploading ${file.name}...`;
+        uploadStatus.style.color = 'blue';
+
+        try {
+            const formData = new FormData();
+            formData.append('file', file);
+            
+            // Note: We're sending FormData, not JSON
+            const response = await fetch(`${workerUrl}/api/upload`, {
+                method: 'POST',
+                body: formData,
+            });
+
+            const data = await response.json();
+
+            if (!response.ok) {
+                throw new Error(data.error || 'Upload failed');
+            }
+            
+            uploadStatus.textContent = `Successfully uploaded ${file.name}!`;
+            uploadStatus.style.color = 'green';
+            uploadForm.reset();
+            await loadReportsList(); // Refresh the list of reports
+
+        } catch (error) {
+            console.error("Upload error:", error);
+            uploadStatus.textContent = `Error: ${error.message}`;
+            uploadStatus.style.color = 'red';
+        }
+    }
+
     /**
-     * --- UPDATED: Submits query to the AI backend ---
+     * Loads the list of available research reports
      */
-    async function handleSubmit() {
+    async function loadReportsList() {
+        try {
+            const response = await fetch(`${workerUrl}/api/documents`);
+            if (!response.ok) throw new Error('Failed to fetch reports list.');
+
+            const documents = await response.json();
+            reportsList.innerHTML = ""; // Clear existing list
+
+            if (documents.length === 0) {
+                reportsList.innerHTML = "<p>No reports uploaded yet.</p>";
+                return;
+            }
+
+            documents.forEach(doc => {
+                const entryDiv = document.createElement("div");
+                entryDiv.className = "report-entry";
+                entryDiv.innerHTML = `
+                    <span>${doc.filename}</span>
+                    <span class="status ${doc.status.toLowerCase()}">${doc.status}</span>
+                `;
+                reportsList.appendChild(entryDiv);
+            });
+
+        } catch (error) {
+            console.error(error);
+            reportsList.innerHTML = "<p>Could not load reports.</p>";
+        }
+    }
+
+    // --- AI Chat Functions (from previous step, unchanged) ---
+
+    async function handleAiQuerySubmit() {
         const query = researchQueryInput.value.trim();
         if (!query) {
             alert("Please enter a research query.");
             return;
         }
 
-        // Show a loading state
         submitButton.disabled = true;
         submitButton.innerText = "Thinking...";
         resultDisplay.innerHTML = "<p>Generating response...</p>";
 
         try {
-            // Call the new AI endpoint
             const response = await fetch(`${workerUrl}/api/query`, {
                 method: "POST",
                 headers: { "Content-Type": "application/json" },
@@ -29,44 +109,28 @@ document.addEventListener("DOMContentLoaded", () => {
             });
 
             const data = await response.json();
-
-            if (!response.ok) {
-                throw new Error(data.error || "An unknown error occurred.");
-            }
-
-            // Display the real result from the AI
+            if (!response.ok) throw new Error(data.error || "An unknown error occurred.");
             resultDisplay.innerText = data.result;
-            
-            // Reload history to include the new entry
             await loadResearchHistory();
-
         } catch (error) {
             console.error("Error fetching AI response:", error);
             resultDisplay.innerText = `Error: ${error.message}`;
         } finally {
-            // Restore the button
             submitButton.disabled = false;
             submitButton.innerText = "Submit";
         }
     }
 
-    /**
-     * Loads and displays the research history from the backend.
-     */
     async function loadResearchHistory() {
         try {
-            // Use the correct endpoint for fetching history
             const response = await fetch(`${workerUrl}/api/research`);
             if (!response.ok) throw new Error(`HTTP error! status: ${response.status}`);
-            
             const logs = await response.json();
-            historyContainer.innerHTML = ""; // Clear existing history
-
+            historyContainer.innerHTML = "";
             if (logs.length === 0) {
                 historyContainer.innerHTML = "<p>No research history found.</p>";
                 return;
             }
-
             logs.forEach(log => {
                 const entryDiv = document.createElement("div");
                 entryDiv.className = "history-entry";
@@ -84,8 +148,10 @@ document.addEventListener("DOMContentLoaded", () => {
     }
 
     // Event Listeners
-    submitButton.addEventListener("click", handleSubmit);
+    uploadForm.addEventListener("submit", handleFileUpload);
+    submitButton.addEventListener("click", handleAiQuerySubmit);
 
-    // Initial load of history
+    // Initial load
+    loadReportsList();
     loadResearchHistory();
 });