Add visualization code

github_resolver/io_utils.py

@@ -0,0 +1,11 @@
+import json
+from github_resolver.resolver_output import ResolverOutput
+
+
+def load_resolver_output(output_jsonl: str, issue_number: int) -> ResolverOutput:
+    with open(output_jsonl, "r") as f:
+        for line in f:
+            data = json.loads(line)
+            if data["issue"]["number"] == issue_number:
+                return ResolverOutput.model_validate(data)
+    raise ValueError(f"Issue number {issue_number} not found in {output_jsonl}")

github_resolver/send_pull_request.py

@@ -1,23 +1,13 @@
 import argparse
 import os
-import json
 import shutil
 from github_resolver.github_issue import GithubIssue
-from github_resolver.resolver_output import ResolverOutput
+from github_resolver.io_utils import load_resolver_output
 import requests
 import subprocess
 import whatthepatch
 
 
-def load_resolver_output(output_jsonl: str, issue_number: int) -> ResolverOutput:
-    with open(output_jsonl, 'r') as f:
-        for line in f:
-            data = json.loads(line)
-            if data['issue']['number'] == issue_number:
-                return ResolverOutput.model_validate(data)
-    raise ValueError(f"Issue number {issue_number} not found in {output_jsonl}")
-
-
 def apply_patch(repo_dir: str, patch: str) -> None:
     diffs = whatthepatch.parse_patch(patch)
 

github_resolver/utils.py

@@ -12,7 +12,6 @@
 from openhands.events.action import Action
 from openhands.events.action.message import MessageAction
 
-
 def codeact_user_response(
     state: State,
     encapsulate_solution: bool = False,

github_resolver/visualize_resolver_output.py

@@ -0,0 +1,42 @@
+import argparse
+import os
+from github_resolver.io_utils import load_resolver_output
+
+
+def visualize_resolver_output(issue_number: int, output_dir: str, vis_method: str):
+    output_jsonl = os.path.join(output_dir, "output.jsonl")
+    resolver_output = load_resolver_output(output_jsonl, issue_number)
+    if vis_method == "json":
+        print(resolver_output.model_dump_json(indent=4))
+    else:
+        raise ValueError(f"Invalid visualization method: {vis_method}")
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Visualize a patch.")
+    parser.add_argument(
+        "--issue-number",
+        type=int,
+        required=True,
+        help="Issue number to send the pull request for.",
+    )
+    parser.add_argument(
+        "--output-dir",
+        type=str,
+        default="output",
+        help="Output directory to write the results.",
+    )
+    parser.add_argument(
+        "--vis-method",
+        type=str,
+        default="json",
+        choices=["json"],
+        help="Method to visualize the patch [json].",
+    )
+    my_args = parser.parse_args()
+
+    visualize_resolver_output(
+        issue_number=my_args.issue_number,
+        output_dir=my_args.output_dir,
+        vis_method=my_args.vis_method,
+    )

pyproject.toml

@@ -16,6 +16,7 @@ whatthepatch = "*"
 [tool.poetry.group.dev.dependencies]
 mypy = "*"
 ruff = "*"
+black = "*"
 types-requests = "*"
 types-whatthepatch = "*"