dev

main.py

@@ -2,6 +2,7 @@ import gradio as gr
 import shutil
 import subprocess
 import tempfile
+import os
 import sys
 
 def new_binary(file):
@@ -11,7 +12,7 @@ def new_binary(file):
 
         # 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)
+        output = subprocess.check_output(f"python /DIRTY/dataset-gen-ghidra/generate.py --ghidra /ghidra/support/analyzeHeadless -t 1 -b {TEMP_DIR} -o {OUTPUT_DIR} 2>&1", shell=True)
         print(output, file=sys.stderr)
 
 with gr.Blocks() as demo:
@@ -52,5 +53,8 @@ with gr.Blocks() as demo:
         
     file_widget.change(file_change_fn, file_widget)
 
+# spaces only shows stderr..
+os.dup2(sys.stdout.fileno(), sys.stderr.fileno())
+
 demo.queue()
 demo.launch(server_name="0.0.0.0", server_port=7860)