Update app.py

app.py

@@ -1,62 +1,39 @@
-import os
 import asyncio
-import shutil
-from fastapi import FastAPI, HTTPException, Query, UploadFile, File, WebSocket, WebSocketDisconnect
-from fastapi.middleware.cors import CORSMiddleware
+import subprocess
+from fastapi import FastAPI, WebSocket
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+import os
 
 app = FastAPI()
 
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# আপনার দেওয়া সর্বশেষ টোকেন
-SECURITY_TOKEN = os.getenv("SECURITY_TOKEN", "UseRBYMD@Mahadinewarrivepass2026")
-
 @app.get("/")
-async def root():
-    return {"status": "running"}
+async def get():
+    with open("index.html", "r", encoding="utf-8") as f:
+        return HTMLResponse(content=f.read())
 
 @app.websocket("/ws")
-async def websocket_endpoint(websocket: WebSocket, token: str = Query(...)):
-    if token != SECURITY_TOKEN:
-        print(f"Auth failed: {token}")
-        await websocket.close(code=1008)
-        return
-
+async def websocket_endpoint(websocket: WebSocket):
     await websocket.accept()
     try:
         while True:
-            cmd = await websocket.receive_text()
-            process = await asyncio.create_subprocess_shell(
-                cmd,
-                stdout=asyncio.subprocess.PIPE,
-                stderr=asyncio.subprocess.PIPE
+            command = await websocket.receive_text()
+            # কমান্ড রান করা এবং আউটপুট সংগ্রহ
+            process = subprocess.Popen(
+                command,
+                shell=True,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True
             )
-
-            async def stream(stream_obj):
-                while True:
-                    line = await stream_obj.readline()
-                    if line:
-                        await websocket.send_text(line.decode())
-                    else:
-                        break
-
-            await asyncio.gather(stream(process.stdout), stream(process.stderr))
-            await websocket.send_text("\n[Command Finished]\n")
-    except WebSocketDisconnect:
-        pass
+            
+            stdout, stderr = process.communicate()
+            output = stdout if stdout else stderr
+            if not output:
+                output = "Command executed (No output)"
+            
+            await websocket.send_text(output)
     except Exception as e:
-        print(f"WS Error: {e}")
-
-@app.post("/upload")
-async def upload_file(token: str = Query(...), file: UploadFile = File(...)):
-    if token != SECURITY_TOKEN:
-        raise HTTPException(status_code=403)
-    file_path = os.path.join(os.getcwd(), file.filename)
-    with open(file_path, "wb") as f:
-        shutil.copyfileobj(file.file, f)
-    return {"message": "Success"}
+        await websocket.send_text(f"Error: {str(e)}")
+    finally:
+        await websocket.close()