update

main.py

@@ -1,5 +1,6 @@
 from hexdump2 import hexdump
 import gradio as gr
+import shlex
 import subprocess
 import tempfile
 
@@ -12,7 +13,7 @@ Currently unhandled features:
 """
 
 
-def compile(source):
+def compile(compiler, flags, source):
     # Create a temporary file for the C source code
     with tempfile.NamedTemporaryFile(suffix=".c", delete=False) as temp_c_file:
         temp_c_file.write(source.encode())
@@ -23,7 +24,7 @@ def compile(source):
         temp_o_file_name = temp_o_file.name
 
     # Compile the C file to an object file
-    subprocess.run(["cc", "-c", temp_c_file_name, "-o", temp_o_file_name], check=True)
+    subprocess.run([compiler, "-c", temp_c_file_name] + shlex.split(flags) + ["-o", temp_o_file_name], check=True)
 
     # Create a temporary file for the raw bytes
     with tempfile.NamedTemporaryFile(suffix=".raw", delete=True) as raw_bytes_file:
@@ -43,8 +44,8 @@ def compile(source):
     return compiled_bytes
 
 
-def predict(target, source):
-    bytes = compile(source)
+def predict(target, source, compiler, flags):
+    bytes = compile(compiler, flags, source)
     return hexdump(bytes), 1.0