Update app.py

app.py

@@ -715,7 +715,7 @@ ner_manager = HybridNERManager()
 def process_text(text, standard_entities, custom_entities_str, confidence_threshold, selected_model, progress=gr.Progress()):
     """Main processing function for Gradio interface with progress tracking"""
     if not text.strip():
-        return "❌ Please enter some text to analyse", "", ""
+        return "❌ Please enter some text to analyse", "", "", gr.update(visible=False)
 
     progress(0.1, desc="Initialising...")
     
@@ -728,7 +728,7 @@ def process_text(text, standard_entities, custom_entities_str, confidence_thresh
     selected_standard = [entity for entity in standard_entities if entity]
 
     if not selected_standard and not custom_entities:
-        return "❌ Please select at least one common entity type OR enter custom entity types", "", ""
+        return "❌ Please select at least one common entity type OR enter custom entity types", "", "", gr.update(visible=False)
 
     progress(0.2, desc="Loading models...")
     
@@ -747,7 +747,7 @@ def process_text(text, standard_entities, custom_entities_str, confidence_thresh
         all_entities.extend(custom_entity_results)
 
     if not all_entities:
-        return "❌ No entities found. Try lowering the confidence threshold or using different entity types.", "", ""
+        return "❌ No entities found. Try lowering the confidence threshold or using different entity types.", "", "", gr.update(visible=False)
 
     progress(0.8, desc="Processing results...")
     
@@ -777,7 +777,7 @@ def process_text(text, standard_entities, custom_entities_str, confidence_thresh
 
     progress(1.0, desc="Complete!")
     
-    return summary, legend_html + highlighted_html, results_html
+    return summary, legend_html + highlighted_html, results_html, gr.update(visible=True)
 
 # Create Gradio interface
 def create_interface():
@@ -825,9 +825,9 @@ def create_interface():
                 
                 # Add confidence threshold explanation
                 gr.HTML("""
-                <details style="margin: 10px 0; padding: 10px; background-color: #f0f9ff; border-radius: 8px; border: 1px solid #bee3f8;">
-                    <summary style="cursor: pointer; font-weight: bold; padding: 5px; color: #2563eb;">
-                        ℹ️ Understanding Confidence Threshold
+                <details style="margin: 10px 0; padding: 10px; background-color: #f8f9fa; border-radius: 8px; border: 1px solid #ddd;">
+                    <summary style="cursor: pointer; font-weight: bold; padding: 5px; color: #1976d2;">
+                        ℹ️ Understanding the Confidence Threshold
                     </summary>
                     <div style="margin-top: 10px; padding: 10px; font-size: 14px;">
                         <p style="margin: 0 0 10px 0;">The <strong>confidence threshold</strong> controls how certain the model needs to be before identifying an entity:</p>
@@ -958,6 +958,9 @@ def create_interface():
         
         analyse_btn = gr.Button("🔍 Analyse Text", variant="primary", size="lg")
         
+        # Add horizontal line separator after the button
+        gr.HTML("<hr style='margin-top: 20px; margin-bottom: 15px;'>")
+        
         # Output sections
         with gr.Row():
             summary_output = gr.Markdown(label="Summary")
@@ -965,8 +968,8 @@ def create_interface():
         with gr.Row():
             highlighted_output = gr.HTML(label="Highlighted Text")
         
-        # Results section
-        with gr.Row():
+        # Results section (initially hidden)
+        with gr.Row(visible=False) as results_row:
             with gr.Column():
                 gr.Markdown("### 📋 Detailed Results")
                 results_output = gr.HTML(label="Entity Results")
@@ -981,14 +984,15 @@ def create_interface():
                 confidence_threshold,
                 model_dropdown
             ],
-            outputs=[summary_output, highlighted_output, results_output]
+            outputs=[summary_output, highlighted_output, results_output, results_row]
         )
         
-        # Updated examples text
-        gr.Markdown("""
-        ### 💡 No example text to test? No problem!
-        Simply click on one of the examples provided below, and the fields will be populated for you.
-        """, elem_id="examples-heading")
+        # Updated examples text with reduced spacing
+        with gr.Column():
+            gr.Markdown("""
+            ### 💡 No example text to test? No problem!
+            Simply click on one of the examples provided below, and the fields will be populated for you.
+            """, elem_id="examples-heading")
         
         gr.Examples(
             examples=[
@@ -1024,7 +1028,7 @@ def create_interface():
             label="Examples"
         )
         
-        # Add custom CSS to make Examples label black
+        # Add custom CSS to make Examples label black and improve entity label readability
         gr.HTML("""
         <style>
             /* Make the Examples label text black */
@@ -1040,6 +1044,20 @@ def create_interface():
             #examples-heading + div .label-text {
                 color: black !important;
             }
+            
+            /* Improve readability of entity checkbox labels */
+            .gradio-checkbox-group label {
+                font-size: 14px !important;
+                font-weight: normal !important;
+                color: #333333 !important;
+                line-height: 1.5 !important;
+            }
+            
+            /* Make the checkbox labels match the entity definitions style */
+            input[type="checkbox"] + span {
+                font-size: 14px !important;
+                color: #333333 !important;
+            }
         </style>
         """)
         
@@ -1084,7 +1102,7 @@ def create_interface():
         funded research project:<br>
         <em>Extracting Keywords from Crowdsourced Collections</em>.
     </p>
-    <p style="font-size: 14px; line-height: 1.8; margin: 0;"><p>
+    <p><p style="font-size: 14px; line-height: 1.8; margin: 0;">
     The code for this tool was built with the aid of Claude Opus 4.</p>
     </p>
 </div>