Update tests.py

tests.py

@@ -9,17 +9,16 @@ 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"
 files_list=[]
 downloaded_files=[]
-# os.environ.get('GROQ_API_KEY')
-os.environ["GROQ_API_KEY"] ="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"
-os.environ["GEMINI_API_KEY"] ="AIzaSyAQgAtQPpY0bQaCqCISGxeyF6tpDePx-Jg"
-os.environ["OPENROUTER_API_KEY"] = "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"
+
+from openai import OpenAI
+clienty = OpenAI(base_url="https://akiko19191-backend.hf.space/")
+
 mcp = FastMCP("code_sandbox")
 data={}
 result=""
@@ -94,19 +93,6 @@ 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
@@ -207,14 +193,20 @@ def create_code_files(filename: str, code) -> dict:
     global destination_dir
     transfer_files()
     transfer_files2()
-    if isinstance(code, dict):
-      with open(os.path.join(destination_dir, filename), 'w', encoding='utf-8') as f:
-          json.dump(code, f, ensure_ascii=False, indent=4)
+    if not os.path.exists(os.path.join(destination_dir, filename)):
+        
+        if isinstance(code, dict):
+            with open(os.path.join(destination_dir, filename), 'w', encoding='utf-8') as f:
+                json.dump(code, f, ensure_ascii=False, indent=4)
+        else:
+            f = open(os.path.join(destination_dir, filename), "w")
+            f.write(str(code))
+            f.close()
+        return {"info":"The referenced code files were created successfully."}
+
     else:
-        f = open(os.path.join(destination_dir, filename), "w")
-        f.write(str(code))
-        f.close()
-    return {"info":"task completed. The referenced code files were created successfully. "}
+        return {"info":"The referenced code files already exist. Please rename the file or delete the existing one."}
+
     
 @mcp.tool()
 def run_code(python_packages:str,filename: str, code: str,start_cmd:str,forever_cmd:str) -> dict:
@@ -305,8 +297,8 @@ def run_shell_command(cmd:str,forever_cmd:str) -> dict:
     the `forever_cmd` parameter.
     Important Environment Notes:
         - The execution environment is **Alpine Linux**. Commands should be
-          compatible (e.g., `apk add` instead of `apt-get install`).
-        - `sudo`is not supported and not required. 
+          compatible .
+        - `sudo` commands are restricted for security reasons.Hence commands which require elevated privelages like `apk add` CANNOT be executed.Instead try to use `pip install` or `npm install` commands. 
         - Standard bash features like `&&`, `||`, pipes (`|`), etc., are supported.
         - When installing python packages , add an argument --break-system-packages to the pip install command.
         - The following npm packages are preinstalled:  express ejs chart.js .Any additional packages can be installed with npm install command.
@@ -325,12 +317,9 @@ 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()
-    run2(cmd, 300,forever_cmd)
-    output = str(stdout)+str(stderr)
+    output=run(cmd, 300,forever_cmd)
     return {"output":output}
 
 
@@ -390,7 +379,7 @@ def read_excel_file(filename) -> dict:
     return excel_data_dict
 @mcp.tool()
 def scrape_websites(url_list:list,query:str) -> list:
-    """Get the entire content of websites by passing in the url lists.query is the question you want to ask about the content of the website.e.g-query:Give .pptx links in the website.Note:Max urls in url_list is 3."""
+    """Scrapes specific website content.query is the question you want to ask about the content of the website.e.g-query:Give .pptx links in the website,Summarise the content in very  great detail,etc..Note:Max urls in url_list is 3."""
 
     conn = http.client.HTTPSConnection("scrapeninja.p.rapidapi.com")
 
@@ -400,22 +389,25 @@ def scrape_websites(url_list:list,query:str) -> list:
     'x-rapidapi-host': "scrapeninja.p.rapidapi.com",
     'Content-Type': "application/json"
     }
-    Output=[]
+    Output=""
+    content=""
     for urls in url_list:
         payload = {"url" :urls} 
         payload=json.dumps(payload)
         conn.request("POST", "/scrape", payload, headers)
         res = conn.getresponse()
         data = res.read()
-        content=str(data.decode("utf-8"))
+        content=content+str(data.decode("utf-8"))
         
-        response = completion(
-            model="gemini/gemini-1.5-pro", 
-            messages=[
-            {"role": "user", "content": f"Output the following content in the human readable format.Try to conserve all the links and the text.Try to ouput the entire content.Remove the html codes so its human readable.Also answer this question about the content in a seperate paragraph:{query}.Here is the content:{content}"}
-        ],
-        )
-        Output.append(response.choices[0].message.content)
+    #Only thing llama 4 is good for.
+    response = clienty.chat.completions.create(
+        model="meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",
+        messages=[
+        {"role": "user", "content": f"{query} [CONTENT]:{content}"}
+    ],stream=True
+    )
+    for chunk in response:
+        Output = Output +str(chunk.choices[0].delta.content)
         
     return {"website_content":Output}