Update app.py

app.py

@@ -13,13 +13,18 @@ class CodeExecutor:
             f.write(inputs)
 
         output_file = f"{self.temp_dir.name}/output.txt"
+        error_file = f"{self.temp_dir.name}/error.txt"
         try:
-            with open(output_file, "w") as f:
+            with open(output_file, "w") as f, open(error_file, "w") as e:
                 subprocess.run(
-                    ["python", "-c", code], check=True, timeout=5, stdin=open(input_file, "r"), stdout=f, stderr=f
+                    ["python", "-c", code], check=True, timeout=5, stdin=open(input_file, "r"), stdout=f, stderr=e
                 )
             with open(output_file, "r") as f:
                 output = f.read()
+            with open(error_file, "r") as f:
+                error = f.read()
+            if error:
+                output = f"Error:\n{error}"
         except subprocess.CalledProcessError as e:
             output = e.output.decode("utf-8")
         except subprocess.TimeoutExpired:
@@ -36,7 +41,7 @@ def create_interface():
         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)
+        output_text = gr.Textbox(label="Output", lines=20)
         run_button = gr.Button("Run")
         run_button.click(
             CodeExecutor().execute, inputs=[code_input, inputs_input], outputs=output_text