run decompiler

main.py

@@ -149,9 +149,32 @@ This is a space to experiment with our quantized 22B neural model for decompilat
                         "addrs": addrs},
             }
         
-    #file_widget.change(
-    #    file_change_fn, file_widget, outputs=[intro, state, col, fun_dropdown]
-    #)
+    @fun_dropdown.change(inputs=[fun_dropdown, state], outputs=[disassembly, original_decompile, decompile, model_output])
+    def function_change_fn(selected_fun, state, progress=gr.Progress()):
+
+        with tempfile.TemporaryDirectory() as TEMP_DIR:
+
+            progress(0, desc=f"Running GhidraFunctionCPPExporter {hex(selected_fun)}...")
+
+            o = subprocess.run(["/code/GhidraFunctionCPPExporter/export.bash", state['file'], "address_set_str", str(selected_fun), "output_dir", TEMP_DIR], shell=False, capture_output=True, encoding="utf8")
+
+            print(o.stdout)
+            print(o.stderr)
+
+            if (o.returncode != 0):
+                raise Exception(f"Ghidra export failed with return code {o.returncode}: {o.stderr}")
+
+            # Show files in TEMP_DIR
+            print("Files in TEMP_DIR:")
+            for f in os.listdir(TEMP_DIR):
+                print(f)
+
+            return {
+                disassembly: gr.Textbox(value=""),
+                original_decompile: gr.Textbox(value=""),
+                decompile: gr.Textbox(value=""),
+                model_output: gr.JSON(value=None),
+            }
 
 print("Lauching now.")