Improve dup location var info

main.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import subprocess
 import tempfile
+import itertools
 import os
 import sys
 import hashlib
@@ -163,18 +164,32 @@ with gr.Blocks() as demo:
             if not os.path.exists(f"{TEMP_DIR}/funcs.json"):
                 raise gr.Error(f"DIRTY Ghidra failed to produce output: {o.stdout}")
 
-            json_info = json.load(open(f"{TEMP_DIR}/funcs.json"))
+            try:
+                json_info = json.load(open(f"{TEMP_DIR}/funcs.json"))
+            except Exception as e:
+                raise gr.Error(f"Unable to parse DIRTY Ghidra output: {e}\n{o.stdout}")
 
             if "exception" in json_info:
                 raise gr.Error(f"DIRTY Ghidra failed: {json_info['exception']}")
 
             #print(json_info)
 
+            # group by location
+            src_filtered = json_info['other_info']['source_filtered']
+            keyfunc = lambda x: x[1]
+            src_filtered = sorted(src_filtered.items(), key=keyfunc)
+            src_filtered = {k: [v1 for v1, v2 in v] for k, v in itertools.groupby(src_filtered, keyfunc)}
+
+            model_output_info = {
+                'model_output': json_info["model_output"],
+                'dup_location_vars': src_filtered
+            }
+
             return {
                 disassembly: gr.Textbox(value=json_info["disassembly"]),
                 original_decompile: gr.Textbox(value=json_info["original_decompile"]),
                 decompile: gr.Textbox(value=json_info["decompile"]),
-                model_output: gr.JSON(value=json_info["model_output"]),
+                model_output: gr.JSON(value=json.dumps(model_output_info)),
             }
 
     # Need to put intro as output to get progress to work!