Spaces:

Liyew
/

Concept_Guided_ARC_solver

Sleeping

App Files Files Community

woletee commited on Mar 19

Commit

0c9440e

1 Parent(s): 729a903

this is the fixed commit

Browse files

Files changed (3) hide show

app.py +20 -20
templates/index.html +3 -4
templates/results.html +106 -165

app.py CHANGED Viewed

@@ -10,9 +10,7 @@ UPLOAD_FOLDER = 'uploads'
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 def tolist_safe(obj):
-    if isinstance(obj, np.ndarray):
-        return obj.tolist()
-    return obj
 @app.route('/')
 def index():
@@ -32,48 +30,50 @@ def upload():
     with open(filepath, 'r') as f:
         data = json.load(f)
     input_output_pairs = []
     predicted_HLCs = []
     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((tolist_safe(input_grid), tolist_safe(output_grid)))
     predicted_HLCs = list(set(predicted_HLCs))
     best_program, generations = genetic_programming(
         input_output_pairs=input_output_pairs,
-        population_size=300,
-        generations=500,
         mutation_rate=0.2,
         crossover_rate=0.7,
         max_depth=3,
         predicted_HLCs=predicted_HLCs
     )
     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))
-print ("server is working ")
 if __name__ == '__main__':
-   app.run(host='0.0.0.0', port=7860)

 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 def tolist_safe(obj):
+    return obj.tolist() if isinstance(obj, np.ndarray) else obj
 @app.route('/')
 def index():
     with open(filepath, 'r') as f:
         data = json.load(f)
+    # Collect training data and predict HLCs
     input_output_pairs = []
     predicted_HLCs = []
     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((tolist_safe(input_grid), tolist_safe(output_grid)))
     predicted_HLCs = list(set(predicted_HLCs))
+    # GP optimization
     best_program, generations = genetic_programming(
         input_output_pairs=input_output_pairs,
+        population_size=200,
+        generations=200,
         mutation_rate=0.2,
         crossover_rate=0.7,
         max_depth=3,
         predicted_HLCs=predicted_HLCs
     )
+    # Evaluate GP program on test inputs
     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:
+            pred = tolist_safe(best_program.evaluate(test_input))
         except Exception as e:
+            print(f"Prediction error: {e}")
+            pred = [["ERROR"]]
+        predicted_test_outputs.append(pred)
+    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)
+    )
 if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=False)

templates/index.html CHANGED Viewed

@@ -2,14 +2,13 @@
 <html lang="en">
 <head>
     <meta charset="UTF-8">
-    <title>ARC Task Upload</title>
-    <link rel="stylesheet" href="{{ url_for('static', filename='style.css') }}">
 </head>
 <body>
-    <h1>Upload ARC Task</h1>
     <form action="/upload" method="post" enctype="multipart/form-data">
         <input type="file" name="file" accept=".json" required>
-        <button type="submit">Submit</button>
     </form>
 </body>
 </html>

 <html lang="en">
 <head>
     <meta charset="UTF-8">
+    <title>Upload ARC Task</title>
 </head>
 <body>
+    <h1>Upload ARC Task File</h1>
     <form action="/upload" method="post" enctype="multipart/form-data">
         <input type="file" name="file" accept=".json" required>
+        <button type="submit">Upload</button>
     </form>
 </body>
 </html>

templates/results.html CHANGED Viewed

@@ -2,180 +2,121 @@
 <html lang="en">
 <head>
     <meta charset="UTF-8">
-    <title>ARC Grid Visualization</title>
     <style>
-        body {
-            font-family: Arial, sans-serif;
-            background-color: white;
-            color: black;
-            text-align: center;
-            margin: 0;
-            padding: 20px;
-        }
-        .grid-container {
-            display: flex;
-            justify-content: center;
-            gap: 40px;
-            flex-wrap: wrap;
-            margin-bottom: 30px;
-        }
-        .grid-box {
-            display: grid;
-            gap: 2px;
-            margin-bottom: 10px;
-        }
         .cell {
-            width: 25px;
-            height: 25px;
             border: 1px solid #ccc;
         }
-        .grid-title {
-            font-weight: bold;
-            margin-bottom: 5px;
-        }
-        .concept-box {
-            display: flex;
-            flex-direction: column;
-            justify-content: center;
-            align-items: center;
-            min-width: 150px;
-        }
-        .concept-title {
-            font-weight: bold;
-            margin-bottom: 5px;
-        }
-        .concept-text {
-            font-size: 16px;
-            padding: 5px 10px;
-            border: 1px solid #888;
-            border-radius: 8px;
-            background-color: #f4f4f4;
-        }
-        pre {
-            background-color: #f5f5f5;
-            padding: 15px;
-            font-family: Consolas, monospace;
-            font-size: 14px;
-            border-radius: 8px;
-            max-width: 900px;
-            margin: auto;
-            white-space: pre-wrap;
-            border: 1px solid #ddd;
-        }
     </style>
 </head>
 <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>
-    <pre>{{ best_program }}</pre>
-    <script>
-        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",
-            4: "#FFDC00", 5: "#AAAAAA", 6: "#F012BE", 7: "#FF851B",
-            8: "#7FDBFF", 9: "#870C25"
-        };
-        function drawGrid(container, gridData) {
-            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");
-                    div.className = "cell";
-                    div.style.backgroundColor = colorMap[cell] || "#000000";
-                    container.appendChild(div);
-                });
-            });
-        }
-        // TRAINING PAIRS
-        const pairsContainer = document.getElementById("pairs-container");
-        inputOutputPairs.forEach((pair, index) => {
-            const container = document.createElement("div");
-            container.className = "grid-container";
-            const inputDiv = document.createElement("div");
-            inputDiv.innerHTML = `<div class="grid-title">Input Grid</div>`;
-            const inputGridBox = document.createElement("div");
-            inputGridBox.className = "grid-box";
-            inputDiv.appendChild(inputGridBox);
-            drawGrid(inputGridBox, pair[0]);
-            const outputDiv = document.createElement("div");
-            outputDiv.innerHTML = `<div class="grid-title">Output Grid</div>`;
-            const outputGridBox = document.createElement("div");
-            outputGridBox.className = "grid-box";
-            outputDiv.appendChild(outputGridBox);
-            drawGrid(outputGridBox, pair[1]);
-            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);
-            container.appendChild(conceptDiv);
-            pairsContainer.appendChild(container);
-        });
-        const testContainer = document.getElementById("test-pairs-container");
-        testPairs.forEach((pair, index) => {
-            const container = document.createElement("div");
-            container.className = "grid-container";
-            const inputDiv = document.createElement("div");
-            inputDiv.innerHTML = `<div class="grid-title">Test Input Grid</div>`;
-            const inputGridBox = document.createElement("div");
-            inputGridBox.className = "grid-box";
-            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>
 </body>
 </html>

 <html lang="en">
 <head>
     <meta charset="UTF-8">
+    <title>Results</title>
     <style>
+        body { font-family: sans-serif; padding: 20px; }
+        .grid-container { display: flex; gap: 30px; flex-wrap: wrap; margin-bottom: 30px; }
+        .grid-box { display: grid; gap: 2px; margin-bottom: 10px; }
         .cell {
+            width: 25px; height: 25px;
             border: 1px solid #ccc;
         }
+        .grid-title { font-weight: bold; margin-bottom: 5px; }
+        pre { background: #f4f4f4; padding: 10px; border-radius: 5px; }
     </style>
 </head>
 <body>
+<h2>Predicted High-Level Concepts</h2>
+<ul>
+  {% for concept in hlcs %}
+    <li>{{ concept }}</li>
+  {% endfor %}
+</ul>
+<h2>Training Grids</h2>
+<div id="train-container"></div>
+<h2>Test Grids</h2>
+<div id="test-container"></div>
+<h2>Best Program</h2>
+<pre>{{ best_program }}</pre>
+<script>
+const inputOutputPairs = {{ input_output_pairs | 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",
+    4: "#FFDC00", 5: "#AAAAAA", 6: "#F012BE", 7: "#FF851B",
+    8: "#7FDBFF", 9: "#870C25"
+};
+function drawGrid(container, gridData) {
+    if (!Array.isArray(gridData)) 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');
+            div.className = "cell";
+            div.style.backgroundColor = colorMap[cell] || "#000";
+            container.appendChild(div);
         });
+    });
+}
+// Show training pairs
+const trainContainer = document.getElementById("train-container");
+inputOutputPairs.forEach((pair, index) => {
+    const container = document.createElement("div");
+    container.className = "grid-container";
+    const inputDiv = document.createElement("div");
+    inputDiv.innerHTML = `<div class="grid-title">Train Input</div>`;
+    const inputBox = document.createElement("div");
+    inputBox.className = "grid-box";
+    inputDiv.appendChild(inputBox);
+    drawGrid(inputBox, pair[0]);
+    const outputDiv = document.createElement("div");
+    outputDiv.innerHTML = `<div class="grid-title">Train Output</div>`;
+    const outputBox = document.createElement("div");
+    outputBox.className = "grid-box";
+    outputDiv.appendChild(outputBox);
+    drawGrid(outputBox, pair[1]);
+    container.appendChild(inputDiv);
+    container.appendChild(outputDiv);
+    trainContainer.appendChild(container);
+});
+// Show test + prediction pairs
+const testContainer = document.getElementById("test-container");
+testPairs.forEach((pair, index) => {
+    const container = document.createElement("div");
+    container.className = "grid-container";
+    const inputDiv = document.createElement("div");
+    inputDiv.innerHTML = `<div class="grid-title">Test Input</div>`;
+    const inputBox = document.createElement("div");
+    inputBox.className = "grid-box";
+    inputDiv.appendChild(inputBox);
+    drawGrid(inputBox, pair[0]);
+    const gtDiv = document.createElement("div");
+    gtDiv.innerHTML = `<div class="grid-title">Ground Truth</div>`;
+    const gtBox = document.createElement("div");
+    gtBox.className = "grid-box";
+    gtDiv.appendChild(gtBox);
+    drawGrid(gtBox, pair[1]);
+    const predDiv = document.createElement("div");
+    predDiv.innerHTML = `<div class="grid-title">Predicted</div>`;
+    const predBox = document.createElement("div");
+    predBox.className = "grid-box";
+    predDiv.appendChild(predBox);
+    drawGrid(predBox, predictedTestOutputs[index]);
+    container.appendChild(inputDiv);
+    container.appendChild(gtDiv);
+    container.appendChild(predDiv);
+    testContainer.appendChild(container);
+});
+</script>
 </body>
 </html>