Update app.py

app.py

@@ -17,9 +17,13 @@ from bm25 import BM25
 from vector_store import Vectorstore
 from retriever import Retriever
 
+print("done")
+
 embedder = Embedder()
 sessions: dict = {}
 
+print("embedder initialized")
+
 MODELS = [
     ("Qwen/Qwen2.5-72B-Instruct"),
     ("meta-llama/Llama-3.2-3B-Instruct"),
@@ -30,6 +34,7 @@ MODELS = [
 
 system_prompt = (
     "You are a helpful study assistant. Answer the user's question based ONLY on the provided context.\n\n"
+    "if the answer is not contained within the context, respond with: 'I couldn't find relevant information in the document.'\n\n(STRICT)"
     "FORMATTING RULES (STRICT):\n"
     "You MUST format your entire response using valid Markdown.\n"
     "1. Use `##` for main section headings.\n"
@@ -106,6 +111,7 @@ async def upload_file(file: UploadFile = File(...)):
 @app.post("/chat")
 async def chat(chat_req: ChatRequest):
     hf_token = os.environ.get("HF_TOKEN")
+    
     if not hf_token:
         raise HTTPException(status_code=500, detail="HF_TOKEN not configured")