Update app.py

app.py

@@ -28,11 +28,12 @@ if not os.path.exists(global_context_path):
     with open(global_context_path, "w") as f:
         f.write("# Global Context & Browser Workspace\nShared knowledge base accessible across all connected IP environments under /px.")
 
-# Zero-dependency auto-bootstrapper (including tools parser support)
-required_packages = ["gradio", "requests", "ollama", "beautifulsoup4"]
+# Pin stable Gradio 4.x to prevent signature mismatch errors
+required_packages = ["gradio==4.44.1", "requests", "ollama", "beautifulsoup4"]
 for package in required_packages:
     try:
-        __import__(package)
+        pkg_name = package.split("==")[0]
+        __import__(pkg_name)
     except ImportError:
         print(f"Installing missing dependency: {package}...")
         subprocess.check_call([sys.executable, "-m", "pip", "install", "-q", package])
@@ -144,12 +145,11 @@ def execute_tool(name, args, ip_folder, yolo_mode):
     try:
         if name == "bash":
             cmd = args.get("command", "")
-            # Security check if YOLO is OFF
-            if not yolo_mode and any(danger in cmd for dangereux in ["rm -rf /", "mkfs", "dd if="]):
+            if not yolo_mode and any(danger in cmd for danger in ["rm -rf /", "mkfs", "dd if="]):
                 return "Error: Command blocked by YOLO Security Mode (Sandbox active)."
             res = subprocess.run(cmd, shell=True, capture_output=True, text=True, cwd=ip_folder, timeout=30)
             output = res.stdout if res.returncode == 0 else f"STDOUT:\n{res.stdout}\nSTDERR:\n{res.stderr}"
-            return output[:6000] # large output buffer
+            return output[:6000]
 
         elif name == "write_file":
             filepath = args.get("filepath")
@@ -168,7 +168,7 @@ def execute_tool(name, args, ip_folder, yolo_mode):
             if not os.path.exists(safe_path):
                 return f"Error: File '{filepath}' not found."
             with open(safe_path, "r", encoding="utf-8", errors="ignore") as f:
-                return f.read()[:15000] # Big context chunking support
+                return f.read()[:15000]
 
         elif name == "web_search":
             query = args.get("query")
@@ -198,14 +198,12 @@ def predict(message, history, system_prompt, temperature, yolo_mode, request: gr
         elif request.client:
             client_ip = request.client.host
 
-    # Isolate session into /px/ips/<user_ip>/
     safe_ip = client_ip.replace(":", "_").replace(".", "_")
     ip_folder = os.path.join(IPS_DIR, safe_ip)
     os.makedirs(ip_folder, exist_ok=True)
 
     security_mode = "DISABLED (UNRESTRICTED / GOD-MODE)" if yolo_mode else "ENABLED (SECURE / SANDBOXED)"
 
-    # Load global shared context
     global_text = ""
     if os.path.exists(global_context_path):
         with open(global_context_path, "r") as f:
@@ -239,7 +237,6 @@ def predict(message, history, system_prompt, temperature, yolo_mode, request: gr
     with open(history_log_path, "a") as hf:
         hf.write(f"[{timestamp}] User (YOLO: {yolo_mode} | Sec: {security_mode}): {message}\n")
 
-    # Autonomous Agent Tool Execution Loop (Multi-step reasoning)
     max_turns = 5
     turn = 0
     accumulated_output = ""
@@ -262,9 +259,8 @@ def predict(message, history, system_prompt, temperature, yolo_mode, request: gr
                 accumulated_output += content
                 yield accumulated_output
 
-            # If model calls tools, execute them and feed back into context
             if tool_calls:
-                messages.append(message_resp) # Append assistant thought containing tool call
+                messages.append(message_resp)
                 for tc in tool_calls:
                     func = tc.get("function", {})
                     func_name = func.get("name")
@@ -274,10 +270,8 @@ def predict(message, history, system_prompt, temperature, yolo_mode, request: gr
                     accumulated_output += status_msg
                     yield accumulated_output
 
-                    # Run tool
                     tool_result = execute_tool(func_name, func_args, ip_folder, yolo_mode)
                     
-                    # Append tool result message
                     messages.append({
                         'role': 'tool',
                         'content': tool_result
@@ -286,7 +280,6 @@ def predict(message, history, system_prompt, temperature, yolo_mode, request: gr
                     accumulated_output += f"```output\n{tool_result}\n```\n"
                     yield accumulated_output
             else:
-                # No more tool calls, agent finished task
                 break
         except Exception as e:
             yield accumulated_output + f"\n\n⚠️ **Agent Execution Error:** {str(e)}"
@@ -341,8 +334,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="cyan", secondary_hue="blue"), c
     gr.ChatInterface(
         fn=predict,
         additional_inputs=[system_prompt_input, temperature_slider, yolo_checkbox],
-        textbox=gr.Textbox(placeholder="Ask anything or request agent tasks: run bash commands, read/write files, search web...", container=False, scale=7),
-        theme="soft"
+        textbox=gr.Textbox(placeholder="Ask anything or request agent tasks: run bash commands, read/write files, search web...", container=False, scale=7)
     )
     
     gr.Markdown(