Update app.py

app.py

@@ -585,17 +585,19 @@ def clean_response(response_text):
         document_name = "Unknown"
         page_number = "Unknown"
 
-    # Remove the document metadata and unnecessary characters from the response
-    response_text = re.sub(r'Document\(metadata=\{.*?\},page_content=', '', response_text, flags=re.DOTALL)
+    # Remove document metadata and other unnecessary parts
+    response_text = re.sub(r"Document\(metadata=\{.*?\}\)", '', response_text, flags=re.DOTALL)
+
+    # Remove unwanted parts like "Nome del documento" and "Numero di pagina"
+    response_text = re.sub(r"Nome del documento:.*\n?", '', response_text)
+    response_text = re.sub(r"Numero di pagina:.*\n?", '', response_text)
 
     # Clean up and extract the relevant response content
     response_text = re.sub(r'\\u[0-9A-Fa-f]{4}', '', response_text)  # Remove any unwanted escape characters
 
-    # Ensure proper spacing
-    response_text = re.sub(r'([a-zA-Z])(\d)', r'\1 \2', response_text)
-    response_text = re.sub(r'(\d)([a-zA-Z])', r'\1 \2', response_text)
-
-    # Strip excess spaces and format the final response content
+    # Ensure proper spacing and remove extra text around the "Contenuto effettivo" and "Risposta" parts
+    response_text = re.sub(r'Contenuto effettivo: ', '', response_text)
+    response_text = re.sub(r'Risposta: ', '', response_text)
     cleaned_response = response_text.strip()
 
     # Handle blank or empty responses
@@ -604,7 +606,7 @@ def clean_response(response_text):
 
     # Create a table format for the document name, page number, and response
     final_response = f"""
-    Here is the response of your Query <br>
+    <br>Here is the response of your Query <br>
     <table style="width:100%;border: 1px solid black;border-collapse: collapse;">
         <tr>
             <th style="border: 1px solid black;padding: 8px;text-align: left;">Document Name</th>
@@ -630,6 +632,7 @@ def clean_response(response_text):
 
 
 
+
 # Define a new template specifically for GPT-4o-mini in VDB Details mode
 gpt4o_mini_template_details = f"""
 As a highly specialized assistant, I provide precise, detailed, and informative responses. On this bright day of {current_date}, I'm equipped to assist with all your queries about Birmingham, Alabama, offering detailed insights tailored to your needs.