refactor UI and visualizer components for improved clarity

- Adjusted import order in ui.py for consistency.
- Removed unused chat_history attribute from ContextVisualizerUI.
- Deleted the format_token_breakdown method from ContextVisualizer to streamline functionality.
- Cleaned up unnecessary type imports in visualizer.py.

src/app/ui.py

@@ -1,7 +1,7 @@
 """Gradio UI for Context Engineering Visualizer"""
 
 import gradio as gr
-from typing import Tuple, List
+from typing import List, Tuple
 
 from .agent import ContextEngineeringAgent
 from config.settings import Settings
@@ -13,7 +13,6 @@ class ContextVisualizerUI:
     
     def __init__(self):
         self.agent = None
-        self.chat_history = []
         logger_ui.info("ContextVisualizerUI initialized")
         
     def initialize_agent(self) -> str:
@@ -109,13 +108,6 @@ class ContextVisualizerUI:
         
         return "\n".join(output)
     
-    def format_token_breakdown(self, visualizer) -> List[Tuple[str, int]]:
-        """Format token breakdown for chart"""
-        if not visualizer.token_counts:
-            return []
-        
-        return [(layer, tokens) for layer, tokens in visualizer.token_counts.items()]
-    
     def process_query(
         self, 
         query: str,

src/app/visualizer.py

@@ -1,6 +1,5 @@
 """Context window visualization component"""
 
-from typing import Dict, Any
 from datetime import datetime
 
 
@@ -24,11 +23,3 @@ class ContextVisualizer:
             "timestamp": datetime.now().isoformat()
         })
         self.token_counts[layer_name] = token_estimate
-        
-    def get_summary(self) -> Dict[str, Any]:
-        """Get structured summary of context"""
-        return {
-            "layers": [l["layer"] for l in self.context_layers],
-            "total_tokens": sum(self.token_counts.values()),
-            "breakdown": self.token_counts
-        }