Update app.py

app.py

@@ -397,6 +397,8 @@ with gr.Blocks(css=custom_css) as demo:
                 value="<i>No analysis yet. Upload a document and click Analyze.</i>"
             )
     
+    gr.Markdown("---")
+    
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Model Performance")
@@ -405,36 +407,32 @@ with gr.Blocks(css=custom_css) as demo:
                 <div style='padding:12px; border:1px solid #444; border-radius:10px; background:var(--background-fill-primary);'>
                     <p style='margin-top:0; margin-bottom:12px;'><b>Trained Model Performance:</b></p>
                     
-                    <b>Segmentation Dice:</b>
+                    <b>Segmentation Dice: {MODEL_METRICS['segmentation']['dice']*100:.2f}%</b>
                     <div style='width:100%; background:#333; height:12px; border-radius:6px; margin-bottom:12px;'>
                         <div style='width:{MODEL_METRICS['segmentation']['dice']*100:.1f}%; background:#4169E1; height:12px; border-radius:6px;'></div>
                     </div>
                     
-                    <b>Classification Accuracy:</b>
-                    <div style='width:100%; background:#333; height:12px; border-radius:6px; margin-bottom:6px;'>
+                    <b>Classification Accuracy: {MODEL_METRICS['classification']['overall_accuracy']*100:.2f}%</b>
+                    <div style='width:100%; background:#333; height:12px; border-radius:6px;'>
                         <div style='width:{MODEL_METRICS['classification']['overall_accuracy']*100:.1f}%; background:#5cb85c; height:12px; border-radius:6px;'></div>
                     </div>
-                    
-                    <small style='opacity:0.8;'>
-                        Dice: {MODEL_METRICS['segmentation']['dice']*100:.1f}% | 
-                        IoU: {MODEL_METRICS['segmentation']['iou']*100:.1f}% | 
-                        Accuracy: {MODEL_METRICS['classification']['overall_accuracy']*100:.1f}%
-                    </small>
                 </div>
                 """
             )
         
         with gr.Column(scale=1):
             gr.Markdown("### Model Architecture")
-            gr.Markdown(
+            gr.HTML(
                 """
-                **Localization:** MobileNetV3-Small + UNet
-                - Dice: 62.1% | IoU: 45.1%
-                
-                **Classification:** LightGBM with 526 features
-                - Accuracy: 88.97%
-                
-                **Training:** 140K samples from DocTamper dataset
+                <div style='padding:12px; border:1px solid #444; border-radius:10px; background:var(--background-fill-primary);'>
+                    <p style='margin-top:0; margin-bottom:8px;'><b>Localization:</b> MobileNetV3-Small + UNet</p>
+                    <p style='margin:4px 0; font-size:0.9em;'>Dice: 62.12% | IoU: 45.06% | Precision: 70.77% | Recall: 55.36%</p>
+                    
+                    <p style='margin-top:12px; margin-bottom:8px;'><b>Classification:</b> LightGBM with 526 features</p>
+                    <p style='margin:4px 0; font-size:0.9em;'>Train Accuracy: 90.53% | Val Accuracy: 88.97%</p>
+                    
+                    <p style='margin-top:12px; margin-bottom:0;'><b>Training:</b> 140K samples from DocTamper dataset</p>
+                </div>
                 """
             )