Update app.py

app.py

@@ -2,15 +2,12 @@ import subprocess
 import asyncio
 import os
 import psutil
-import json
-from fastapi import FastAPI, WebSocket, WebSocketDisconnect, HTTPException, Depends
+import shutil
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 from fastapi.responses import HTMLResponse, JSONResponse
-from fastapi.staticfiles import StaticFiles
 
 app = FastAPI()
-
-# সিকিউরিটি: এখানে আপনার নিজের পাসওয়ার্ড সেট করুন
-ACCESS_PASSWORD = "admin" 
+ACCESS_PASSWORD = "admin" # আপনার পাসওয়ার্ড
 
 @app.get("/")
 async def get():
@@ -19,25 +16,26 @@ async def get():
 
 @app.get("/api/metrics")
 async def get_metrics():
-    # সিস্টেম রিসোর্স ডাটা
     return {
         "cpu": psutil.cpu_percent(),
         "ram": psutil.virtual_memory().percent,
-        "disk": psutil.disk_usage('/').percent
+        "disk": psutil.disk_usage('/').percent,
+        "uptime": shutil.disk_usage("/").total // (1024**3) # Storage limit approx
     }
 
 @app.get("/api/files")
 async def list_files(path: str = "."):
-    # ফাইল এক্সপ্লোরার ডাটা
     try:
+        abs_path = os.path.abspath(path)
         files = []
-        for entry in os.scandir(path):
+        for entry in os.scandir(abs_path):
             files.append({
                 "name": entry.name,
                 "is_dir": entry.is_dir(),
-                "size": entry.stat().st_size if entry.is_file() else "-"
+                "path": os.path.join(path, entry.name),
+                "size": f"{entry.stat().st_size / 1024:.1f} KB" if entry.is_file() else "-"
             })
-        return files
+        return sorted(files, key=lambda x: not x['is_dir'])
     except Exception as e:
         return JSONResponse(status_code=500, content={"error": str(e)})
 
@@ -45,24 +43,28 @@ async def list_files(path: str = "."):
 async def websocket_endpoint(websocket: WebSocket):
     await websocket.accept()
     try:
-        # প্রথম মেসেজটি পাসওয়ার্ড হতে হবে
-        auth_data = await websocket.receive_text()
-        if auth_data != ACCESS_PASSWORD:
-            await websocket.send_text("AUTH_FAILED")
+        auth = await websocket.receive_text()
+        if auth != ACCESS_PASSWORD:
+            await websocket.send_text("[-] AUTH_FAILED")
             await websocket.close()
             return
         
-        await websocket.send_text("AUTH_SUCCESS")
+        await websocket.send_text("[+] AUTH_SUCCESS")
         
         while True:
             command = await websocket.receive_text()
+            # ক্লিয়ার কমান্ড হ্যান্ডলিং
+            if command.strip() == "clear":
+                await websocket.send_text("\033[H\033[2J")
+                continue
+
             process = await asyncio.create_subprocess_shell(
                 command,
                 stdout=asyncio.subprocess.PIPE,
                 stderr=asyncio.subprocess.PIPE
             )
             stdout, stderr = await process.communicate()
-            output = stdout.decode().strip() or stderr.decode().strip() or "Success (No Output)"
+            output = stdout.decode().strip() or stderr.decode().strip() or "Done."
             await websocket.send_text(output)
     except WebSocketDisconnect:
         pass