Update tests.py

tests.py

@@ -15,7 +15,6 @@ from google import genai
 import pexpect
 
 client = genai.Client(api_key="AIzaSyDtP05TyoIy9j0uPL7_wLEhgQEE75AZQSc")
-
 source_dir = "/app/uploads/temp"
 destination_dir = "/app/code_interpreter"
 files_list=[]
@@ -31,6 +30,7 @@ import requests
 import os
 from bs4 import BeautifulSoup  # For parsing HTML
 
+Parent=pexpect.spawn('bash')
 
 def download_all_files(base_url, files_endpoint, download_directory):
     """Downloads all files listed on the server's /upload page."""
@@ -140,6 +140,15 @@ from requests_futures.sessions import FuturesSession
 session = FuturesSession()
 
 def run(cmd, timeout_sec,forever_cmd):
+    global Parent
+    if forever_cmd == 'true':
+        Parent.close()
+        Parent = pexpect.spawn("bash")
+        command="cd /app/code_interpreter/ && "+cmd
+
+        Parent.sendline(command)
+        Parent.readline.decode()
+        return str(Parent.readline.decode())       
     t=time.time()
     child = pexpect.spawn("bash")
     output=""
@@ -231,7 +240,7 @@ def run_code(python_packages:str,filename: str, code: str,start_cmd:str,forever_
         code:           Full code to write to the file.
         start_cmd:      Command to execute the file (e.g., "python /app/code_interpreter/app.py" 
                         or "bash /app/code_interpreter/app.py").
-        forever_cmd:    If 'true', the command will run indefinitely.Set to 'true', when runnig a website/server. If 'false', the command will time out after 300 second and the result will be returned.
+        forever_cmd:    If 'true', the command will run indefinitely.Set to 'true', when runnig a website/server.Run all servers/website on port 1337. If 'false', the command will time out after 300 second and the result will be returned.
     Notes:
         - All user-uploaded files are in /app/code_interpreter/.
         - After execution, embed a download link (or display images/gifs/videos directly in markdown format) in your response.
@@ -418,4 +427,5 @@ def deepthinking3(query:str,info:str) -> dict:
 
 if __name__ == "__main__":
     # Initialize and run the server
+    pexpect.spawn('ngrok http 1337')
     mcp.run(transport='stdio')