Create app.py

app.py

@@ -0,0 +1,91 @@
+import json
+import os
+import numpy as np
+import matplotlib.pyplot as plt
+import gradio as gr
+
+# Update this to the correct path of your folder
+task_folder = "./ARC_AGI"  # Use absolute path if necessary
+
+def plot_arc_problem(task_file_path, data_type="train"):
+    """
+    Visualize input-output pairs for a given ARC task file.
+    Args:
+        task_file_path (str): Path to the ARC task JSON file.
+        data_type (str): 'train' or 'test' to visualize respective examples.
+    Returns:
+        matplotlib.figure.Figure or str: The plotted figure or an error message.
+    """
+    try:
+        # Load the JSON file
+        with open(task_file_path, 'r') as f:
+            task = json.load(f)
+
+        # Get the data type (train or test) pairs
+        pairs = task.get(data_type, [])
+        if not pairs:  # Check if the section exists and has data
+            return f"No '{data_type}' data found in the selected file."
+
+        # Create a figure with subplots for each pair
+        fig, axes = plt.subplots(len(pairs), 2, figsize=(10, 5 * len(pairs)))
+        fig.suptitle(f"ARC Task: {os.path.basename(task_file_path)} ({data_type.capitalize()})", fontsize=16)
+
+        # Handle case where there is only one pair
+        if len(pairs) == 1:
+            axes = [axes]
+
+        for idx, pair in enumerate(pairs):
+            input_grid = np.array(pair['input'])
+            output_grid = np.array(pair['output'])
+
+            # Plot input grid
+            axes[idx][0].imshow(input_grid, cmap='tab20', interpolation='none')
+            axes[idx][0].set_title(f"Input Pair {idx + 1} ({data_type.capitalize()})")
+            axes[idx][0].axis('off')
+
+            # Plot output grid
+            axes[idx][1].imshow(output_grid, cmap='tab20', interpolation='none')
+            axes[idx][1].set_title(f"Output Pair {idx + 1} ({data_type.capitalize()})")
+            axes[idx][1].axis('off')
+
+        plt.tight_layout(rect=[0, 0, 1, 0.96])
+        return fig
+    except Exception as e:
+        print("Error in plot_arc_problem:", e)
+        return f"An error occurred while plotting the {data_type} data: {e}"
+
+def visualize_task(file_name, data_type):
+    """
+    Load and visualize the ARC task for a given file and data type.
+    Args:
+        file_name (str): Name of the JSON file in the folder.
+        data_type (str): 'train' or 'test'.
+    Returns:
+        matplotlib.figure.Figure or str: Figure of the visualized task or an error message.
+    """
+    try:
+        print(f"Selected file: {file_name}, Data type: {data_type}")  # Debugging
+        task_file_path = os.path.join(task_folder, file_name)
+        result = plot_arc_problem(task_file_path, data_type)
+        return result
+    except Exception as e:
+        print("Error in visualize_task:", e)
+        return f"An error occurred while visualizing the task: {e}"
+
+# Gradio Interface
+task_files = [f for f in os.listdir(task_folder) if f.endswith('.json')]
+
+interface = gr.Interface(
+    fn=visualize_task,
+    inputs=[
+        gr.Dropdown(choices=task_files, label="Select ARC Task File"),
+        gr.Radio(choices=["train", "test"], label="Select Data Type to Visualize", value="train")
+    ],
+    outputs="plot",
+    title="ARC Task Visualizer",
+    description="Select a task file and data type (train or test) to visualize its input-output grids."
+)
+
+if __name__ == "__main__":
+    print("Task files:", task_files)  # Debugging
+    interface.launch()