Update tests.py

tests.py

@@ -115,7 +115,7 @@ def run(cmd, timeout_sec,forever_cmd):
 
     child.sendline(command)
 
-    while (not child.eof() ) and (time.time()-t<120):
+    while (not child.eof() ) and (time.time()-t<timeout_sec):
         x=child.readline().decode()
         output=output+x
         print(x)
@@ -222,7 +222,7 @@ def run_code(language:str,packages:str,filename: str, code: str,start_cmd:str,fo
     Execute code in a controlled environment with package installation and file handling.
     Args:
         language:Programming language of the code (eg:"python", "nodejs", "bash","html",etc).
-        packages: Space-separated list of packages to install (e.g., "numpy matplotlib").
+        packages: Space-separated list of packages to install.(python packages are installed if language set to python and npm packages are installed if language set to nodejs).
                   Preinstalled python packages: gradio, XlsxWriter, openpyxl , mpxj , jpype1.
                   Preinstalled npm packages: express, ejs, chart.js.
         filename:Name of the file to create (stored in /app/code_interpreter/).
@@ -248,9 +248,11 @@ def run_code(language:str,packages:str,filename: str, code: str,start_cmd:str,fo
     else:
         command="ls"
     if  packages != "" or packages != " ":
-        stdot=run(
+        package_logs=run(
             f"{command} {package_names}", timeout_sec=300,forever_cmd= 'false' 
         )
+        if "ERROR" in package_logs:
+            return {"package_installation_log":package_logs,"info":"Package installation failed. Please check the package names. Tip:Try using another package/method to accomplish the task."}
     transfer_files2()
     transfer_files()
     f = open(os.path.join(destination_dir, filename), "w")
@@ -258,7 +260,7 @@ def run_code(language:str,packages:str,filename: str, code: str,start_cmd:str,fo
     f.close()
     global files_list
     if start_cmd != "" or start_cmd != " ":
-        stdot=run(start_cmd, 300,forever_cmd)
+        stdot=run(start_cmd, 120,forever_cmd)
     else:
         stdot="File created successfully."
     onlyfiles = glob.glob("/app/code_interpreter/*")