Update tests.py

tests.py

@@ -208,12 +208,23 @@ def create_code_files(filename: str, code: str) -> dict:
     f.write(code)
     f.close()
     return {"info":"task completed. The referenced code files were created successfully. "}
-
+    
 @mcp.tool()
 def run_code(python_packages:str,filename: str, code: str,start_cmd:str) -> dict:
-    """python_packages:python_packages to install seperated by space.eg-(python packages:numpy matplotlib).The following python packages are preinstalled:gradio XlsxWriter openpyxl
-    filename,code:Create code files by passing the the filename as well the entire code to write.The file is created by default in the /app/code_interpreter directory.Note:All user uploaded files that you might need to work upon are stored in the /app/code_interpreter directory.
-    start_cmd:Example-  python /app/code_interpreter/app.py or bash /app/code_interpreter/app.py.The files must be inside the /app/code_interpreter directory.
+    """  
+    Execute Python code in a controlled environment with package installation and file handling.
+
+    Args:
+        python_packages: Space-separated list of packages to install (e.g., "numpy matplotlib").
+                        Preinstalled packages: gradio, XlsxWriter, openpyxl.
+        filename:        Name of the file to create (stored in /app/code_interpreter/).
+        code:           Full Python 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").
+
+    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.
     """
 
     global files_list
@@ -236,7 +247,6 @@ def run_code(python_packages:str,filename: str, code: str,start_cmd:str) -> dict
     run(start_cmd, 300)
     while stderr=="" and stdout=="":
         pass
-    time.sleep(1.5)
     onlyfiles = glob.glob("/app/code_interpreter/*")
     onlyfiles=list(set(onlyfiles)-set(files_list))
     uploaded_filenames=[]