dev

main.py

@@ -2,7 +2,7 @@ import gradio as gr
 import shutil
 import subprocess
 import tempfile
-
+import sys
 
 def new_binary(file):
 
@@ -10,10 +10,9 @@ def new_binary(file):
         shutil.copy2(file.name, TEMP_DIR)
 
         # python3 generate.py --ghidra PATH_TO_GHIDRA_ANALYZEHEADLESS -t NUM_THREADS -n [NUM_FILES|None] -b BINARIES_DIR -o OUTPUT_DIR
+        print("Running DIRTY-Ghidra...", file=sys.stderr)
         output = subprocess.check_output(f"python /DIRTY/dataset-gen-ghidra/generate.py --ghidra /ghidra/support/analyzeHeadless -t 1 -b {TEMP_DIR} -o {OUTPUT_DIR}", shell=True)
-        print(output)
-
-
+        print(output, file=sys.stderr)
 
 with gr.Blocks() as demo:
 
@@ -30,8 +29,6 @@ with gr.Blocks() as demo:
 
     def file_change_fn(file, progress=gr.Progress()):
 
-        print("hi")
-        new_binary(file)
 
         if file is None:
             return {
@@ -40,6 +37,8 @@ with gr.Blocks() as demo:
         else:
 
             #fun_data = {42: 2, 43: 3}
+            print("hi")
+            new_binary(file)
             progress(0, desc="Decompiling binary...")
             #fun_data = get_all_dis(file.name)