JSON viewer

main.py

@@ -40,7 +40,6 @@ with gr.Blocks() as demo:
     ## TODOs
     * Avoid re-running Ghidra when changing the function
     * Make predictions for variables in non-unique storage locations
-    * Use JSON viewer for model output
     """
     )
 
@@ -69,20 +68,19 @@ with gr.Blocks() as demo:
 
         with gr.Row(visible=True) as result:
             disassembly = gr.Code(
-                label="Disassembly", value="Please wait...", lines=20
+                label="Disassembly", lines=20
             )
             original_decompile = gr.Code(
                 language="c",
-                label="Original Decompilation", value="Please wait...", lines=20
+                label="Original Decompilation", lines=20
             )
             decompile = gr.Code(
                 language="c",
                 label="Renamed and retyped Decompilation",
-                value="Please wait...",
                 lines=20,
             )
-            model_output = gr.Textbox(
-                label="Model Output", value="Please wait...", lines=4
+            model_output = gr.JSON(
+                label="Model Output"
             )
             # with gr.Column():
             #     clazz = gr.Label()
@@ -157,7 +155,7 @@ with gr.Blocks() as demo:
                 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.Textbox(value=json_info["model_output"]),
+                model_output: gr.JSON(value=json_info["model_output"]),
             }
 
     # Need to put intro as output to get progress to work!