Update app.py

app.py

@@ -271,27 +271,6 @@ def reset_interface():
     """Reset all interface components."""
     return gr.update(value=None), gr.update(value=None), gr.update(value="")
 
-def create_legend_html():
-    """Create HTML for the class legend."""
-    cmap = colormap(N=len(classes_map), normalized=False)
-    legend_items = []
-    
-    for class_id, class_name in classes_map.items():
-        color_rgb = cmap[class_id % len(cmap)]
-        color_hex = f"#{color_rgb[0]:02x}{color_rgb[1]:02x}{color_rgb[2]:02x}"
-        legend_items.append(f"""
-        <div style='display: flex; align-items: center; padding: 8px; margin: 4px; background-color: #f8f9fa; border-radius: 6px;'>
-            <div style='width: 24px; height: 24px; background-color: {color_hex}; margin-right: 12px; border: 2px solid #dee2e6; border-radius: 4px;'></div>
-            <span style='font-weight: 500; color: #495057;'>{class_name}</span>
-        </div>
-        """)
-    
-    return f"""
-    <div style='display: grid; grid-template-columns: repeat(auto-fit, minmax(200px, 1fr)); gap: 8px; padding: 16px; background-color: #ffffff; border-radius: 8px; border: 1px solid #e9ecef;'>
-        {''.join(legend_items)}
-    </div>
-    """
-
 if __name__ == "__main__":
     print(f"🚀 Starting Document Layout Analysis App")
     print(f"📱 Device: {device}")
@@ -430,17 +409,6 @@ if __name__ == "__main__":
                         label="Overlay Transparency",
                         info="Transparency of detection overlays"
                     )
-                
-                # Visualization Options Section
-                with gr.Group(elem_classes=["control-section"]):
-                    gr.HTML("<h3>🎨 Visualization Options</h3>")
-                    
-                    show_labels_checkbox = gr.Checkbox(
-                        value=True,
-                        label="Show Class Labels",
-                        info="Display class names and confidence scores on detections",
-                        interactive=True
-                    )
             
             # RIGHT COLUMN - Results and Output
             with gr.Column(scale=1, elem_classes=["panel-right"]):
@@ -463,11 +431,17 @@ if __name__ == "__main__":
                         lines=3,
                         placeholder="Detection results will appear here..."
                     )
-        
-        # Legend Section (Full Width)
-        with gr.Group():
-            with gr.Accordion("📋 Class Legend - All Detectable Elements", open=False):
-                gr.HTML(create_legend_html())
+
+                    # Visualization Options Section
+                with gr.Group(elem_classes=["control-section"]):
+                    gr.HTML("<h3>🎨 Visualization Options</h3>")
+                    
+                    show_labels_checkbox = gr.Checkbox(
+                        value=True,
+                        label="Show Class Labels",
+                        info="Display class names and confidence scores on detections",
+                        interactive=True
+                    )
         
         # Event Handlers
         load_btn.click(