Upload app.py with huggingface_hub

app.py

@@ -1,5 +1,9 @@
 import os
+import pty
 import asyncio
+import struct
+import fcntl
+import termios
 import json
 import subprocess
 
@@ -10,8 +14,6 @@ from fastapi.staticfiles import StaticFiles
 app = FastAPI()
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
-SESSION_NAME = "main"
-
 @app.get("/")
 async def index():
     return FileResponse("static/index.html")
@@ -20,55 +22,64 @@ async def index():
 async def get_pass():
     return {"pass": os.environ.get("TERMINAL_PASS", "")}
 
-def ensure_session():
-    result = subprocess.run(
-        ["tmux", "has-session", "-t", SESSION_NAME],
-        capture_output=True
-    )
-    if result.returncode != 0:
-        subprocess.run([
-            "tmux", "new-session", "-d", "-s", SESSION_NAME,
-            "-x", "200", "-y", "50"
-        ])
-
 @app.websocket("/ws")
 async def terminal(websocket: WebSocket):
     await websocket.accept()
-    ensure_session()
+
+    master_fd, slave_fd = pty.openpty()
 
     proc = await asyncio.create_subprocess_exec(
-        "tmux", "attach", "-t", SESSION_NAME,
-        stdin=asyncio.subprocess.PIPE,
-        stdout=asyncio.subprocess.PIPE,
-        stderr=asyncio.subprocess.PIPE
+        "/bin/bash",
+        stdin=slave_fd,
+        stdout=slave_fd,
+        stderr=slave_fd,
+        env={
+            **os.environ,
+            "TERM": "xterm-256color",
+            "HOME": "/persistent",
+            "USER": "ubuntu"
+        }
     )
 
-    async def read_output():
+    os.close(slave_fd)
+
+    loop = asyncio.get_event_loop()
+
+    async def read_pty():
         while True:
             try:
-                data = await proc.stdout.read(4096)
-                if not data:
-                    break
-                await websocket.send_text(data.decode("utf-8", errors="ignore"))
-            except:
+                data = await loop.run_in_executor(
+                    None,
+                    lambda: os.read(master_fd, 4096)
+                )
+                if data:
+                    try:
+                        text = data.decode("utf-8")
+                    except UnicodeDecodeError:
+                        text = data.decode("utf-8", errors="ignore")
+                    await websocket.send_text(text)
+            except OSError:
                 break
 
-    read_task = asyncio.create_task(read_output())
+    read_task = asyncio.create_task(read_pty())
 
     try:
         while True:
             message = await websocket.receive_text()
-            msg = json.loads(message)
-            if msg.get("type") == "input":
-                proc.stdin.write(msg["data"].encode("utf-8"))
-                await proc.stdin.drain()
-            elif msg.get("type") == "resize":
-                cols = msg.get("cols", 80)
-                rows = msg.get("rows", 24)
-                subprocess.run([
-                    "tmux", "resize-window", "-t", SESSION_NAME,
-                    "-x", str(cols), "-y", str(rows)
-                ])
+            try:
+                msg = json.loads(message)
+                if msg.get("type") == "input":
+                    os.write(master_fd, msg["data"].encode("utf-8"))
+                elif msg.get("type") == "resize":
+                    cols = msg.get("cols", 80)
+                    rows = msg.get("rows", 24)
+                    fcntl.ioctl(
+                        master_fd,
+                        termios.TIOCSWINSZ,
+                        struct.pack("HHHH", rows, cols, 0, 0)
+                    )
+            except (json.JSONDecodeError, OSError):
+                break
     except (WebSocketDisconnect, Exception):
         pass
     finally:
@@ -76,5 +87,9 @@ async def terminal(websocket: WebSocket):
         try:
             proc.terminate()
             await asyncio.wait_for(proc.wait(), timeout=2)
-        except:
+        except Exception:
             proc.kill()
+        try:
+            os.close(master_fd)
+        except:
+            pass