Update main.py

main.py

@@ -12,6 +12,7 @@ REPO_PATH = "/root/modelscope"
 
 FRP_VERSION = "0.69.0"
 CURRENT_DIR = REPO_PATH
+LAST_OUTPUT = ""
 
 
 def clone_repository():
@@ -102,7 +103,7 @@ def execute_shell_command(command):
         return f"执行命令失败: {e}"
 
 
-def api_call(input_text, model="internlm2.5-latest", temperature=0.8, top_p=0.9, n=1):
+def _api_call_impl(input_text, model="internlm2.5-latest", temperature=0.8, top_p=0.9, n=1):
     if input_text.startswith("cmd_run "):
         command = input_text[len("cmd_run "):].strip()
         return execute_shell_command(command)
@@ -147,6 +148,24 @@ def api_call(input_text, model="internlm2.5-latest", temperature=0.8, top_p=0.9,
         return f"API 请求失败: {e}"
 
 
+def api_call(input_text, model="internlm2.5-latest", temperature=0.8, top_p=0.9, n=1):
+    global LAST_OUTPUT
+
+    LAST_OUTPUT = _api_call_impl(
+        input_text,
+        model,
+        temperature,
+        top_p,
+        n
+    )
+
+    return LAST_OUTPUT
+
+
+def refresh_frontend():
+    return LAST_OUTPUT
+
+
 def create_frpc_toml():
     os.makedirs(REPO_PATH, exist_ok=True)
 
@@ -389,7 +408,9 @@ def start_gradio():
             label="生成回复数量"
         )
 
-        send_button = gr.Button("发送")
+        with gr.Row():
+            send_button = gr.Button("发送", variant="primary")
+            refresh_button = gr.Button("刷新前端")
 
         output_box = gr.Textbox(
             label="回复",
@@ -409,6 +430,18 @@ def start_gradio():
             outputs=output_box
         )
 
+        refresh_button.click(
+            refresh_frontend,
+            outputs=output_box
+        )
+
+        timer = gr.Timer(5)
+
+        timer.tick(
+            refresh_frontend,
+            outputs=output_box
+        )
+
     print("Gradio 正在启动。")
 
     demo.launch(