Update for multiple predictions

Dockerfile

@@ -27,7 +27,7 @@ RUN unzip ghidrathon/Ghidrathon-v4.0.0.zip -d /ghidra/Ghidra/Extensions
 
 WORKDIR /
 
-RUN git clone -b main https://github.com/edmcman/DIRTY # 09-25-24
+RUN git clone -b nonbest https://github.com/edmcman/DIRTY # 10-3-24 3.3
 #ADD ./DIRTY /DIRTY
 
 RUN --mount=type=cache,target=/root/.cache pip install --break-system-packages --upgrade -r /DIRTY/requirements.txt

main.py

@@ -184,14 +184,16 @@ with gr.Blocks() as demo:
             #print(json_info)
 
             # group by location
-            src_filtered = json_info['other_info']['source_filtered']
+            src_filtered = json_info['other_info']['example_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
+                'model_output_multi': json_info["model_output_multi"],
+                'dup_location_vars': src_filtered,
+                'other_outputs': json_info['other_info']['other_outputs']
             }
 
             return {