Update app.py

app.py

@@ -32,7 +32,7 @@ tokenizer.pad_token = tokenizer.eos_token
 tokenizer.pad_token_id = 1
 
 # Define variables 
-temperature = 0.25  
+temperature = 0.0  
 max_new_tokens = 1200
 top_p = 0.95
 repetition_penalty = 1.0
@@ -129,8 +129,10 @@ def format_references(text):
     ref_number = 1
 
     import re
-    # Remove leading dots from lines
-    text = re.sub(r'^\s*\.\s*', '', text, flags=re.MULTILINE)
+    # Remove leading dots and whitespace from lines
+    lines = text.split('\n')
+    cleaned_lines = [line.lstrip('. ').rstrip() for line in lines]
+    text = '\n'.join(cleaned_lines)
     
     for match in re.finditer(ref_pattern, text):
         # Add text before the reference
@@ -141,7 +143,7 @@ def format_references(text):
         ref_id = match.group(1)
         ref_text = match.group(2).strip()
         
-        # Add the reference with period and line break 
+        # Add the reference with period and line break
         tooltip_html = f'<span class="tooltip"><strong>[{ref_number}]</strong><span class="tooltiptext"><strong>{ref_id}</strong>: {ref_text}</span></span>.<br>'
         parts.append(tooltip_html)
         
@@ -173,7 +175,7 @@ css = """
 .tooltip {
     position: relative;
     display: inline-block;
-    color: #007FFF;  
+    color: #0000FF;  
     font-weight: bold;
     cursor: pointer;
 }
@@ -232,9 +234,14 @@ with demo:
     <div style="display: flex; justify-content: center; width: 100%; background-color: black; padding: 5px 0;">
         <pre style="font-family: monospace; line-height: 1.2; font-size: 24px; color: #00ffea; margin: 0;">
 
-╔═══════════════════╗
-║   pleias-RAG 1.0  ║
-╚═══════════════════╝
+       _      _                     ______  ___  _____ 
+      | |    (_)                    | ___ \/ _ \|  __ \
+ _ __ | | ___ _  __ _ ___   ______  | |_/ / /_\ \ |  \/
+| '_ \| |/ _ \ |/ _` / __| |______| |    /|  _  | | __ 
+| |_) | |  __/ | (_| \__ \          | |\ \| | | | |_\ \
+| .__/|_|\___|_|\__,_|___/          \_| \_\_| |_/\____/
+| |                                                    
+|_|                                                    
         </pre>
     </div>
     """)