update

main.py

@@ -27,12 +27,12 @@ def compile(compiler, flags, source):
         temp_o_file_name = temp_o_file.name
 
     # Compile the C file to an object file
-    subprocess.run(
-        [compiler, "-c", temp_c_file_name]
-        + shlex.split(flags)
-        + ["-o", temp_o_file_name],
-        check=True,
+    result = subprocess.run(
+        [compiler, "-c", temp_c_file_name] + shlex.split(flags) + ["-o", temp_o_file_name],
+        capture_output=True,
+        text=True
     )
+    compile_output = result.stdout + result.stderr
 
     # Create a temporary file for the raw bytes
     with tempfile.NamedTemporaryFile(suffix=".raw", delete=True) as raw_bytes_file:
@@ -48,18 +48,19 @@ def compile(compiler, flags, source):
             ]
         )
         compiled_bytes = raw_bytes_file.read()
-        print(compiled_bytes)
 
-    return compiled_bytes
+    print(compiled_bytes)
+    return compiled_bytes, compile_output
 
 
 def predict(target_bytes, source, compiler, flags):
     target_bytes = bytes.fromhex(target_bytes)
-    compiled_bytes = compile(compiler, flags, source)
+    compiled_bytes, compile_output = compile(compiler, flags, source)
     return (
         hexdump(compiled_bytes, result="return"),
         hexdump(target_bytes, result="return"),
-        editdistance.eval(compiled_bytes, target_bytes)
+        editdistance.eval(compiled_bytes, target_bytes),
+        compile_output
     )
 
 
@@ -85,6 +86,7 @@ def run():
             gr.Textbox(label="Compiled bytes"),
             gr.Textbox(label="Target bytes"),
             gr.Number(label="Edit distance (lower is better)"),
+            gr.Textbox(label="Compiler Output"),
         ],
     )