Update src/app.py

src/app.py

@@ -237,19 +237,39 @@ with tab1:
         
         # RAG Search
         context_txt = ""
-        sys_p = "You are a helpful assistant."
+        # 1. Default System Prompt (No RAG)
+        sys_p = "You are a helpful AI assistant."
         
         if use_rag:
-            with st.spinner("Searching DB..."):
+            with st.spinner("Searching Knowledge Base..."):
                 docs = rag_engine.search_knowledge_base(prompt, st.session_state.username)
                 if docs:
+                    # 2. Strict System Prompt (With RAG)
+                    # We relax the strictness slightly to allow for inference, 
+                    # while still demanding evidence.
                     sys_p = (
-                        "You are a Navy Document Analyst. Answer using ONLY the Context below. "
-                        "If the answer is not in the context, say 'I cannot find that information'."
+                        "You are a Navy Document Analyst. "
+                        "You must answer the user's question based PRIMARILY on the provided Context. "
+                        "If the Context contains the answer, output it clearly. "
+                        "If the Context does NOT contain the answer, simply state: "
+                        "'I cannot find that specific information in the documents provided.'"
                     )
-                    for d in docs: context_txt += f"\n---\n{d.page_content}"
                     
-        final_prompt = f"{prompt}\n\nCONTEXT:\n{context_txt}" if context_txt else prompt
+                    # 3. XML-Formatted Context Construction
+                    # This helps the model "see" the start and end of each chunk clearly.
+                    for i, d in enumerate(docs):
+                        src = d.metadata.get('source', 'Unknown')
+                        context_txt += f"<document index='{i+1}' source='{src}'>\n{d.page_content}\n</document>\n"
+                    
+        # 4. Construct Final User Payload
+        if context_txt:
+            final_prompt = (
+                f"User Question: {prompt}\n\n"
+                f"<context>\n{context_txt}\n</context>\n\n"
+                "Instruction: Answer the question using the context above."
+            )
+        else:
+            final_prompt = prompt
         
         # Generation
         with st.chat_message("assistant"):