Upload 4 files

static/script.js

@@ -8,15 +8,16 @@ document.getElementById("uploadForm").addEventListener("submit", async (e) => {
     const loadingElement = document.getElementById("loading");
     const errorElement = document.getElementById("error");
 
+    // Reset UI
     visualizationImage.src = "";
     visualizationImage.classList.remove("loaded");
     generatedCodeElement.textContent = "";
     errorElement.textContent = "";
     errorElement.classList.add("hidden");
-
     loadingElement.classList.remove("hidden");
 
     try {
+        // Upload file
         const uploadFormData = new FormData();
         uploadFormData.append("file", fileInput.files[0]);
 
@@ -33,6 +34,7 @@ document.getElementById("uploadForm").addEventListener("submit", async (e) => {
         const uploadResult = await uploadResponse.json();
         console.log("✅ File uploaded:", uploadResult);
 
+        // Generate visualization
         const visFormData = new FormData();
         visFormData.append("filename", uploadResult.filename);
         visFormData.append("prompt", promptInput.value);
@@ -50,14 +52,22 @@ document.getElementById("uploadForm").addEventListener("submit", async (e) => {
         const result = await visResponse.json();
         console.log("🔹 Response from API:", result);
 
-        if (result.plot_url) {
-            visualizationImage.src = result.plot_url;  
+        // Always display generated code if present
+        if (result.generated_code) {
+            generatedCodeElement.textContent = result.generated_code;
+        }
+
+        // Display plot or error
+        if (result.plot_base64) {
+            visualizationImage.src = `data:image/png;base64,${result.plot_base64}`;
             visualizationImage.onload = () => {
                 visualizationImage.classList.add("loaded");
             };
-            generatedCodeElement.textContent = result.generated_code;
+        } else if (result.error) {
+            errorElement.textContent = `Error: ${result.error}`;
+            errorElement.classList.remove("hidden");
         } else {
-            throw new Error("No plot URL received.");
+            throw new Error("No plot or error message received.");
         }
     } catch (error) {
         console.error("Error:", error);