Update app.py

app.py

@@ -46,18 +46,14 @@ SYSTEM_PROMPT = (
 )
 
 def build_prompt(messages, context=""):
-    prompt = f"<|im_start|>system\n{SYSTEM_PROMPT}\n
-Context:
-{context}<|im_end|>
-"
+    prompt = f"<|im_start|>system\n{SYSTEM_PROMPT}\n\nContext:\n{context}<|im_end|>\n"
     for msg in messages:
         role = msg["role"]
-        prompt += f"<|im_start|>{role}\n{msg['content']}<|im_end|>
-"
-    prompt += "<|im_start|>assistant
-"
+        prompt += f"<|im_start|>{role}\n{msg['content']}<|im_end|>\n"
+    prompt += "<|im_start|>assistant\n"
     return prompt
 
+
 # --- RAG Embedding and Search ---
 @st.cache_resource
 def embed_uploaded_files(files):