Update app.py

app.py

@@ -37,7 +37,17 @@ def chat_with_model(user_input, history):
     command = f"./llama.cpp/build/bin/main -m models/ZhongJing1_5-1_8b-fp16.gguf -n 256 --repeat_penalty 1.0 --color -i -r \"User:\" -f llama.cpp/prompts/TcmChat.txt"
     result = subprocess.run(command, shell=True, capture_output=True, text=True)
     
-    response = result.stdout.split("User:")[-1].strip()
+    # 解析输出
+    output_lines = result.stdout.split("\n")
+    response = ""
+    capture = False
+    for line in output_lines:
+        if "User:" in line:
+            capture = True
+        if capture:
+            response += line + "\n"
+    
+    response = response.strip()
     history.append((user_input, response))
     return history, history