Update main.py

main.py

@@ -110,7 +110,27 @@ remote_port = 8188
     except Exception as e:
         print(f"创建 frpc.toml 配置文件时出错: {e}")
 
+def install_frp():
+    frp_tar_path = os.path.join(REPO_PATH, "frp_0.61.0_linux_amd64.tar.gz")
+    frp_install_path = os.path.join(REPO_PATH, "frp_0.61.0_linux_amd64")
 
+    if os.path.exists(frp_install_path):
+        print(f"frp 已安装: {frp_install_path}")
+        return True
+
+    if not os.path.exists(frp_tar_path):
+        print(f"未找到 frp 安装包: {frp_tar_path}")
+        return False
+
+    try:
+        print(f"正在解压 frp 安装包到 {REPO_PATH}...")
+        subprocess.run(["tar", "-xvzf", frp_tar_path, "-C", REPO_PATH], check=True)
+        print("frp 安装成功！")
+        return True
+    except subprocess.CalledProcessError as e:
+        print(f"解压 frp 时出错: {e}")
+        return False
+    
 # 启动 start_jupyterlab.py 的线程函数
 def start_jupyterlab_process():
     jupyter_script = os.path.join(REPO_PATH, "start_jupyterlab.py")
@@ -179,6 +199,7 @@ def start_gradio():
 # 主函数
 def main():
     clone_repository()
+    install_frp()
 
     # 创建 frp 配置文件（如果不存在）
     create_frpc_toml()