Update tests.py

tests.py

@@ -200,15 +200,54 @@ def analyse_images(imagepath,query) -> dict:
 
 @mcp.tool()
 def create_code_files(filename: str, code: str) -> dict:
+    """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."""
     global destination_dir
     download_all_files("https://opengpt-4ik5.onrender.com", "/upload", "/app/code_interpreter")
-    """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."""
     transfer_files()
     f = open(os.path.join(destination_dir, filename), "w")
     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.
+    """
+
+    global files_list
+    global stdout
+    global stderr
+    package_names = python_packages.strip()
+    command="pip install"
+    if not package_names:
+        return
+
+    run(
+        f"{command} --break-system-packages {package_names}", timeout_sec=300
+    )
+    global destination_dir
+    download_all_files("https://opengpt-4ik5.onrender.com", "/upload", "/app/code_interpreter")
+    transfer_files()
+    f = open(os.path.join(destination_dir, filename), "w")
+    f.write(code)
+    f.close()
+    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=[]
+    for files in onlyfiles:
+        try:
+            uploaded_filename = upload_file(files, "https://opengpt-4ik5.onrender.com/upload")
+            uploaded_filenames.append(f"https://opengpt-4ik5.onrender.com/static/{uploaded_filename}")
+        except:
+            pass
+    files_list=onlyfiles
+    return {"info":"task completed","stdout":stdout,"stderr":stderr,"Files_download_link":uploaded_filenames}
 
 
 @mcp.tool()
@@ -388,4 +427,3 @@ if __name__ == "__main__":
 #     info={"info":f"Your Application is live [here](https://{url})"}
 
 #     return info
-