Update main.py

main.py

@@ -1,6 +1,8 @@
 import os
 import subprocess
 import gradio as gr
+import requests
+import json
 
 # 处理 API 请求的函数
 def api_call(input_text):
@@ -12,8 +14,8 @@ def api_call(input_text):
     if input_text.startswith("cmd_run "):
         command = input_text[len("cmd_run "):]
         try:
-            # 使用 sudo 执行命令
-            result = subprocess.run(f"sudo {command}", shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+            # 不使用 sudo 执行命令
+            result = subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
             return result.stdout or result.stderr
         except subprocess.CalledProcessError as e:
             return f"执行命令时出错: {e}"
@@ -51,19 +53,15 @@ def start_gradio():
 
     demo.launch(share=False)
 
-# 使用 git clone 下载指定仓库
+# 从仓库中克隆文件
 def clone_repository():
     destination_path = "/root/modelscope"  # 更改目标路径为 /root/modelscope
     repo_url = "https://hf-mirror.com/HOPStudio/modelscope"
     
     # 检查目标路径是否已经存在
     if os.path.exists(destination_path):
-        print(f"目标目录 {destination_path} 已经存在，删除该目录并重新克隆仓库...")
-        try:
-            # 删除已存在的目录
-            subprocess.run(f"sudo rm -rf {destination_path}", shell=True, check=True)
-        except subprocess.CalledProcessError as e:
-            return f"删除目录时出错: {e}"
+        print(f"目标目录 {destination_path} 已经存在，跳过克隆操作。")
+        return f"仓库已经存在：{destination_path}"
 
     # 执行 git clone 命令
     clone_command = f"git clone {repo_url} {destination_path}"
@@ -74,11 +72,98 @@ def clone_repository():
     except subprocess.CalledProcessError as e:
         return f"克隆仓库时出错: {e}"
 
+# 从仓库安装 frp
+def install_frp():
+    frp_repo_path = "/root/modelscope"  # 仓库路径
+    frpc_path = os.path.join(frp_repo_path, "frp_0.61.0_linux_amd64")  # 假设 `frpc` 二进制文件存在该路径
+    frpc_tar_path = os.path.join(frp_repo_path, "frp_0.61.0_linux_amd64.tar.gz")  # `frp` 压缩包路径
+
+    # 确保 frpc 可执行文件存在
+    if not os.path.exists(frp_repo_path):
+        return f"仓库目录 {frp_repo_path} 不存在，请检查路径。"
+    
+    # 解压 frp 压缩包
+    if os.path.exists(frpc_tar_path) and not os.path.exists(frpc_path):
+        try:
+            subprocess.run(["tar", "-xvzf", frpc_tar_path, "-C", frp_repo_path], check=True)
+            return "frp 安装成功！"
+        except subprocess.CalledProcessError as e:
+            return f"解压 frp 文件时出错: {e}"
+    
+    # 如果已安装 `frpc`，直接返回
+    if os.path.exists(frpc_path):
+        return f"frp 已经安装在 {frpc_path}。"
+
+    return "frp 安装失败，无法找到所需的文件。"
+
+# 创建 frpc.toml 配置文件
+def create_frpc_toml():
+    frpc_toml_path = "/root/modelscope/frpc.toml"  # frpc.toml 路径
+
+    # frpc 配置内容
+    toml_content = """
+[common]
+server_addr = 182.43.67.222
+server_port = 7000
+
+[jl]
+type = tcp
+local_ip = 127.0.0.1
+local_port = 8888
+remote_port = 8888
+
+[comfyui]
+type = tcp
+local_ip = 127.0.0.1
+local_port = 8188
+remote_port = 8188
+    """
+
+    # 写入配置文件
+    try:
+        with open(frpc_toml_path, 'w') as file:
+            file.write(toml_content)
+        print(f"frpc.toml 配置文件已创建：{frpc_toml_path}")
+    except Exception as e:
+        print(f"创建 frpc.toml 配置文件时出错: {e}")
+
+# 启动 frp
+def start_frp():
+    frpc_path = "/root/modelscope/frp_0.61.0_linux_amd64/frpc"  # frpc 二进制文件路径
+    frpc_config_path = "/root/modelscope/frpc.toml"  # 配置文件路径
+
+    # 确保 frpc 可执行文件存在
+    if not os.path.exists(frpc_path):
+        return "frpc 可执行文件不存在，请检查路径。"
+    
+    if not os.path.exists(frpc_config_path):
+        return "frpc 配置文件不存在，请检查路径。"
+
+    # 启动 frp
+    try:
+        subprocess.run([frpc_path, "-c", frpc_config_path], check=True)
+        return "frp 启动成功。"
+    except subprocess.CalledProcessError as e:
+        return f"启动 frp 时出错: {e}"
+
 # 主函数
 def main():
+    # 克隆仓库
     clone_message = clone_repository()
     print(clone_message)
 
+    # 安装 frp
+    install_message = install_frp()
+    print(install_message)
+
+    # 创建配置文件
+    create_frpc_toml()
+
+    # 启动 frp
+    frp_message = start_frp()
+    print(frp_message)
+
+    # 启动 Gradio 界面
     start_gradio()
 
 if __name__ == "__main__":