Update rag_service.py

rag_service.py

@@ -110,6 +110,7 @@ class RAGService:
         user_question = kwargs["question"]
     
         context_text = ""
+        prompt_content = []
         if len(docs_by_type["texts"]) > 0:
             for text_element in docs_by_type["texts"]:
                 context_text += str(text_element)
@@ -150,10 +151,57 @@ class RAGService:
                 [HumanMessage(content=prompt_template.strip())]  # plain string
             )
 
+    # def ask_question(self, question: str):
+    #     """Process a question and return response"""
+    #     response = self.chain_with_sources.invoke(question)
+    #     return response['response']
+
     def ask_question(self, question: str):
-        """Process a question and return response"""
+    """Process a question and return response"""
+    try:
+        # First attempt
         response = self.chain_with_sources.invoke(question)
+        
+        # Check if response is valid
+        if not response or not response.get('response') or len(response['response'].strip()) < 0:
+            # Try to clarify what the user might be asking
+            return self._handle_ambiguous_question(question)
+        
         return response['response']
+        
+    except Exception as e:
+        print(f"Error in ask_question: {e}")
+        return self._handle_ambiguous_question(question)
+
+    def _handle_ambiguous_question(self, question: str):
+        """Handle ambiguous or complex questions"""
+        try:
+            # Use LLM to understand what the user might be asking
+            llm = ChatGroq(model="llama-3.1-8b-instant", groq_api_key=self.groq_key)
+            
+            clarification_prompt = f"""
+            The user asked: "{question}"
+            
+            This question seems unclear or complex. Please help clarify by:
+            1. Identifying what the user might be asking about
+            2. Suggesting a clearer way to phrase the question
+            3. If possible, provide a brief answer based on your best understanding
+            
+            Format your response as:
+            "I think you're asking about: [topic]
+            
+            Suggested clearer question: [clearer version]
+            
+            Based on my understanding: [brief answer if possible]
+            
+            Please rephrase your question if this isn't what you meant."
+            """
+            
+            response = llm.invoke(clarification_prompt)
+            return response.content
+            
+        except Exception as e:
+            return "I'm having trouble understanding your question. Could you please rephrase it more clearly? For example, be more specific about what you're looking for."
 
 
 # Create a global instance