update

main.py

@@ -28,9 +28,11 @@ def compile(compiler, flags, source):
 
     # Compile the C file to an object file
     result = subprocess.run(
-        [compiler, "-c", temp_c_file_name] + shlex.split(flags) + ["-o", temp_o_file_name],
+        [compiler, "-c", temp_c_file_name]
+        + shlex.split(flags)
+        + ["-o", temp_o_file_name],
         capture_output=True,
-        text=True
+        text=True,
     )
     compile_output = result.stdout + result.stderr
 
@@ -49,19 +51,30 @@ def compile(compiler, flags, source):
         )
         compiled_bytes = raw_bytes_file.read()
 
-    print(compiled_bytes)
-    return compiled_bytes, compile_output
+    if result.returncode == 0:
+        return compiled_bytes, compile_output
+    else:
+        return None, compile_output
 
 
 def predict(target_bytes, source, compiler, flags):
     target_bytes = bytes.fromhex(target_bytes)
     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),
-        compile_output
-    )
+
+    if compiled_bytes is not None:
+        return (
+            hexdump(compiled_bytes, result="return"),
+            hexdump(target_bytes, result="return"),
+            editdistance.eval(compiled_bytes, target_bytes),
+            compile_output,
+        )
+    else:
+        return (
+            "Compilation failed",
+            hexdump(target_bytes, result="return"),
+            -1,
+            compile_output,
+        )
 
 
 def run():