added for the test pairs as well

app.py

@@ -38,13 +38,13 @@ def upload():
     for sample in data.get("train", []):
         input_grid = sample["input"]
         output_grid = sample["output"]
-
         concept_label, _ = run_inference(model, input_grid, output_grid)
         predicted_HLCs.append(concept_label)
         input_output_pairs.append((input_grid, output_grid))
 
     predicted_HLCs = list(set(predicted_HLCs))
 
+    # Run Genetic Programming
     best_program, generations = genetic_programming(
         input_output_pairs=input_output_pairs,
         population_size=300,
@@ -55,30 +55,32 @@ def upload():
         predicted_HLCs=predicted_HLCs
     )
 
-    # Convert train pairs to plain lists
-    input_output_pairs = [(tolist_safe(i), tolist_safe(o)) for i, o in input_output_pairs]
+    # Prepare TEST PAIRS and predicted test outputs
+    test_pairs = []
+    predicted_test_outputs = []
+
+    for sample in data.get("test", []):
+        input_grid = tolist_safe(sample["input"])
+        output_grid = tolist_safe(sample["output"])
+        test_pairs.append((input_grid, output_grid))
 
-    # Prepare test pairs as well
-    test_pairs = [(tolist_safe(sample["input"]), tolist_safe(sample["output"])) for sample in data.get("test", [])]
+        try:
+            predicted = tolist_safe(best_program.evaluate(input_grid))
+        except Exception as e:
+            print("Error during prediction on test input:", e)
+            predicted = [["ERROR"]]
 
-    # Prepare final program evaluation display
-    last_input = input_output_pairs[-1][0] if input_output_pairs else []
-    last_ground_truth = input_output_pairs[-1][1] if input_output_pairs else []
+        predicted_test_outputs.append(predicted)
 
-    try:
-        predicted_output = tolist_safe(best_program.evaluate(last_input))
-    except Exception as e:
-        print("Error during best_program evaluation:", e)
-        predicted_output = [["ERROR"]]
+    # Clean train pairs for rendering
+    input_output_pairs = [(tolist_safe(i), tolist_safe(o)) for i, o in input_output_pairs]
 
     return render_template("results.html",
                            hlcs=predicted_HLCs,
                            input_output_pairs=input_output_pairs,
-                           test_pairs=test_pairs,
                            best_program=str(best_program),
-                           last_input=last_input,
-                           last_ground_truth=last_ground_truth,
-                           predicted_output=predicted_output)
+                           test_pairs=test_pairs,
+                           predicted_test_outputs=predicted_test_outputs)
 
 if __name__ == '__main__':
     app.run(host="0.0.0.0", port=7860)

templates/results.html

@@ -68,10 +68,9 @@
 
     <h1>Predicted High-Level Concepts</h1>
     <h2>Input / Output Pairs with Concepts</h2>
-
     <div id="pairs-container"></div>
 
-    <h2>Test Pairs</h2>
+    <h2>Test Set Evaluation</h2>
     <div id="test-pairs-container"></div>
 
     <h2>Best Program</h2>
@@ -81,6 +80,7 @@
         const inputOutputPairs = {{ input_output_pairs | tojson | safe }};
         const hlcs = {{ hlcs | tojson | safe }};
         const testPairs = {{ test_pairs | tojson | safe }};
+        const predictedTestOutputs = {{ predicted_test_outputs | tojson | safe }};
 
         const colorMap = {
             0: "#000000", 1: "#0074D9", 2: "#FF4136", 3: "#2ECC40",
@@ -103,7 +103,6 @@
         }
 
         const pairsContainer = document.getElementById("pairs-container");
-
         inputOutputPairs.forEach((pair, index) => {
             const container = document.createElement("div");
             container.className = "grid-container";
@@ -152,17 +151,16 @@
             drawGrid(inputGridBox, pair[0]);
 
             const outputDiv = document.createElement("div");
-            outputDiv.innerHTML = `<div class="grid-title">Test Ground Truth Output Grid</div>`;
+            outputDiv.innerHTML = `<div class="grid-title">Ground Truth Output Grid</div>`;
             const outputGridBox = document.createElement("div");
             outputGridBox.className = "grid-box";
             outputDiv.appendChild(outputGridBox);
             drawGrid(outputGridBox, pair[1]);
 
-            container.appendChild(inputDiv);
-            container.appendChild(outputDiv);
-            testContainer.appendChild(container);
-        });
-    </script>
+            const predDiv = document.createElement("div");
+            predDiv.innerHTML = `<div class="grid-title">Predicted Output by Program</div>`;
+            const predGridBox = document.createElement("div");
+            predGridBox.className = "grid-box";
+            predDiv.appendChild(predGridBox);
+            drawGrid(predGridBox, predictedTestOutputs[index]);
 
-</body>
-</html>