Update app.py

app.py

@@ -125,6 +125,7 @@ def determine_response_type(message):
 
 def post_process_response(response, is_short_response):
     def count_questions(text):
+        # Compte le nombre de phrases interrogatives
         return len(re.findall(r'\?', text))
 
     def truncate_to_questions(text, max_questions):
@@ -142,13 +143,16 @@ def post_process_response(response, is_short_response):
         
         return ' '.join(truncated_sentences)
 
+    # Limiter à 2 questions, quelle que soit la longueur de la réponse
+    truncated_response = truncate_to_questions(response, 2)
+    
+    # Appliquer la limitation de longueur si nécessaire
     if is_short_response:
-        # Pour les réponses courtes, on ne limite pas le nombre de questions
-        return response.strip()
-    else:
-        # Pour les réponses moyennes et longues, on limite à 2 questions
-        return truncate_to_questions(response, 2).strip()
-
+        sentences = re.split(r'(?<=[.!?])\s+', truncated_response)
+        if len(sentences) > 2:
+            return ' '.join(sentences[:2]).strip()
+    
+    return truncated_response.strip()
 def check_coherence(response):
     sentences = re.split(r'(?<=[.!?])\s+', response)
     unique_sentences = set(sentences)