Create app.py

app.py

@@ -0,0 +1,99 @@
+import os
+import subprocess
+import threading
+import gradio as gr
+import time
+import urllib.request
+import tarfile
+import zipfile
+
+# 下载并解压工具函数
+def download_and_extract(url, extract_to, is_zip=False):
+    filename = url.split("/")[-1]
+    if not os.path.exists(extract_to):
+        os.makedirs(extract_to)
+    filepath = os.path.join(extract_to, filename)
+    if not os.path.exists(filepath):
+        print(f"Downloading {filename} ...")
+        urllib.request.urlretrieve(url, filepath)
+    if is_zip:
+        with zipfile.ZipFile(filepath, "r") as zip_ref:
+            zip_ref.extractall(extract_to)
+    else:
+        with tarfile.open(filepath, "r:*") as tar:
+            tar.extractall(path=extract_to)
+
+# 下载并解压 fluxbox
+def setup_fluxbox():
+    url = "https://github.com/void-linux/void-packages/files/12496441/fluxbox-1.3.7-x86_64.tar.gz"
+    extract_to = "./fluxbox"
+    download_and_extract(url, extract_to)
+    return os.path.join(extract_to, "fluxbox-1.3.7-x86_64", "bin", "fluxbox")
+
+# 下载并解压 TigerVNC
+def setup_tigervnc():
+    url = "https://github.com/TigerVNC/tigervnc/releases/download/v1.13.1/tigervnc-1.13.1.x86_64.tar.gz"
+    extract_to = "./tigervnc"
+    download_and_extract(url, extract_to)
+    return os.path.join(extract_to, "tigervnc-1.13.1.x86_64", "vncserver")
+
+# 下载并解压 noVNC
+def setup_novnc():
+    url = "https://github.com/novnc/noVNC/archive/refs/heads/master.zip"
+    extract_to = "./novnc"
+    download_and_extract(url, extract_to, is_zip=True)
+    return os.path.join(extract_to, "noVNC-master")
+
+# 启动 Xvfb
+def start_xvfb():
+    return subprocess.Popen(["Xvfb", ":1", "-screen", "0", "1024x768x16"], env={**os.environ, "DISPLAY": ":1"})
+
+# 启动 fluxbox
+def start_fluxbox(fluxbox_path):
+    return subprocess.Popen([fluxbox_path], env={**os.environ, "DISPLAY": ":1"})
+
+# 启动 VNC server
+def start_vncserver(vncserver_path):
+    return subprocess.Popen([vncserver_path, ":1", "-geometry", "1024x768", "-SecurityTypes", "None"], env={**os.environ, "DISPLAY": ":1"})
+
+# 启动 noVNC
+def start_novnc(novnc_path):
+    websockify = os.path.join(novnc_path, "utils", "websockify", "run")
+    return subprocess.Popen(["python", websockify, "7860", "localhost:5901", "--web", novnc_path], env=os.environ)
+
+# 启动 ssh_client.py
+def start_ssh_client():
+    if os.path.exists("ssh_client.py"):
+        return subprocess.Popen(["python", "ssh_client.py"])
+    return None
+
+def setup_and_run():
+    fluxbox_path = setup_fluxbox()
+    vncserver_path = setup_tigervnc()
+    novnc_path = setup_novnc()
+    # 启动 Xvfb
+    start_xvfb()
+    time.sleep(2)
+    # 启动 fluxbox
+    start_fluxbox(fluxbox_path)
+    time.sleep(2)
+    # 启动 VNC server
+    start_vncserver(vncserver_path)
+    time.sleep(2)
+    # 启动 noVNC
+    start_novnc(novnc_path)
+    # 启动 ssh_client.py
+    # start_ssh_client()
+
+# 用线程后台启动桌面环境
+threading.Thread(target=setup_and_run, daemon=True).start()
+
+def hello():
+    return "欢迎来到 Huggingface Space！已启动 Fluxbox+TigerVNC+noVNC+SSH 客户端。\n" \
+           "请通过 Web VNC 访问桌面环境：\n" \
+           "http://<your-space-url>:7860/vnc.html"
+
+demo = gr.Interface(fn=hello, inputs=None, outputs="text", title="Fluxbox TigerVNC Space")
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)