Update app.py

app.py

@@ -75,7 +75,7 @@ class EnhancedContextDrivenChatbot:
     def is_follow_up_question(self, question):
         doc = nlp(question.lower())
         follow_up_indicators = set(['it', 'this', 'that', 'these', 'those', 'he', 'she', 'they', 'them'])
-        return any(token.text in follow_up_indicators for token in doc)
+        return any(token.text in follow_up_indicators for token in doc) or question.strip().startswith("What about")
 
     def extract_topics(self, text):
         doc = nlp(text)
@@ -115,7 +115,6 @@ class EnhancedContextDrivenChatbot:
         
         # Otherwise, it might be a new topic
         return question
-
     
     def rephrase_query(self, question, instructions=None):
         if not self.model:
@@ -124,7 +123,7 @@ class EnhancedContextDrivenChatbot:
         instruction_prompt = f"Instructions: {instructions}\n" if instructions else ""
         
         prompt = f"""
-        Given the conversation context, the current question, and any provided instructions, rephrase the question to include relevant context:
+        Given the conversation context, the current question, and any provided instructions, rephrase the question to include relevant context and rephrase it to more search-engine-friendly query:
         
         Conversation context: {self.get_context()}
         Current question: {question}
@@ -139,10 +138,11 @@ class EnhancedContextDrivenChatbot:
     def process_question(self, question):
         core_question, instructions = self.extract_instructions(question)
         
-        contextualized_question = self.get_most_relevant_context(core_question)
-        
         if self.is_follow_up_question(core_question):
+            contextualized_question = self.get_most_relevant_context(core_question)
             contextualized_question = self.rephrase_query(contextualized_question, instructions)
+        else:
+            contextualized_question = core_question
         
         topics = self.extract_topics(contextualized_question)
         
@@ -358,8 +358,12 @@ def ask_question(question: str, temperature: float, top_p: float, repetition_pen
 
     if web_search:
         contextualized_question, topics, entity_tracker, instructions = chatbot.process_question(question)
-        search_results = google_search(contextualized_question, num_results=3)
         
+        # Log the contextualized question for debugging
+        print(f"Contextualized question: {contextualized_question}")
+        
+        search_results = google_search(contextualized_question, num_results=3)
+
         context_chunks = []
         for result in search_results:
             if result["text"]: