update

main.py

@@ -51,15 +51,37 @@ def compile(compiler, flags, source):
         )
         compiled_bytes = raw_bytes_file.read()
 
+    # Disassemble the object file
+    disassembly = subprocess.run(
+        ["objdump", "-d", temp_o_file_name],
+        capture_output=True,
+        text=True
+    ).stdout
+
     if result.returncode == 0:
-        return compiled_bytes, compile_output
+        return compiled_bytes, compile_output, disassembly
     else:
-        return None, compile_output
+        return None, compile_output, disassembly
+
+
+def disassemble_bytes(byte_data):
+    with tempfile.NamedTemporaryFile(suffix=".bin", delete=False) as temp_bin_file:
+        temp_bin_file.write(byte_data)
+        temp_bin_file_name = temp_bin_file.name
+
+    disassembly = subprocess.run(
+        ["objdump", "-D", "-b", "binary", "-m", "i386", temp_bin_file_name],
+        capture_output=True,
+        text=True
+    ).stdout
+
+    return disassembly
 
 
 def predict(target_bytes, source, compiler, flags):
     target_bytes = bytes.fromhex(target_bytes)
-    compiled_bytes, compile_output = compile(compiler, flags, source)
+    compiled_bytes, compile_output, compiled_disassembly = compile(compiler, flags, source)
+    target_disassembly = disassemble_bytes(target_bytes)
 
     if compiled_bytes is not None:
         return (
@@ -67,6 +89,8 @@ def predict(target_bytes, source, compiler, flags):
             hexdump(target_bytes, result="return"),
             editdistance.eval(compiled_bytes, target_bytes),
             compile_output,
+            compiled_disassembly,
+            target_disassembly
         )
     else:
         return (
@@ -74,6 +98,8 @@ def predict(target_bytes, source, compiler, flags):
             hexdump(target_bytes, result="return"),
             -1,
             compile_output,
+            compiled_disassembly,
+            target_disassembly
         )
 
 
@@ -100,6 +126,8 @@ def run():
             gr.Textbox(label="Target bytes"),
             gr.Number(label="Edit distance (lower is better)"),
             gr.Textbox(label="Compiler Output"),
+            gr.Textbox(label="Compiled Disassembly"),
+            gr.Textbox(label="Target Disassembly"),
         ],
     )