Update app.py

app.py

@@ -604,8 +604,6 @@ import re
 
 #     return final_response
 
-import re
-
 def clean_response(response_text):
     # Remove system and user tags
     response_text = re.sub(r'<\|system\|>.*?<\|end\|>', '', response_text, flags=re.DOTALL)
@@ -624,9 +622,9 @@ def clean_response(response_text):
 
     # Remove the entire 'Document(metadata=...' and any mention of it from the response
     response_text = re.sub(r'Document\(metadata=\{.*?\},page_content=', '', response_text, flags=re.DOTALL)
-    
-    # Remove any mention of "Document:" in the response
-    response_text = re.sub(r'- Document:.*', '', response_text)
+
+    # Extract top 5 fetched results (based on some identifier you have)
+    top_results = response_text.split('\n')[:5]  # Assuming top 5 results are first 5 lines of content
 
     # Remove any unwanted escape characters like \u and \u00
     response_text = re.sub(r'\\u[0-9A-Fa-f]{4}', '', response_text)
@@ -638,24 +636,21 @@ def clean_response(response_text):
     # Remove the phrase "Sure! The Responses are as follows:" from the actual content
     response_text = re.sub(r'Sure! The Responses are as follows:', '', response_text).strip()
 
-    # Extract and limit to top 5 results
-    result_lines = response_text.splitlines()[:5]
-    top_5_results = "\n".join(result_lines)
-
     # Clean up the text by removing extra whitespace
     cleaned_response = re.sub(r'\s+', ' ', response_text).strip()
 
-    # Format the final response with bullet points in Italian
+    # Format the final response with bullet points
     final_response = f"""
-Nome del documento: {document_name}
-Numero di pagina: {page_number}
+Sure! Here is the response for your Query:
+• Document name - {document_name}
+• Page No - {page_number}
+• Top 5 Fetched Results:
+  {''.join([f'{i+1}. {result.strip()}\n' for i, result in enumerate(top_results)])}
+• Actual Response: {cleaned_response}
+"""
 
-Risultati principali (Top 5):
-{top_5_results}
+    return final_response
 
-Risultato effettivo:
-{cleaned_response}
-"""