Update app.py

app.py

@@ -1,47 +1,44 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
 import subprocess
-from playwright.sync_api import sync_playwright
-import base64
-import os
 
 app = FastAPI()
 
-# -------- TERMINAL TOOL --------
+
+@app.get("/")
+def root():
+    return {"status": "ok", "message": "Terminal API is running"}
+
+
 @app.post("/terminal")
-def terminal(cmd: str):
-    result = subprocess.run(
-        cmd,
-        shell=True,
-        capture_output=True,
-        text=True
-    )
-    return {
-        "stdout": result.stdout,
-        "stderr": result.stderr,
-        "code": result.returncode
-    }
-
-# -------- BROWSER TOOL --------
-@app.get("/browse")
-def browse(url: str):
-    with sync_playwright() as p:
-        browser = p.chromium.launch(headless=True)
-        page = browser.new_page()
-        page.goto(url, timeout=60000)
-        screenshot = page.screenshot()
-        browser.close()
-
-    return {
-        "screenshot": base64.b64encode(screenshot).decode()
-    }
-
-# -------- SCREENSHOT OF SYSTEM --------
-@app.get("/screen")
-def screen():
-    os.system("apt-get update && apt-get install -y scrot")
-    os.system("scrot screen.png")
-
-    with open("screen.png","rb") as f:
+async def run_terminal(req: Request):
+    data = await req.json()
+    cmd = data.get("cmd")
+
+    if not cmd:
+        return JSONResponse(
+            status_code=400,
+            content={"error": "cmd field is required"}
+        )
+
+    try:
+        result = subprocess.run(
+            cmd,
+            shell=True,
+            capture_output=True,
+            text=True,
+            timeout=30
+        )
+
         return {
-            "image": base64.b64encode(f.read()).decode()
-        }
+            "command": cmd,
+            "stdout": result.stdout,
+            "stderr": result.stderr,
+            "returncode": result.returncode
+        }
+
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={"error": str(e)}
+        )