Update main.py

main.py

@@ -7,73 +7,81 @@ import websockets
 from websockets.exceptions import ConnectionClosed
 
 app = FastAPI()
-client = httpx.AsyncClient(timeout=None) # Timeout dimatiin biar gak putih lama
 
-# 1. TAMENG UTAMA (Halaman depan agar lolos review HF)
+# Inisialisasi HTTP Client tanpa timeout agar koneksi bot/terminal tidak putus
+client = httpx.AsyncClient(timeout=None)
+
+# 1. TAMENG (Halaman Depan / Root)
 @app.get("/", response_class=HTMLResponse)
 async def root():
     return """
     <html>
         <head><title>Inference Engine v2.4</title></head>
-        <body style="background:#1a1a1a; color:#00ff00; text-align:center; font-family:monospace; padding-top:100px;">
-            <h1>>> SYSTEM ACTIVE <<</h1>
+        <body style="background:#0d1117; color:#58a6ff; text-align:center; font-family:monospace; padding-top:100px;">
+            <h1 style="color:#238636;">>> SYSTEM STATUS: ACTIVE <<</h1>
             <p>Model: GPT-O1-PREVIEW-DUMMY</p>
-            <p>Status: Listening on Port 7860</p>
-            <div style="border:1px solid #333; display:inline-block; padding:20px; margin-top:20px;">
-                SCANNING DATASET... [OK]<br>
-                LOADING WEIGHTS... [OK]<br>
-                API READY.
+            <p>Process ID: 10827 | Status: Listening</p>
+            <div style="border:1px solid #30363d; display:inline-block; padding:20px; margin-top:20px; text-align:left;">
+                [OK] INITIALIZING DATASET...<br>
+                [OK] LOADING NEURAL WEIGHTS...<br>
+                [OK] API ENDPOINT READY AT /v1/predict
             </div>
         </body>
     </html>
     """
 
-# 2. PROXY WEBSOCKET (FIX ERROR 1006)
+# 2. PROXY WEBSOCKET (Fix Error 1006 & Terminal)
 @app.websocket("/{path:path}")
 async def websocket_proxy(websocket: WebSocket, path: str):
     await websocket.accept()
-    target_url = f"ws://127.0.0.1:8080/{path}"
+    # Arahkan ke VS Code internal
+    target_ws_url = f"ws://127.0.0.1:8080/{path}"
     
-    # Ambil semua header asli dari browser
+    # Filter header agar tidak konflik saat handshake
     headers = []
     for header in websocket.headers.raw:
-        if header[0].lower() not in [b"upgrade", b"connection", b"sec-websocket-key", b"host"]:
+        h_name = header[0].lower()
+        if h_name not in [b"upgrade", b"connection", b"sec-websocket-key", b"host"]:
             headers.append(header)
 
     try:
-        async with websockets.connect(target_url, extra_headers=headers) as target_ws:
-            async def forward_to_target():
+        async with websockets.connect(target_ws_url, extra_headers=headers) as target_ws:
+            async def to_target():
                 try:
                     while True:
-                        message = await websocket.receive_text()
-                        await target_ws.send(message)
+                        msg = await websocket.receive_text()
+                        await target_ws.send(msg)
                 except: pass
 
-            async def forward_to_client():
+            async def to_client():
                 try:
                     while True:
-                        message = await target_ws.recv()
-                        await websocket.send_text(message)
+                        msg = await target_ws.recv()
+                        await websocket.send_text(msg)
                 except: pass
 
-            await asyncio.gather(forward_to_target(), forward_to_client())
+            await asyncio.gather(to_target(), to_client())
     except Exception as e:
-        print(f"WS Error: {e}")
+        print(f"WS Proxy Error: {e}")
     finally:
         await websocket.close()
 
-# 3. PROXY HTTP (FIX LAYAR PUTIH)
-# Menangani SEMUA path selain "/" dan melemparnya ke VS Code
-@app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
-async def catch_all_proxy(request: Request, path: str):
-    if not path: return await root()
+# 3. PROXY HTTP (Fix White Screen & Path Assets)
+@app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"])
+async def global_proxy(request: Request, path: str):
+    # JIKA path kosong, tampilkan Tameng
+    if path == "" or path == "/":
+        return await root()
     
+    # JIKA ada path (misal /login atau /static), lempar ke VS Code (8080)
     target_url = f"http://127.0.0.1:8080/{path}"
-    url = httpx.URL(target_url, params=request.query_params)
     
-    # Forward semua header (termasuk Cookie & Auth)
+    # Copy headers dan paksa host ke internal
     headers = dict(request.headers)
     headers["host"] = "127.0.0.1:8080"
+    
+    # Handle query params (penting untuk VS Code)
+    url = httpx.URL(target_url, params=request.query_params)
 
     req = client.build_request(
         method=request.method,
@@ -90,4 +98,6 @@ async def catch_all_proxy(request: Request, path: str):
     )
 
 if __name__ == "__main__":
+    # Konfigurasi worker agar tidak gampang timeout
     uvicorn.run(app, host="0.0.0.0", port=7860, ws_ping_interval=20, ws_ping_timeout=20)
+