Update app.py

app.py

@@ -397,40 +397,46 @@ with gr.Blocks(css=custom_css) as demo:
                 value="<i>No analysis yet. Upload a document and click Analyze.</i>"
             )
     
-    gr.Markdown("### Model Performance")
-    gr.HTML(
-        f"""
-        <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>
-            <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;'>
-                <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>
-        """
-    )
-    
-    gr.Markdown(
-        """
-        ---
-        **Model Architecture:**
-        - **Localization:** MobileNetV3-Small + UNet (Dice: 62.1%, IoU: 45.1%)
-        - **Classification:** LightGBM with 526 features (Accuracy: 88.97%)
-        - **Training:** 140K samples (DocTamper + SCD + FCD datasets)
-        """
-    )
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### Model Performance")
+            gr.HTML(
+                f"""
+                <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>
+                    <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;'>
+                        <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(
+                """
+                **Localization:** MobileNetV3-Small + UNet
+                - Dice: 62.1% | IoU: 45.1%
+                
+                **Classification:** LightGBM with 526 features
+                - Accuracy: 88.97%
+                
+                **Training:** 140K samples from DocTamper dataset
+                """
+            )
     
     # Event handlers
     analyze_btn.click(