update

main.py

@@ -2,25 +2,46 @@ import gradio as gr
 import subprocess
 import tempfile
 
+description = """This is a space testing a method for evaluating the quality of decompilation.
+
+Currently unhandled features:
+* Global references
+* Function calls
+* Compiler and flag selection
+"""
+
+
 def compile(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())
         temp_c_file_name = temp_c_file.name
-    
+
     # Create a temporary file for the object file
     with tempfile.NamedTemporaryFile(suffix=".o", delete=False) as temp_o_file:
         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)
-    
-    # Read the compiled object file and load the bytes into a bytestring
-    with open(temp_o_file_name, "rb") as f:
-        compiled_bytes = f.read()
-    
+
+    # Create a temporary file for the raw bytes
+    with tempfile.NamedTemporaryFile(suffix=".raw", delete=True) as raw_bytes_file:
+        subprocess.run(
+            [
+                "objcopy",
+                "--only-section",
+                ".text",
+                "-O",
+                "binary",
+                temp_o_file_name,
+                raw_bytes_file.name,
+            ]
+        )
+        compiled_bytes = raw_bytes_file.read()
+
     return compiled_bytes
 
+
 def predict(source):
     bytes = compile(source)
     return bytes, 1.0
@@ -29,10 +50,12 @@ def predict(source):
 def run():
     demo = gr.Interface(
         fn=predict,
+        description=description,
         inputs=gr.Textbox(lines=10, label="Decompiled C Source Code"),
         outputs=[gr.Textbox(), gr.Number()],
     )
 
     demo.launch(server_name="0.0.0.0", server_port=7860)
 
-run()
+
+run()