Update app.py

app.py

@@ -42,19 +42,32 @@ class Agent1:
                 text.strip().endswith('?') or 
                 any(word in self.question_words for word in words))
 
+    def find_noun_phrases(self, sentence):
+        tokens = nltk.pos_tag(word_tokenize(sentence))
+        noun_phrases = []
+        current_phrase = []
+        for word, tag in tokens:
+            if tag.startswith('NN'):
+                current_phrase.append(word)
+            elif current_phrase:
+                noun_phrases.append(' '.join(current_phrase))
+                current_phrase = []
+        if current_phrase:
+            noun_phrases.append(' '.join(current_phrase))
+        return noun_phrases
+
     def replace_pronoun(self, questions: List[str]) -> List[str]:
         if len(questions) < 2:
             return questions
 
-        # Simple NLP to identify potential nouns in the first question
-        tokens = nltk.pos_tag(word_tokenize(questions[0]))
-        nouns = [word for word, pos in tokens if pos.startswith('NN')]
-
-        if not nouns:
+        # Find noun phrases in the first question
+        noun_phrases = self.find_noun_phrases(questions[0])
+        
+        if not noun_phrases:
             return questions
 
-        # Use the last noun as the antecedent
-        antecedent = nouns[-1]
+        # Use the last noun phrase as the antecedent
+        antecedent = noun_phrases[-1]
 
         # Replace pronouns in subsequent questions
         for i in range(1, len(questions)):