Spaces:

Liyew
/

Concept_Guided_ARC_solver

Sleeping

App Files Files Community

woletee commited on Mar 19

Commit

0ecdc68

1 Parent(s): a37b981

I hope this would be the last

Browse files

Files changed (3) hide show

app.py +30 -25
templates/index.html +5 -28
templates/results.html +4 -4

app.py CHANGED Viewed

@@ -16,40 +16,43 @@ def tolist_safe(obj):
 def index():
     return render_template('index.html')
-@app.route('/submit_json', methods=['POST'])
-def submit_json():
-    json_data = request.form.get('json_data')
-    if not json_data:
-        return "No JSON data provided"
-    try:
-        data = json.loads(json_data)
-    except json.JSONDecodeError:
-        return "Invalid JSON format!"
-    # Continue processing exactly as before:
     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
     )
-    # Evaluate test pairs (same as before)
     test_pairs = []
     predicted_test_outputs = []
     for sample in data.get("test", []):
@@ -57,18 +60,20 @@ def submit_json():
         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:
-            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))
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=False)

 def index():
     return render_template('index.html')
+@app.route('/upload', methods=['POST'])
+def upload():
+    if 'file' not in request.files:
+        return "No file part"
+    file = request.files['file']
+    if file.filename == '':
+        return "No selected file"
+    filepath = os.path.join(UPLOAD_FOLDER, file.filename)
+    file.save(filepath)
+    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_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,36 +2,13 @@
 <html lang="en">
 <head>
     <meta charset="UTF-8">
-    <title>Paste ARC Task JSON</title>
-    <style>
-        body {
-            font-family: Arial, sans-serif;
-            padding: 20px;
-            text-align: center;
-        }
-        textarea {
-            width: 90%;
-            height: 400px;
-            font-family: monospace;
-            font-size: 14px;
-            padding: 10px;
-            margin-bottom: 20px;
-        }
-        button {
-            padding: 10px 20px;
-            font-size: 16px;
-            cursor: pointer;
-        }
-    </style>
 </head>
 <body>
-    <h1>Paste ARC Task JSON</h1>
-    <form action="/submit_json" method="post">
-        <textarea name="json_data" placeholder="Paste your ARC JSON here..." required></textarea>
-        <br>
-        <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

@@ -64,15 +64,15 @@
 </head>
 <body>
-<h1>ARC Grid Visualization</h1>
-<h2>Training Input / Output Pairs with Concepts</h2>
 <div id="pairs-container"></div>
-<h2>Best Program</h2>
 <pre>{{ best_program }}</pre>
-<h2>Test Input / Ground Truth / Predicted Outputs</h2>
 <div id="test-pairs-container"></div>

 </head>
 <body>
+<h1>Concept Guided GP</h1>
+<h2>Step1: Training Pairs (left) with Concepts by recognition module (right most)</h2>
 <div id="pairs-container"></div>
+<h2>Step2: Best Program  by GP</h2>
 <pre>{{ best_program }}</pre>
+<h2>Test Input / Ground Truth / Predicted Outputs (by applying best program)</h2>
 <div id="test-pairs-container"></div>