Update script.js

script.js

@@ -1,106 +1,233 @@
-// Handle Generate button click
-document.getElementById("generateBtn").addEventListener("click", async () => {
-  const question = document.getElementById("questionInput").value.trim();
-  const responseBox = document.getElementById("responseOutput");
-  const historyBox = document.getElementById("history");
-
-  if (!question) {
-    responseBox.value = "Please enter your legal question.";
-    return;
-  }
-
-  responseBox.value = "Generating...";
-
-  try {
-    const res = await fetch("/answer", {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ question })
-    });
-
-    const data = await res.json();
-    const answer = data.answer || "No answer received.";
-
-    responseBox.value = answer;
-
-    // Update top label
-    document.getElementById("topLabel").innerText = question;
-
-    // Update history
-    historyBox.value += `\n\nYou: ${question}\nBot: ${answer}`;
-  } catch (err) {
-    responseBox.value = `Error: ${err}`;
-  }
-});
-
-// Handle Reset button
-function resetApp() {
-  document.getElementById("questionInput").value = "";
-  document.getElementById("responseOutput").value = "";
-  document.getElementById("topLabel").innerText = "Dictate your legal question!";
-}
-
-// Handle Read Aloud
-function readAloud() {
-  const text = document.getElementById("responseOutput").value;
-  if (!text.trim()) return;
-  const synth = window.speechSynthesis;
-  const utterance = new SpeechSynthesisUtterance(text);
-  synth.speak(utterance);
-}
-
-// Handle Save
-function saveQA() {
-  const question = document.getElementById("questionInput").value.trim();
-  const answer = document.getElementById("responseOutput").value.trim();
-
-  if (!question || !answer) {
-    alert("Nothing to save.");
-    return;
-  }
-
-  const blob = new Blob([`Question:\n${question}\n\nAnswer:\n${answer}`], { type: "text/plain" });
-  const link = document.createElement("a");
-  link.href = URL.createObjectURL(blob);
-  link.download = "QnA.txt";
-  document.body.appendChild(link);
-  link.click();
-  document.body.removeChild(link);
-}
-
-// Handle Upload MP3 (Disabled in web)
-function uploadMP3() {
-  alert("📁 MP3 upload is only supported in the desktop assistant.");
-}
-
-// Handle Dictate (Web Speech API)
-function handleDictate() {
-  if (!('webkitSpeechRecognition' in window)) {
-    alert("Speech recognition not supported in this browser. Use Chrome.");
-    return;
-  }
-
-  const recognition = new webkitSpeechRecognition();
-  recognition.lang = "en-US";
-  recognition.interimResults = false;
-  recognition.maxAlternatives = 1;
-
-  document.getElementById("topLabel").innerText = "Listening... 🎙";
-
-  recognition.onresult = function (event) {
-    const transcript = event.results[0][0].transcript;
-    document.getElementById("questionInput").value = transcript;
-    document.getElementById("topLabel").innerText = transcript;
-  };
-
-  recognition.onerror = function (event) {
-    console.error("Speech recognition error:", event.error);
-    document.getElementById("topLabel").innerText = "Could not recognize speech.";
-  };
-
-  recognition.onend = function () {
-    console.log("Speech recognition ended.");
-  };
-
-  recognition.start();
-}
+// Function to update the top label
+function updateTopLabel(text) {
+  document.getElementById("topLabel").innerText = text;
+}
+
+// Function to update the history textarea
+function updateHistory(userQuestion, botAnswer) {
+  const historyBox = document.getElementById("history");
+  const timestamp = new Date().toLocaleTimeString();
+  historyBox.value += `\n\n[${timestamp}] You: ${userQuestion}\n[${timestamp}] Bot: ${botAnswer}`;
+  historyBox.scrollTop = historyBox.scrollHeight; // Scroll to bottom
+}
+
+// Handle Generate button click
+document.getElementById("generateBtn").addEventListener("click", async () => {
+  const questionInput = document.getElementById("question"); // Changed from questionInput
+  const answerOutput = document.getElementById("answer");   // Changed from responseOutput
+  const question = questionInput.value.trim();
+
+  if (!question) {
+    answerOutput.value = "Please enter your legal question.";
+    return;
+  }
+
+  answerOutput.value = "Generating...";
+  updateTopLabel("Generating answer...");
+
+  try {
+    const res = await fetch("/answer", {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ question })
+    });
+
+    if (!res.ok) {
+        const errorData = await res.json();
+        throw new Error(errorData.answer || "Failed to generate answer.");
+    }
+
+    const data = await res.json();
+    const answer = data.answer || "No answer received.";
+
+    answerOutput.value = answer;
+    updateTopLabel(question); // Update top label with the question
+    updateHistory(question, answer); // Update history with Q&A
+
+  } catch (err) {
+    answerOutput.value = `Error: ${err.message || err}`;
+    updateTopLabel("Error generating answer.");
+  }
+});
+
+// Handle Reset button
+function resetApp() {
+  document.getElementById("question").value = "";
+  document.getElementById("answer").value = "";
+  document.getElementById("topLabel").innerText = "Dictate your legal question!";
+  document.getElementById("history").value = ""; // Clear history on reset
+}
+
+// Handle Read Aloud (using Web Speech API)
+function readAloud() {
+  const text = document.getElementById("answer").value;
+  if (!text.trim()) {
+      alert("No answer to read aloud.");
+      return;
+  }
+  const synth = window.speechSynthesis;
+  // Stop any ongoing speech before starting a new one
+  synth.cancel();
+  const utterance = new SpeechSynthesisUtterance(text);
+  synth.speak(utterance);
+}
+
+// Handle Save/Print
+function saveQA() {
+  const question = document.getElementById("question").value.trim();
+  const answer = document.getElementById("answer").value.trim();
+
+  if (!question || !answer) {
+    alert("Nothing to save.");
+    return;
+  }
+
+  const blob = new Blob([`Question:\n${question}\n\nAnswer:\n${answer}`], { type: "text/plain" });
+  const link = document.createElement("a");
+  link.href = URL.createObjectURL(blob);
+  link.download = "Legal_Assistant_QnA.txt";
+  document.body.appendChild(link);
+  link.click();
+  document.body.removeChild(link);
+  URL.revokeObjectURL(link.href); // Clean up the URL object
+}
+
+// Handle Upload MP3
+async function uploadMP3() {
+  const fileInput = document.createElement('input');
+  fileInput.type = 'file';
+  fileInput.accept = '.mp3';
+  fileInput.onchange = async (e) => {
+    const file = e.target.files[0];
+    if (!file) {
+      document.getElementById("answer").value = "No file selected.";
+      return;
+    }
+
+    document.getElementById("answer").value = "Uploading and transcribing MP3...";
+    updateTopLabel("Processing MP3...");
+
+    const formData = new FormData();
+    formData.append('file', file);
+
+    try {
+      const res = await fetch("/upload-mp3", {
+        method: "POST",
+        body: formData
+      });
+
+      if (!res.ok) {
+          const errorData = await res.json();
+          throw new Error(errorData.message || "Failed to upload and transcribe MP3.");
+      }
+
+      const data = await res.json();
+      const transcription = data.transcription || "Could not transcribe audio.";
+
+      updateTopLabel(transcription);
+      document.getElementById("question").value = transcription; // Put transcription in question box
+      document.getElementById("answer").value = "Transcription complete. You can now summarize or generate an answer.";
+
+      // Optional: Prompt user to summarize or generate an answer
+      const summarizeConfirm = confirm("MP3 transcribed. Do you want to summarize it?");
+      if (summarizeConfirm) {
+          const summaryRes = await fetch("/summarize", {
+              method: "POST",
+              headers: { "Content-Type": "application/json" },
+              body: JSON.stringify({ text: transcription })
+          });
+          const summaryData = await summaryRes.json();
+          if (summaryRes.ok) {
+              document.getElementById("question").value = summaryData.summary;
+              document.getElementById("answer").value = "Summary pasted in question box. You can now generate an answer.";
+              updateTopLabel("Summary ready.");
+          } else {
+              document.getElementById("answer").value = `Error summarizing: ${summaryData.summary}`;
+          }
+      } else {
+          const generateConfirm = confirm("Do you want to generate an answer from the full transcription?");
+          if (generateConfirm) {
+              // Trigger generateAnswer function with the transcription
+              document.getElementById("generateBtn").click(); // Simulate click on generate button
+          }
+      }
+
+    } catch (err) {
+      document.getElementById("answer").value = `Error: ${err.message || err}`;
+      updateTopLabel("Error processing MP3.");
+    }
+  };
+  fileInput.click(); // Open file dialog
+}
+
+// Handle Dictate (Web Speech API)
+let recognition; // Keep recognition object in scope to stop it
+
+function handleDictate() {
+  if (!('webkitSpeechRecognition' in window)) {
+    alert("Speech recognition not supported in this browser. Use Chrome.");
+    return;
+  }
+
+  // If already listening, stop it
+  if (recognition && (recognition.recognizing || recognition.continuous)) {
+    recognition.stop();
+    document.getElementById("topLabel").innerText = "Dictate your legal question!";
+    // If you want to automatically trigger answer generation after stopping dictation:
+    // document.getElementById("generateBtn").click();
+    return;
+  }
+
+  recognition = new webkitSpeechRecognition();
+  recognition.lang = "en-US";
+  recognition.interimResults = false;
+  recognition.maxAlternatives = 1;
+  // recognition.continuous = true; // For continuous listening
+
+  document.getElementById("topLabel").innerText = "Listening... 🎙 (Click Dictate again to stop)";
+  document.getElementById("question").value = ""; // Clear previous question
+
+  recognition.onresult = function (event) {
+    const transcript = event.results[0][0].transcript;
+    document.getElementById("question").value = transcript;
+    updateTopLabel(transcript);
+    // Automatically generate answer after dictation
+    document.getElementById("generateBtn").click();
+  };
+
+  recognition.onerror = function (event) {
+    console.error("Speech recognition error:", event.error);
+    document.getElementById("topLabel").innerText = "Could not recognize speech or an error occurred.";
+  };
+
+  recognition.onend = function () {
+    console.log("Speech recognition ended.");
+    // Only revert label if not actively listening or if an error occurred during listening
+    if (!recognition.recognizing) {
+        document.getElementById("topLabel").innerText = "Dictate your legal question!";
+    }
+  };
+
+  recognition.start();
+}
+
+
+// Event listeners (ensure they are properly mapped to the new HTML IDs)
+document.addEventListener("DOMContentLoaded", () => {
+    // The existing event listener for "generateBtn" is already set up.
+    // Ensure the button IDs in HTML match what's used in JS.
+    // current IDs:
+    // Dictate: handleDictate()
+    // Generate Response: generateAnswer() -> now mapped to generateBtn click event
+    // Read Aloud: readAloud()
+    // Upload MP3: uploadMP3()
+    // Save/Print: saveQA()
+    // Reset: resetApp()
+
+    // Add event listeners for other buttons if they don't have inline onclick
+    // For example, if you replace onclick="generateAnswer()" with id="generateBtn"
+    // then the document.getElementById("generateBtn").addEventListener("click", ...) handles it.
+
+    // If your HTML buttons use onclick, make sure the function names match these JS functions.
+    // Example: <button onclick="handleDictate()">
+});