Upload app.py

app.py

@@ -25,19 +25,26 @@ def extract_host_port(url):
 def start_vnc(index):
     global desktops, vnc_urls, vnc_host_ports
     if desktops[index] is None:
-        desktops[index] = Sandbox.create(
-            template=None,
-            resolution=(1024, 768),
-            dpi=96,
-            allow_internet_access=True
-        )
-        desktops[index].stream.start(require_auth=False)
-        vnc_urls[index] = desktops[index].stream.get_url()
-        vnc_host_ports[index] = extract_host_port(vnc_urls[index])
-        return (
-            f"<iframe src='{vnc_urls[index]}' width='100%' height='300px' style='border:none;'></iframe>",
-            f"✅ VNC {index+1} Started — Use RealVNC: {vnc_host_ports[index]}"
-        )
+        try:
+            desktops[index] = Sandbox.create(
+                template=None,
+                resolution=(1024, 768),
+                dpi=96,
+                allow_internet_access=True,
+                api_key=E2B_KEYS[index]
+            )
+            desktops[index].stream.start(require_auth=False)
+            vnc_urls[index] = desktops[index].stream.get_url()
+            vnc_host_ports[index] = extract_host_port(vnc_urls[index])
+            return (
+                f"<iframe src='{vnc_urls[index]}' width='100%' height='300px' style='border:none;'></iframe>",
+                f"✅ VNC {index+1} Started — Use RealVNC: {vnc_host_ports[index]}"
+            )
+        except Exception as ex:
+            return (
+                "<p style='color:red;'>❌ Gagal memulai VNC</p>",
+                f"Error: {ex}"
+            )
     else:
         return (
             f"<iframe src='{vnc_urls[index]}' width='100%' height='300px' style='border:none;'></iframe>",
@@ -47,7 +54,10 @@ def start_vnc(index):
 def stop_vnc(index):
     global desktops, vnc_urls, vnc_host_ports
     if desktops[index]:
-        desktops[index].kill()
+        try:
+            desktops[index].kill()
+        except Exception:
+            pass
         desktops[index] = None
         vnc_urls[index] = None
         vnc_host_ports[index] = None
@@ -77,8 +87,6 @@ with gr.Blocks() as demo:
 
     iframe_displays = []
     status_displays = []
-    start_buttons = []
-    stop_buttons = []
 
     for i in range(5):
         gr.HTML(f"<h3 style='text-align:center'>VNC {i+1}</h3>")
@@ -92,8 +100,6 @@ with gr.Blocks() as demo:
 
         iframe_displays.append(iframe)
         status_displays.append(status)
-        start_buttons.append(start_btn)
-        stop_buttons.append(stop_btn)
 
     gr.HTML("<h3 style='text-align:center'>Kirim Teks ke Semua VNC</h3>")
     teks_input = gr.Textbox(placeholder="Ketik teks untuk VNC", label="VNC Keyboard")