Update rag_service.py

rag_service.py

@@ -113,19 +113,28 @@ class RAGService:
             for text_element in docs_by_type["texts"]:
                 context_text += str(text_element)
 
-        prompt_template = f"""
-       You are a helpful AI assistant. You have access to the following context from uploaded documents:
-       Context: {context_text}
-       Question: {user_question}
-       
-       Instructions:
-       - If the question is related to the provided context, use that information to answer
-       - If the question is not related to the context or is a general query, answer based on your general knowledge
-       - Be clear about whether you're using the provided context or general knowledge in your response
-       """
+        # If no relevant context found, make it fully general
+        if not context_text.strip():
+            prompt_template = f""" 
+            You are a helpful AI assistant. Answer the following question:
+            Question: {user_question}
+            """
+        else:
+            prompt_template = f"""
+            You are a helpful AI assistant. You have access to the following context from uploaded documents:
+
+            Context: {context_text}
+
+            Question: {user_question}
+
+            Instructions:
+            - If the question is related to the provided context, use that information to answer
+            - If the question is not related to the context or is a general query, answer based on your general knowledge
+            - Be clear about whether you're using the provided context or general knowledge in your response
+            """
 
         prompt_content = [{"type": "text", "text": prompt_template}]
-
+        
         if len(docs_by_type["images"]) > 0:
             for image in docs_by_type["images"]:
                 prompt_content.append(