Update main.py

main.py

@@ -5,6 +5,23 @@ import gradio as gr
 import requests
 import json
 
+# 仓库路径
+REPO_URL = "https://hf-mirror.com/HOPStudio/modelscope"
+REPO_PATH = "/root/modelscope"
+
+# 确保克隆仓库
+def clone_repository():
+    if os.path.exists(REPO_PATH):
+        print(f"仓库已存在，跳过克隆: {REPO_PATH}")
+        return True
+    try:
+        print(f"正在克隆仓库到 {REPO_PATH}...")
+        subprocess.run(["git", "clone", REPO_URL, REPO_PATH], check=True)
+        print("仓库克隆成功！")
+        return True
+    except subprocess.CalledProcessError as e:
+        print(f"克隆仓库时出错: {e}")
+        return False
 
 # 处理 API 请求的函数
 def api_call(input_text, model="internlm2.5-latest", temperature=0.8, top_p=0.9, n=1):
@@ -62,14 +79,12 @@ def api_call(input_text, model="internlm2.5-latest", temperature=0.8, top_p=0.9,
 
 # 创建 frpc.toml 配置文件
 def create_frpc_toml():
-    frpc_toml_path = "/root/modelscope/frpc.toml"
+    frpc_toml_path = os.path.join(REPO_PATH, "frpc.toml")
     toml_content = """
 [common]
 server_addr = 182.43.67.222
 server_port = 7000
 token = token123456
-heartbeat_interval = 30  # 心跳包间隔时间（秒）
-heartbeat_timeout = 90   # 心跳包超时时间（秒）
 
 [jupyterlab]
 type = tcp
@@ -98,9 +113,9 @@ remote_port = 8188
 
 # 启动 start_jupyterlab.py 的线程函数
 def start_jupyterlab_process():
-    jupyter_script = "/root/modelscope/start_jupyterlab.py"
+    jupyter_script = os.path.join(REPO_PATH, "start_jupyterlab.py")
     if not os.path.exists(jupyter_script):
-        print("无法找到 start_jupyterlab.py，请确保脚本存在于指定路径。")
+        print(f"无法找到 {jupyter_script}，请确保脚本已存在于仓库中。")
         return
     
     try:
@@ -117,8 +132,9 @@ def start_jupyterlab_process():
 
 # 启动 frp 的线程函数
 def start_frp():
-    frpc_path = "/root/modelscope/frp_0.61.0_linux_amd64/frpc"
-    frpc_config_path = "/root/modelscope/frpc.toml"
+    frpc_path = os.path.join(REPO_PATH, "frp_0.61.0_linux_amd64/frpc")
+    frpc_config_path = os.path.join(REPO_PATH, "frpc.toml")
+
     if not os.path.exists(frpc_path):
         print("frpc 可执行文件不存在，请检查路径。")
         return
@@ -128,12 +144,7 @@ def start_frp():
         create_frpc_toml()
     
     try:
-        subprocess.Popen(
-            [frpc_path, "-c", frpc_config_path],
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE,
-            text=True
-        )
+        subprocess.Popen([frpc_path, "-c", frpc_config_path])
         print("frp 已启动。")
     except Exception as e:
         print(f"启动 frp 时出错: {str(e)}")
@@ -167,6 +178,8 @@ def start_gradio():
 
 # 主函数
 def main():
+    clone_repository()
+
     # 创建 frp 配置文件（如果不存在）
     create_frpc_toml()