Update app.py

app.py

@@ -1,8 +1,25 @@
 import gradio as gr
+from io import StringIO
+import sys
+
+original_stdout = sys.stdout
+sys.stdout = StringIO()
+
+def exc(source, token):
+    if token != os.environ["special_token"]:
+        return "Token Incorrect"
+    else:
+        consoleop = ""
+        try:
+            op = exec(source=source)
+            consoleop = sys.stdout.getvalue().strip()
+            return f"""['''ExecOP: {op}''', '''ConsoleOP: {consoleop}''']"""
+        except Exception as e:
+            return f"{e}"
 
 app = gr.Interface(
-    fn=exec,
-    inputs=["text"],
+    fn=exc,
+    inputs=["text", "text"],
     outputs=["text"],
 )