Update app.py

app.py

@@ -97,14 +97,14 @@ class pleiasBot:
                 analysis = parts[0].strip()
                 answer = parts[1].replace("<|answer_start|>", "").replace("<|answer_end|>", "").strip()
                 
-                # Format each section separately
-                analysis_text = f'<div class="generation">' + format_references(analysis) + "</div>"
-                answer_text = '<h2 style="text-align:center">Réponse</h3>\n<div class="generation">' + format_references(answer) + "</div>"
+                # Format each section with matching h2 titles
+                analysis_text = '<h2 style="text-align:center">Analyse des sources</h2>\n<div class="generation">' + format_references(analysis) + "</div>"
+                answer_text = '<h2 style="text-align:center">Réponse</h2>\n<div class="generation">' + format_references(answer) + "</div>"
             else:
                 analysis_text = ""
                 answer_text = format_references(generated_text)
             
-            fiches_html = '<h2 style="text-align:center">Sources</h3>\n' + fiches_html
+            fiches_html = '<h2 style="text-align:center">Sources</h2>\n' + fiches_html
             return analysis_text, answer_text, fiches_html
 
         except Exception as e:
@@ -216,7 +216,7 @@ demo = gr.Blocks(css=css)
 with demo:
     # Header with black bar
     gr.HTML("""
-    <div style="display: flex; justify-content: center; width: 100%; background-color: black; padding: 7px 0;">
+    <div style="display: flex; justify-content: center; width: 100%; background-color: black; padding: 10px 0;">
         <pre style="font-family: monospace; line-height: 1.2; font-size: 24px; color: #00ffea; margin: 0;">
 ╔═══════════════════╗
 ║   pleias-RAG 1.0  ║
@@ -225,18 +225,17 @@ with demo:
     </div>
     """)
     
-    # Input section
-    with gr.Column(scale=2):
-        text_input = gr.Textbox(label="Votre question ou votre instruction", lines=3)
-        text_button = gr.Button("Interroger pleias-RAG")
-    
-    # Two columns for output: Analysis and Response
     with gr.Row():
+        # Left column for input, button and answer
         with gr.Column(scale=2):
-            text_output = gr.HTML(label="Analyse des sources")
-        with gr.Column(scale=3):
+            text_input = gr.Textbox(label="Votre question ou votre instruction", lines=3)
+            text_button = gr.Button("Interroger pleias-RAG")
             response_output = gr.HTML(label="Réponse")
             
+        # Right column for analysis
+        with gr.Column(scale=3):
+            text_output = gr.HTML(label="Analyse des sources")
+            
     # Sources at the bottom
     with gr.Row():
         embedding_output = gr.HTML(label="Les sources utilisées")