Update app.py

app.py

@@ -32,11 +32,11 @@ tokenizer.pad_token = tokenizer.eos_token
 tokenizer.pad_token_id = 1
 
 # Define variables 
-temperature = 0.0  
-max_new_tokens = 1400
+temperature = 0.25  
+max_new_tokens = 1200
 top_p = 0.95
 repetition_penalty = 1.0
-min_new_tokens = 700
+min_new_tokens = 600
 early_stopping = False
 
 # Connect to the LanceDB database
@@ -130,15 +130,18 @@ def format_references(text):
 
     import re
     for match in re.finditer(ref_pattern, text):
-        # Add text before the reference
-        parts.append(text[current_pos:match.start()])
+        # Add text before the reference up to the period
+        text_before = text[current_pos:match.start()].rstrip()
+        if text_before.endswith('.'):
+            text_before = text_before[:-1]  # Remove the period temporarily
+        parts.append(text_before)
         
         # Extract reference components
-        ref_id = match.group(1)  # The source ID
-        ref_text = match.group(2).strip()  # The reference text
+        ref_id = match.group(1)
+        ref_text = match.group(2).strip()
         
-        # Create tooltip HTML with source ID in bold and add <br> after each reference
-        tooltip_html = f'<span class="tooltip">[{ref_number}]<span class="tooltiptext"><strong>{ref_id}</strong>: {ref_text}</span></span><br>'
+        # Add the reference with period and line break
+        tooltip_html = f'<span class="tooltip">[{ref_number}]<span class="tooltiptext"><strong>{ref_id}</strong>: {ref_text}</span></span>.<br>'
         parts.append(tooltip_html)
         
         current_pos = match.end()
@@ -169,7 +172,8 @@ css = """
 .tooltip {
     position: relative;
     display: inline-block;
-    color: #2563eb;
+    color: navy;  /* Changed from #2563eb to navy */
+    font-weight: bold;  /* Make it bold for better visibility */
     cursor: pointer;
 }
 .tooltip .tooltiptext {
@@ -234,16 +238,19 @@ with demo:
     </div>
     """)
     
+    # Centered input section
+    with gr.Column(scale=1):
+        text_input = gr.Textbox(label="Votre question ou votre instruction", lines=3)
+        text_button = gr.Button("Interroger pleias-RAG")
+    
+    # Analysis and Response in side-by-side columns
     with gr.Row():
-        # Left column for input, button and analysis
+        # Left column for analysis
         with gr.Column(scale=2):
-            text_input = gr.Textbox(label="Votre question ou votre instruction", lines=3)
-            text_button = gr.Button("Interroger pleias-RAG")
-            text_output = gr.HTML(label="Analyse des sources")  # Moved here
-            
+            text_output = gr.HTML(label="Analyse des sources")
         # Right column for response
         with gr.Column(scale=3):
-            response_output = gr.HTML(label="Réponse")  # Moved here
+            response_output = gr.HTML(label="Réponse")
             
     # Sources at the bottom
     with gr.Row():