Spaces:

Liyew
/

Concept_Guided_ARC_solver

Sleeping

App Files Files Community

woletee commited on Mar 20

Commit

2f1b584

1 Parent(s): 0ecdc68

I am not sure if this change could be applicable or not

Browse files

Files changed (12) hide show

GP.py +2 -20
Vit_concept.py +1 -12
__pycache__/GP.cpython-38.pyc +0 -0
__pycache__/Nods.cpython-38.pyc +0 -0
__pycache__/Vit_concept.cpython-38.pyc +0 -0
__pycache__/custom_t5_vit.cpython-38.pyc +0 -0
__pycache__/dsl.cpython-38.pyc +0 -0
__pycache__/task_loader.cpython-38.pyc +0 -0
app.py +7 -13
templates/index.html +28 -5
templates/results.html +1 -1
uploads/AboveBelow3 (6).json +1 -0

GP.py CHANGED Viewed

@@ -350,29 +350,11 @@ def genetic_programming(input_output_pairs, population_size, generations, mutati
 if __name__ == "__main__":
  if __name__ == "__main__":
     input_output_pairs = [
-        (
-            [[0,0,0,0,0,0,0,0],
-             [0,3,3,3,0,0,0,0],
-             [0,0,0,0,0,0,0,0],
-             [0,4,4,4,0,0,0,0],
-             [0,4,4,4,0,0,0,0],
-             [0,4,4,4,0,0,0,0],
-             [0,0,0,0,0,3,3,3],
-             [0,0,3,3,3,0,0,0]],
-            [[0,0,0,0,0,0,0,0],
-             [0,0,0,0,0,0,0,0],
-             [0,3,3,3,0,0,0,0],
-             [0,4,4,4,0,0,0,0],
-             [0,4,4,4,0,0,0,0],
-             [0,4,4,4,0,0,0,0],
-             [0,0,0,0,0,3,3,3],
-             [0,0,3,3,3,0,0,0]]
-        )
     ]
-    predicted_HLCs = ["Above Beloww"]
     best_program, generations = genetic_programming(

 if __name__ == "__main__":
  if __name__ == "__main__":
     input_output_pairs = [
     ]
+    predicted_HLCs = []
     best_program, generations = genetic_programming(

Vit_concept.py CHANGED Viewed

@@ -83,12 +83,8 @@ def preprocess_for_inference(input_grid, output_grid):
     input_tokens = "<s> Input Grid: " + reformat_arc_tokens(input_grid) + " </s>"
     output_tokens = " Output Grid: " + reformat_arc_tokens(output_grid) + " </s>"
     return input_tokens + output_tokens
-# Concept Label Mapping
 CONCEPT_LABELS =  {'Above_below': 0, 'Below_row_line': 1, 'Center': 2, 'Copy': 3, 'Horizontal_vertical': 4, 'Inside_outside': 5, 'Remove_below_horizontal_line': 6}
 CONCEPT_LABELS_INV = {v: k for k, v in CONCEPT_LABELS.items()}
-# Map ViT Concept to GP Function
 CONCEPT_TO_FUNCTION_MAP = {
     'Center': 'find_center_pixel',
     'Copy': 'identity',
@@ -114,14 +110,9 @@ def run_inference(model, input_grid, output_grid):
     return concept_label, mapped_function
 if __name__ == "__main__":
-    # Path to your JSON file
     JSON_DATA_PATH = r"C:\Users\gebre\OneDrive - GIST\문서\KakaoTalk Downloads\GPARC_concept_with_vit\GPARC\SRC\data\AboveBelow3.json"
-    # Load JSON data
     with open(JSON_DATA_PATH, "r") as f:
         data = json.load(f)
-    # Loop through both train and test sets
     results = []
     for split_name in ["train", "test"]:
@@ -136,14 +127,12 @@ if __name__ == "__main__":
                     "input": input_grid,
                     "output": output_grid,
                     "predicted_label": predicted_label,
-                    "mapped_function": str(mapped_function)  # in case it's a callable
                 })
             results.append({
                 "split": split_name,
                 "predictions": split_results
             })
-    # Optionally: save the result to a JSON file
     with open("inference_results.json", "w") as f:
         json.dump(results, f, indent=2)

     input_tokens = "<s> Input Grid: " + reformat_arc_tokens(input_grid) + " </s>"
     output_tokens = " Output Grid: " + reformat_arc_tokens(output_grid) + " </s>"
     return input_tokens + output_tokens
 CONCEPT_LABELS =  {'Above_below': 0, 'Below_row_line': 1, 'Center': 2, 'Copy': 3, 'Horizontal_vertical': 4, 'Inside_outside': 5, 'Remove_below_horizontal_line': 6}
 CONCEPT_LABELS_INV = {v: k for k, v in CONCEPT_LABELS.items()}
 CONCEPT_TO_FUNCTION_MAP = {
     'Center': 'find_center_pixel',
     'Copy': 'identity',
     return concept_label, mapped_function
 if __name__ == "__main__":
     JSON_DATA_PATH = r"C:\Users\gebre\OneDrive - GIST\문서\KakaoTalk Downloads\GPARC_concept_with_vit\GPARC\SRC\data\AboveBelow3.json"
     with open(JSON_DATA_PATH, "r") as f:
         data = json.load(f)
     results = []
     for split_name in ["train", "test"]:
                     "input": input_grid,
                     "output": output_grid,
                     "predicted_label": predicted_label,
+                    "mapped_function": str(mapped_function)
                 })
             results.append({
                 "split": split_name,
                 "predictions": split_results
             })
     with open("inference_results.json", "w") as f:
         json.dump(results, f, indent=2)

__pycache__/GP.cpython-38.pyc ADDED Viewed

Binary file (11.6 kB). View file

__pycache__/Nods.cpython-38.pyc ADDED Viewed

Binary file (3.33 kB). View file

__pycache__/Vit_concept.cpython-38.pyc ADDED Viewed

Binary file (6.25 kB). View file

__pycache__/custom_t5_vit.cpython-38.pyc ADDED Viewed

Binary file (34 kB). View file

__pycache__/dsl.cpython-38.pyc ADDED Viewed

Binary file (18.5 kB). View file

__pycache__/task_loader.cpython-38.pyc ADDED Viewed

Binary file (958 Bytes). View file

app.py CHANGED Viewed

@@ -16,19 +16,13 @@ def tolist_safe(obj):
 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 = []

 def index():
     return render_template('index.html')
+@app.route('/submit_json', methods=['POST'])
+def submit_json():
+    try:
+        raw_json = request.form.get('json_input')
+        data = json.loads(raw_json)
+    except Exception as e:
+        return f"Invalid JSON input: {e}"
     # Collect training data and predict HLCs
     input_output_pairs = []

templates/index.html CHANGED Viewed

@@ -2,13 +2,36 @@
 <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>

 <html lang="en">
 <head>
     <meta charset="UTF-8">
+    <title>Paste ARC Task JSON</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            background-color: white;
+            padding: 30px;
+            text-align: center;
+        }
+        textarea {
+            width: 80%;
+            height: 300px;
+            padding: 10px;
+            font-family: monospace;
+            font-size: 14px;
+            margin-bottom: 20px;
+        }
+        button {
+            padding: 10px 20px;
+            font-size: 16px;
+            cursor: pointer;
+        }
+    </style>
 </head>
 <body>
+    <h1>Concept Guided GP</h1>
+    <form action="/submit_json" method="post">
+        <textarea name="json_input" placeholder="ARC task" required></textarea><br>
+        <button type="submit">What concept is in this task</button>
     </form>
 </body>
 </html>

templates/results.html CHANGED Viewed

@@ -64,7 +64,7 @@
 </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>

 </head>
 <body>
+<h1>Concept Guided PS</h1>
 <h2>Step1: Training Pairs (left) with Concepts by recognition module (right most)</h2>
 <div id="pairs-container"></div>

uploads/AboveBelow3 (6).json ADDED Viewed

	@@ -0,0 +1 @@

+ {"train":[{"input":[[0,0,0,0,0,0,0,0],[0,3,3,3,0,0,0,0],[0,0,0,0,0,0,0,0],[0,4,4,4,0,0,0,0],[0,4,4,4,0,0,0,0],[0,4,4,4,0,0,0,0],[0,0,0,0,0,3,3,3],[0,0,3,3,3,0,0,0]],"output":[[0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0],[0,3,3,3,0,0,0,0],[0,4,4,4,0,0,0,0],[0,4,4,4,0,0,0,0],[0,4,4,4,0,0,0,0],[0,0,0,0,0,3,3,3],[0,0,3,3,3,0,0,0]]},{"input":[[0,0,2,2,2,2,0,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,6,6,6,6,0,0,0,0],[0,0,6,6,6,6,0,0,0,0],[0,0,6,6,6,6,0,0,0,0],[0,0,6,6,6,6,0,0,0,0],[2,2,2,2,2,0,0,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,0,0,2,2,2,0,0,0]],"output":[[0,0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,2,2,2,2,0,0,0,0],[0,0,6,6,6,6,0,0,0,0],[0,0,6,6,6,6,0,0,0,0],[0,0,6,6,6,6,0,0,0,0],[0,0,6,6,6,6,0,0,0,0],[2,2,2,2,2,0,0,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,0,0,2,2,2,0,0,0]]},{"input":[[0,0,4,4,0,0,0],[0,0,0,0,0,0,0],[0,0,1,1,0,0,0],[0,0,1,1,0,0,0],[0,4,4,4,0,0,0],[0,0,0,0,0,0,0],[0,0,0,0,0,0,0]],"output":[[0,0,0,0,0,0,0],[0,0,4,4,0,0,0],[0,0,1,1,0,0,0],[0,0,1,1,0,0,0],[0,4,4,4,0,0,0],[0,0,0,0,0,0,0],[0,0,0,0,0,0,0]]}],"test":[{"input":[[0,0,1,1,1,1,1,0,0],[0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0],[0,0,5,5,5,5,5,0,0],[0,0,5,5,5,5,5,0,0],[0,0,5,5,5,5,5,0,0],[0,0,5,5,5,5,5,0,0],[0,0,5,5,5,5,5,0,0],[0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0],[0,0,1,1,1,1,1,0,0]],"output":[[0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0],[0,0,1,1,1,1,1,0,0],[0,0,5,5,5,5,5,0,0],[0,0,5,5,5,5,5,0,0],[0,0,5,5,5,5,5,0,0],[0,0,5,5,5,5,5,0,0],[0,0,5,5,5,5,5,0,0],[0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0],[0,0,1,1,1,1,1,0,0]]},{"input":[[0,0,0,0,1,1,1,0,0],[0,0,0,0,0,0,0,0,0],[0,0,4,4,4,4,4,0,0],[0,0,4,0,0,0,4,0,0],[0,0,4,0,0,0,4,0,0],[0,0,4,0,0,0,4,0,0],[0,0,4,4,4,4,4,0,0],[0,0,0,0,0,0,0,0,0],[0,0,0,1,1,1,1,0,0]],"output":[[0,0,0,0,0,0,0,0,0],[0,0,0,0,1,1,1,0,0],[0,0,4,4,4,4,4,0,0],[0,0,4,0,0,0,4,0,0],[0,0,4,0,0,0,4,0,0],[0,0,4,0,0,0,4,0,0],[0,0,4,4,4,4,4,0,0],[0,0,0,0,0,0,0,0,0],[0,0,0,1,1,1,1,0,0]]},{"input":[[0,0,0,0,0,0,0,0,0,0],[0,0,0,5,5,5,5,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,0,5,5,5,5,0,0,0],[0,0,0,5,5,5,5,0,0,0],[0,0,0,5,5,5,5,0,0,0],[0,0,0,5,5,5,5,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0,0]],"output":[[0,0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,0,5,5,5,5,0,0,0],[0,0,0,5,5,5,5,0,0,0],[0,0,0,5,5,5,5,0,0,0],[0,0,0,5,5,5,5,0,0,0],[0,0,0,5,5,5,5,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0,0],[0,0,0,0,0,0,0,0,0,0]]}]}