Update chatbot.py

chatbot.py

@@ -139,6 +139,17 @@ class RAGChatbot:
             
             context = "\n\n".join(context_parts)
             
+            # ✅ Fallback: if no meaningful context remains after filtering
+            if not context.strip():
+                return {
+                    'response': "I couldn't find any good content to answer your question. Try ingesting a more informative page.",
+                    'sources': [],
+                    'retrieval_time': round(retrieval_time, 3),
+                    'generation_time': 0,
+                    'total_time': round(retrieval_time, 3),
+                    'context_used': 0
+                }
+
             # Step 3: Generate response using Groq
             generation_start = time.time()
             response = self._generate_response(message, context)
@@ -174,16 +185,19 @@ class RAGChatbot:
         Returns:
             Generated response
         """
-        system_prompt = """You are a helpful AI assistant that answers questions based on the provided context. 
+        system_prompt = """You are a helpful AI assistant.
+
+You must answer user questions based strictly on the provided context below. Do not use outside knowledge, do not make up facts, and do not guess.
 
-Guidelines:
-- Use ONLY the information provided in the context to answer questions
-- If the context doesn't contain enough information, say so clearly
-- Be accurate and cite specific details from the context
-- Provide comprehensive answers but stay focused on the question
-- If asked about sources, refer to the context provided
-- Be conversational and helpful in your tone
+If the context does not contain enough information, say clearly: "I don’t have enough information in the context to answer that."
+
+When you do answer:
+- Be accurate, concise, and truthful
+- Use facts and phrases from the context only
+- If asked for a source, refer to the matching context
+- Keep your tone friendly and professional
 """
+
         
         user_prompt = f"""Context:
 {context}