Update app.py

app.py

@@ -8,23 +8,20 @@ from aiohttp import web
 from aiortc import RTCPeerConnection, RTCSessionDescription, RTCIceServer, RTCConfiguration
 
 logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger("BuiltVNC-Bridge")
+logger = logging.getLogger("HF-VNC-Bridge")
 
+# Configuration
 VNC_PORT = 5900
+# Cloudflare TURN helps bypass Hugging Face's strict network/firewall
+TURN_USER = "g08abe68c81a07f098bb5f0914549bb32440e5aad0b216c7fba2b61e76fd62c6"
+TURN_PASS = "aed1a10dd10eba9401ad9d99e5c66036d8a970eab5ba8e6dc9845ab57c771a7d"
 
 def start_system():
     os.environ["DISPLAY"] = ":99"
-    
-    # 1. Start Xvfb
     subprocess.Popen(["Xvfb", ":99", "-screen", "0", "1280x720x24", "-ac"])
     time.sleep(1)
-
-    # 2. Start YOUR BUILT VNC TOOL
-    logger.info("Starting custom built minivnc...")
     subprocess.Popen(["/usr/local/bin/minivnc"])
     time.sleep(1)
-
-    # 3. Start Window Manager & Opera
     subprocess.Popen("matchbox-window-manager -use_titlebar no", shell=True)
     subprocess.Popen("opera --no-sandbox --user-data-dir=/home/user/opera-data", shell=True)
 
@@ -36,29 +33,53 @@ async def bridge_vnc(channel):
                 data = await reader.read(65536)
                 if not data: break
                 channel.send(data)
-        
         @channel.on("message")
         def on_message(msg):
             if isinstance(msg, bytes): writer.write(msg)
-
         await vnc_to_rtc()
     except Exception as e: logger.error(f"Bridge fail: {e}")
 
 async def offer(request):
     params = await request.json()
-    pc = RTCPeerConnection() # Simplified for brevity, add TURN if needed
+    
+    # Crucial: Use ICE Servers for remote connection
+    config = RTCConfiguration(iceServers=[
+        RTCIceServer(urls=["stun:stun.l.google.com:19302"]),
+        RTCIceServer(urls=["turns:turn.cloudflare.com:443?transport=tcp"], username=TURN_USER, credential=TURN_PASS)
+    ])
+    
+    pc = RTCPeerConnection(config)
+    pcs.add(pc)
+
     @pc.on("datachannel")
     def on_dc(channel):
         asyncio.create_task(bridge_vnc(channel))
-    
+
     await pc.setRemoteDescription(RTCSessionDescription(sdp=params["sdp"], type=params["type"]))
-    await pc.setLocalDescription(await pc.createAnswer())
-    return web.Response(content_type="application/json", text=json.dumps({
-        "sdp": pc.localDescription.sdp, "type": pc.localDescription.type
-    }), headers={"Access-Control-Allow-Origin": "*"})
+    answer = await pc.createAnswer()
+    await pc.setLocalDescription(answer)
+
+    return web.Response(
+        content_type="application/json", 
+        text=json.dumps({"sdp": pc.localDescription.sdp, "type": pc.localDescription.type}),
+        headers={
+            "Access-Control-Allow-Origin": "*", # Allow connections from your other site
+            "Access-Control-Allow-Methods": "POST, OPTIONS",
+            "Access-Control-Allow-Headers": "Content-Type"
+        }
+    )
+
+async def options(request):
+    return web.Response(headers={
+        "Access-Control-Allow-Origin": "*",
+        "Access-Control-Allow-Methods": "POST, OPTIONS",
+        "Access-Control-Allow-Headers": "Content-Type"
+    })
 
+pcs = set()
 if __name__ == "__main__":
     start_system()
     app = web.Application()
     app.router.add_post("/offer", offer)
-    web.run_app(app, port=7860)
+    app.router.add_options("/offer", options)
+    web.run_app(app, host="0.0.0.0", port=7860)