Update app.py

app.py

@@ -1,16 +1,25 @@
 import gradio as gr
+import sys
+from io import StringIO
 
 def run_code(code):
     try:
+        # Redirect stdout to a buffer
+        output_buffer = StringIO()
+        sys.stdout = output_buffer
+
         # Execute the code in a separate namespace
         namespace = {}
         exec(code, namespace)
-        # Capture the output
-        output = namespace.get('output', None)
-        if output is None:
-            output = "No output captured."
+
+        # Get the output from the buffer
+        output = output_buffer.getvalue()
     except Exception as e:
         output = f"Error: {str(e)}"
+    finally:
+        # Reset stdout to its original value
+        sys.stdout = sys.__stdout__
+
     return output
 
 # Define the Gradio interface