Update tests.py

tests.py

@@ -9,7 +9,8 @@ import openpyxl
 import shutil
 from google import genai
 import pexpect
-
+stdout=""
+stderr=""
 client = genai.Client(api_key="AIzaSyDtP05TyoIy9j0uPL7_wLEhgQEE75AZQSc")
 source_dir = "/app/uploads/temp"
 destination_dir = "/app/code_interpreter"
@@ -93,6 +94,19 @@ TOKEN = "5182224145:AAEjkSlPqV-Q3rH8A9X8HfCDYYEQ44v_qy0"
 chat_id = "5075390513"
 from requests_futures.sessions import FuturesSession
 session = FuturesSession()
+from subprocess import Popen, PIPE
+import shlex
+from threading import Timer
+def run2(cmd, timeout_sec,forever_cmd):
+    global stdout 
+    global stderr
+    proc = Popen(shlex.split(cmd), stdout=PIPE, stderr=PIPE,cwd="/app/code_interpreter/")
+    timer = Timer(timeout_sec, proc.kill)
+    try:
+        timer.start()
+        stdout, stderr = proc.communicate()
+    finally:
+        timer.cancel()
 
 def run(cmd, timeout_sec,forever_cmd):
     global Parent
@@ -108,16 +122,10 @@ def run(cmd, timeout_sec,forever_cmd):
     child = pexpect.spawn("bash")
     output=""
     command="cd /app/code_interpreter/ && "+cmd
-    child.sendline('su root')
-    print(child.readline().decode())
-    child.sendline('password')
-    print(child.readline().decode())
-    print(child.readline().decode())
 
     child.sendline('PROMPT_COMMAND="echo END"')
     child.readline().decode()
     child.readline().decode()
-    child.readline().decode()
 
     child.sendline(command)
 
@@ -317,9 +325,12 @@ def run_shell_command(cmd:str,forever_cmd:str) -> dict:
               - ``'output'`` (str): The captured standard output (stdout) and potentially
                 standard error (stderr) from the command.
     """    
+    global stderr
+    global stdout
     transfer_files()
     transfer_files2()
-    output=run(cmd, 300,forever_cmd)
+    run2(cmd, 300,forever_cmd)
+    output = str(stdout)+str(stderr)
     return {"output":output}