Update app.py

app.py

@@ -347,27 +347,27 @@ class RAGPipeline:
 
     ### Added on Nov 2, 2024
     def postprocess_response(self, response: str) -> str:
-    """Clean up the generated response"""
-    try:
-        # Remove datetime patterns and other unwanted content
-        response = re.sub(r'\d{4}-\d{2}-\d{2}(?:T|\s)\d{2}:\d{2}:\d{2}(?:\.\d+)?(?:Z|[+-]\d{2}:?\d{2})?', '', response)
-        response = re.sub(r'User \d+:.*?(?=User \d+:|$)', '', response)
-        response = re.sub(r'\d{2}:\d{2}(?::\d{2})?(?:\s?(?:AM|PM))?', '', response)
-        response = re.sub(r'\d{1,2}[-/]\d{1,2}[-/]\d{2,4}', '', response)
-        response = re.sub(r'(?m)^User \d+:', '', response)
-        
-        # Clean up spacing but preserve intentional paragraph breaks
-        # Replace multiple newlines with two newlines (one paragraph break)
-        response = re.sub(r'\n\s*\n\s*\n+', '\n\n', response)
-        # Replace multiple spaces with single space
-        response = re.sub(r' +', ' ', response)
-        # Clean up beginning/end
-        response = response.strip()
-        
-        return response
-    except Exception as e:
-        logging.error(f"Error in postprocess_response: {str(e)}")
-        return response
+        """Clean up the generated response"""
+        try:
+            # Remove datetime patterns and other unwanted content
+            response = re.sub(r'\d{4}-\d{2}-\d{2}(?:T|\s)\d{2}:\d{2}:\d{2}(?:\.\d+)?(?:Z|[+-]\d{2}:?\d{2})?', '', response)
+            response = re.sub(r'User \d+:.*?(?=User \d+:|$)', '', response)
+            response = re.sub(r'\d{2}:\d{2}(?::\d{2})?(?:\s?(?:AM|PM))?', '', response)
+            response = re.sub(r'\d{1,2}[-/]\d{1,2}[-/]\d{2,4}', '', response)
+            response = re.sub(r'(?m)^User \d+:', '', response)
+            
+            # Clean up spacing but preserve intentional paragraph breaks
+            # Replace multiple newlines with two newlines (one paragraph break)
+            response = re.sub(r'\n\s*\n\s*\n+', '\n\n', response)
+            # Replace multiple spaces with single space
+            response = re.sub(r' +', ' ', response)
+            # Clean up beginning/end
+            response = response.strip()
+            
+            return response
+        except Exception as e:
+            logging.error(f"Error in postprocess_response: {str(e)}")
+            return response
 
     def process_query(self, query: str, placeholder) -> str:
         try: