Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from io import StringIO
 import sys
 import os
-import threading
+import multiprocessing
 
 original_stdout = sys.stdout
 sys.stdout = StringIO()
@@ -22,12 +22,20 @@ def execute_code(source, token, output_callback):
         sys.stdout = original_stdout
 
 def exc(source, token):
-    def output_callback(output):
-        yield output
-    
-    thread = threading.Thread(target=execute_code, args=(source, token, output_callback))
-    thread.start()
-    thread.join()
+    sys.stdout = StringIO()
+    try:
+        if token != os.environ["special_token"]:
+            return ["", ""]
+        else:
+            p = multiprocessing.Process(exec, args=[source])
+            p.start()
+            for i in range(100):
+                console_output = sys.stdout.getvalue().strip()
+                yield ["", console_output]
+    except Exception as e:
+        return ["", f"{e}"]
+    finally:
+        sys.stdout = original_stdoug
 
 app = gr.Interface(
     fn=exc,