Spaces:

Liyew
/

Concept_Guided_ARC_solver

Sleeping

App Files Files Community

woletee commited on Mar 19

Commit

6fb64b4

1 Parent(s): 0c77885

this is the alst updated code for both the index.html and for both the app.py code

Browse files

Files changed (2) hide show

app.py +20 -8
templates/results.html +31 -17

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ def upload():
         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))
@@ -55,13 +55,7 @@ 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 as well
-    test_pairs = [(tolist_safe(sample["input"]), tolist_safe(sample["output"])) for sample in data.get("test", [])]
-    # 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 []
@@ -71,10 +65,28 @@ def upload():
         print("Error during best_program evaluation:", e)
         predicted_output = [["ERROR"]]
     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,

         concept_label, _ = run_inference(model, input_grid, output_grid)
         predicted_HLCs.append(concept_label)
+        input_output_pairs.append((tolist_safe(input_grid), tolist_safe(output_grid)))
     predicted_HLCs = list(set(predicted_HLCs))
         predicted_HLCs=predicted_HLCs
     )
+    # Last grid for optional evaluation
     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 []
         print("Error during best_program evaluation:", e)
         predicted_output = [["ERROR"]]
+    # Evaluate program on test pairs
+    test_pairs = []
+    predicted_test_outputs = []
+    for sample in data.get("test", []):
+        test_input = tolist_safe(sample["input"])
+        test_output = tolist_safe(sample["output"])
+        test_pairs.append((test_input, test_output))
+        try:
+            predicted = tolist_safe(best_program.evaluate(test_input))
+        except Exception as e:
+            print("Error evaluating test input:", e)
+            predicted = [["ERROR"]]
+        predicted_test_outputs.append(predicted)
     return render_template("results.html",
                            hlcs=predicted_HLCs,
                            input_output_pairs=input_output_pairs,
                            test_pairs=test_pairs,
+                           predicted_test_outputs=predicted_test_outputs,
                            best_program=str(best_program),
                            last_input=last_input,
                            last_ground_truth=last_ground_truth,

templates/results.html CHANGED Viewed

@@ -67,11 +67,16 @@
 <body>
     <h1>Predicted High-Level Concepts</h1>
-    <h2>Input / Output Pairs with Concepts</h2>
     <div id="pairs-container"></div>
-    <h2>Test Pairs</h2>
     <div id="test-pairs-container"></div>
     <h2>Best Program</h2>
@@ -81,6 +86,7 @@
         const inputOutputPairs = {{ input_output_pairs | tojson | safe }};
         const hlcs = {{ hlcs | tojson | safe }};
         const testPairs = {{ test_pairs | tojson | safe }};
         const colorMap = {
             0: "#000000", 1: "#0074D9", 2: "#FF4136", 3: "#2ECC40",
@@ -92,6 +98,7 @@
             if (!Array.isArray(gridData) || !gridData.length || !Array.isArray(gridData[0])) return;
             container.style.gridTemplateRows = `repeat(${gridData.length}, 25px)`;
             container.style.gridTemplateColumns = `repeat(${gridData[0].length}, 25px)`;
             gridData.forEach(row => {
                 row.forEach(cell => {
                     const div = document.createElement("div");
@@ -102,6 +109,7 @@
             });
         }
         const pairsContainer = document.getElementById("pairs-container");
         inputOutputPairs.forEach((pair, index) => {
@@ -124,14 +132,10 @@
             const conceptDiv = document.createElement("div");
             conceptDiv.className = "concept-box";
-            const conceptTitle = document.createElement("div");
-            conceptTitle.className = "concept-title";
-            conceptTitle.innerText = "Predicted Concept";
-            const conceptText = document.createElement("div");
-            conceptText.className = "concept-text";
-            conceptText.innerText = hlcs[index] || "N/A";
-            conceptDiv.appendChild(conceptTitle);
-            conceptDiv.appendChild(conceptText);
             container.appendChild(inputDiv);
             container.appendChild(outputDiv);
@@ -139,7 +143,9 @@
             pairsContainer.appendChild(container);
         });
         const testContainer = document.getElementById("test-pairs-container");
         testPairs.forEach((pair, index) => {
             const container = document.createElement("div");
             container.className = "grid-container";
@@ -151,15 +157,23 @@
             inputDiv.appendChild(inputGridBox);
             drawGrid(inputGridBox, pair[0]);
-            const outputDiv = document.createElement("div");
-            outputDiv.innerHTML = `<div class="grid-title">Test 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>

 <body>
     <h1>Predicted High-Level Concepts</h1>
+    <ul>
+        {% for concept in hlcs %}
+            <li>{{ concept }}</li>
+        {% endfor %}
+    </ul>
+    <h2>Training Input / Output Pairs with Concepts</h2>
     <div id="pairs-container"></div>
+    <h2>Test Input / Ground Truth / Predicted Outputs</h2>
     <div id="test-pairs-container"></div>
     <h2>Best Program</h2>
         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",
             if (!Array.isArray(gridData) || !gridData.length || !Array.isArray(gridData[0])) return;
             container.style.gridTemplateRows = `repeat(${gridData.length}, 25px)`;
             container.style.gridTemplateColumns = `repeat(${gridData[0].length}, 25px)`;
+            container.innerHTML = '';
             gridData.forEach(row => {
                 row.forEach(cell => {
                     const div = document.createElement("div");
             });
         }
+        // TRAINING PAIRS
         const pairsContainer = document.getElementById("pairs-container");
         inputOutputPairs.forEach((pair, index) => {
             const conceptDiv = document.createElement("div");
             conceptDiv.className = "concept-box";
+            conceptDiv.innerHTML = `
+                <div class="concept-title">Concept</div>
+                <div class="concept-text">${hlcs[index] || 'N/A'}</div>
+            `;
             container.appendChild(inputDiv);
             container.appendChild(outputDiv);
             pairsContainer.appendChild(container);
         });
+        // TEST PAIRS AND PREDICTIONS
         const testContainer = document.getElementById("test-pairs-container");
         testPairs.forEach((pair, index) => {
             const container = document.createElement("div");
             container.className = "grid-container";
             inputDiv.appendChild(inputGridBox);
             drawGrid(inputGridBox, pair[0]);
+            const gtDiv = document.createElement("div");
+            gtDiv.innerHTML = `<div class="grid-title">Ground Truth Output</div>`;
+            const gtGridBox = document.createElement("div");
+            gtGridBox.className = "grid-box";
+            gtDiv.appendChild(gtGridBox);
+            drawGrid(gtGridBox, pair[1]);
+            const predDiv = document.createElement("div");
+            predDiv.innerHTML = `<div class="grid-title">Predicted Output</div>`;
+            const predGridBox = document.createElement("div");
+            predGridBox.className = "grid-box";
+            predDiv.appendChild(predGridBox);
+            drawGrid(predGridBox, predictedTestOutputs[index]);
             container.appendChild(inputDiv);
+            container.appendChild(gtDiv);
+            container.appendChild(predDiv);
             testContainer.appendChild(container);
         });
     </script>