Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from huggingface_hub import hf_hub_download
 import os
+import subprocess
 
 # 模型下载链接
 model_url = "https://huggingface.co/CMLL/ZhongJing-2-1_8b-GGUF/resolve/main/ZhongJing1_5-1_8b-fp16.gguf"
@@ -26,8 +27,16 @@ with open(os.path.join(prompts_dir, "TcmChat.txt"), "w") as f:
 
 # Gradio 接口
 def chat_with_model(user_input, history):
-    prompt = f"You are a helpful TCM medical assistant named 仲景中医大语言模型.\nUser: {user_input}\nAssistant:"
-    response = os.popen(f"./llama.cpp/build/bin/main -m {model_path} -n 256 --repeat_penalty 1.0 --color -i -r \"User:\" -f {os.path.join(prompts_dir, 'TcmChat.txt')}").read()
+    prompt = f"User: {user_input}\nAssistant:"
+    with open(os.path.join(prompts_dir, "TcmChat.txt"), "a") as f:
+        f.write(prompt + "\n")
+    
+    # 执行命令并捕获输出
+    command = f"./llama.cpp/build/bin/main -m {model_path} -n 256 --repeat_penalty 1.0 --color -i -r \"User:\" -f {os.path.join(prompts_dir, 'TcmChat.txt')}"
+    result = subprocess.run(command, shell=True, capture_output=True, text=True)
+    
+    response = result.stdout.split("User:")[-1].strip()
+    
     history.append((user_input, response))
     return history, history