Update app.py

app.py

@@ -47,6 +47,11 @@ h2 {
     color:darkgreen !important;
     margin-top: -5px;  /* Adjust if needed to align with annotation */
 }
+
+.bibliography-content {
+    color:#a4a4a4;
+    margin-top: -5px;  /* Adjust if needed to align with annotation */
+}
 </style>
 """
 
@@ -122,6 +127,8 @@ def transform_chunks(marianne_segmentation):
             html_output.append(f'<div class="manuscript"><div class="annotation">{result_entity}</div><div class="content title-content"><h2>{word}</h2></div></div>')
         elif entity_group == 'bibliography':
             html_output.append(f'<div class="manuscript"><div class="annotation">{result_entity}</div><div class="content bibliography-content">{word}</div></div>')
+        elif entity_group == 'paratext':
+            html_output.append(f'<div class="manuscript"><div class="annotation">{result_entity}</div><div class="content paratext-content">{word}</div></div>')
         else:
             html_output.append(f'<div class="manuscript"><div class="annotation">{result_entity}</div><div class="content">{word}</div></div>')
 
@@ -151,7 +158,7 @@ class MistralChatBot:
     
         classified_list = pd.concat(classified_list)
         out = transform_chunks(classified_list)
-        generated_text = f'{css}<h2 style="text-align:center">Réponse</h2>\n<div class="generation">{out}</div>'
+        generated_text = f'{css}<h2 style="text-align:center">Edited text</h2>\n<div class="generation">{out}</div>'
         return generated_text
 
 # Create the Falcon chatbot instance
@@ -170,7 +177,7 @@ examples = [
 demo = gr.Blocks()
 
 with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as demo:
-    gr.HTML("""<h1 style="text-align:center">Editorialize your text/h1>""")
+    gr.HTML("""<h1 style="text-align:center">Editorialize your text</h1>""")
     text_input = gr.Textbox(label="Your text", type="text", lines=1)
     text_button = gr.Button("Identify editorial structures")
     text_output = gr.HTML(label="Corrected text")