in output there is code copy in output, so llm will understand better ig

app.py

@@ -1,54 +1,51 @@
 import gradio as gr
 import subprocess
+import tempfile
 
-def run_code(code, inputs, packages):
-    try:
-        # Create a temporary Python script file
-        with open('temp_script.py', 'w') as script_file:
-            script_file.write(code)
-        
-        # Install packages if specified
-        installed_packages = []
-        if packages:
-            for package in packages.split(','):
-                package = package.strip()
-                subprocess.check_call(['pip', 'install', package])
-                installed_packages.append(package)
-        
-        # Execute the script file with inputs and capture the output and error
-        input_values = '\n'.join(inputs.split(','))
+
+class CodeExecutor:
+    def __init__(self):
+        self.temp_dir = tempfile.TemporaryDirectory()
+
+    def execute(self, code, inputs):
+        input_file = f"{self.temp_dir.name}/input.txt"
+        with open(input_file, "w") as f:
+            f.write(inputs)
+
+        output_file = f"{self.temp_dir.name}/output.txt"
         try:
-            output = subprocess.check_output(['python', 'temp_script.py'], input=input_values.encode(), stderr=subprocess.STDOUT, timeout=5)
-            output = output.decode('utf-8')
-            error = ''
-        except subprocess.CalledProcessError as cpe:
-            output = cpe.output.decode('utf-8')
-            error = f'Error: {output}'
+            with open(output_file, "w") as f:
+                subprocess.run(
+                    ["python", "-c", code], check=True, timeout=5, stdin=open(input_file, "r"), stdout=f, stderr=f
+                )
+            with open(output_file, "r") as f:
+                output = f.read()
+        except subprocess.CalledProcessError as e:
+            output = e.output.decode("utf-8")
         except subprocess.TimeoutExpired:
-            output = ''
-            error = 'Error: Execution timed out after 5 seconds'
-    except Exception as e:
-        output = ''
-        error = str(e)
-    finally:
-        # Uninstall packages if they were installed
-        if installed_packages:
-            subprocess.check_call(['pip', 'uninstall', '-y'] + installed_packages)  # Added closing parenthesis here
-            # subprocess.check_call(['pip', 'uninstall', '-y'] + installed_packages) can also be written as 
-            # subprocess.check_call(['pip', 'uninstall', '-y', *installed_packages]) 
-
-    return output, error
-
-demo = gr.Blocks()
-with demo:
-    gr.Markdown("# Code Interpreter")
-    gr.Markdown("Enter your Python code, inputs separated by commas, and optionally specify packages to install")
-    code_input = gr.Code(language="python", label="Code")
-    inputs_input = gr.Textbox(label="Inputs (separated by commas)", placeholder="e.g. 1,2,3")
-    packages_input = gr.Textbox(label="Packages (optional, separated by commas)", placeholder="e.g. numpy,pandas")
-    output_text = gr.Textbox(label="Output")
-    error_text = gr.Textbox(label="Error")
-
-    gr.Button("Run").click(fn=run_code, inputs=[code_input, inputs_input, packages_input], outputs=[output_text, error_text])
-
-demo.launch()
+            output = "Execution timed out."
+        except Exception as e:
+            output = str(e)
+
+        return f"User's Code:\n```python\n{code}\n```\n\nOutput:\n{output}"
+
+
+def create_interface():
+    with gr.Blocks() as demo:
+        gr.Markdown("# Code Interpreter")
+        gr.Markdown("Enter Python code and inputs to execute.")
+        code_input = gr.Code(language="python", label="Code")
+        inputs_input = gr.Textbox(label="Inputs")
+        output_text = gr.Textbox(label="Output", lines=10)
+        run_button = gr.Button("Run")
+        run_button.click(
+            CodeExecutor().execute, inputs=[code_input, inputs_input], outputs=output_text
+        )
+        demo.inputs = [code_input, inputs_input]
+        demo.outputs = [output_text]
+    return demo
+
+
+if __name__ == "__main__":
+    app = create_interface()
+    app.launch()