Update timeline_generator.py

timeline_generator.py

@@ -62,16 +62,15 @@ def generate_timeline_diagram(json_input: str, output_format: str) -> str:
             }
         )
         
-        base_color = '#19191a' # Hardcoded base color
+        base_color = '#19191a'
         
         title = data.get('title', '')
         events = data.get('events', [])
-        events_per_row = data.get('events_per_row', 4)  # Default to 4 events per row
+        events_per_row = data.get('events_per_row', 4)  
         
         if not events:
             raise ValueError("Timeline must contain at least one event")
 
-        # Add title node if provided
         if title:
             dot.node(
                 'title',
@@ -80,19 +79,17 @@ def generate_timeline_diagram(json_input: str, output_format: str) -> str:
                 fontsize='18',
                 fontweight='bold',
                 fontcolor=base_color,
-                pos="6,2!"  # Centered at top (adjust x-coordinate based on timeline width)
+                pos="6,2!" 
             )
         
         total_events = len(events)
         
-        # Create all event nodes first
         for i, event in enumerate(events):
             event_id = event.get('id', f'event_{i}')
             event_label = event.get('label', f'Event {i+1}')
             event_date = event.get('date', '')
             event_description = event.get('description', '')
             
-            # Create full label with date and description
             if event_date and event_description:
                 full_label = f"{event_date}\\n{event_label}\\n{event_description}"
             elif event_date:
@@ -102,7 +99,6 @@ def generate_timeline_diagram(json_input: str, output_format: str) -> str:
             else:
                 full_label = event_label
             
-            # Calculate color opacity based on position in timeline
             if total_events == 1:
                 opacity = 'FF'
             else:
@@ -112,17 +108,14 @@ def generate_timeline_diagram(json_input: str, output_format: str) -> str:
             node_color = f"{base_color}{opacity}"
             font_color = 'white' if i < total_events * 0.7 else 'black'
             
-            # Calculate position for serpentine layout
             row = i // events_per_row
             col = i % events_per_row
             
-            # For odd rows, reverse the column position to create serpentine effect
             if row % 2 == 1:
                 visual_col = events_per_row - 1 - col
             else:
                 visual_col = col
             
-            # Add the event node with position attributes for layout
             dot.node(
                 event_id,
                 full_label,
@@ -133,10 +126,9 @@ def generate_timeline_diagram(json_input: str, output_format: str) -> str:
                 fontsize='12',
                 width='2.5',
                 height='1.2',
-                pos=f"{visual_col * 4.5},{-row * 3}!"  # Increased spacing for serpentine layout
+                pos=f"{visual_col * 4.5},{-row * 3}!"  
             )
         
-        # Connect events in chronological order (1→2→3→4...)
         for i in range(len(events) - 1):
             current_event_id = events[i].get('id', f'event_{i}')
             next_event_id = events[i + 1].get('id', f'event_{i + 1}')
@@ -149,9 +141,6 @@ def generate_timeline_diagram(json_input: str, output_format: str) -> str:
                 penwidth='2'
             )
         
-        # No need to connect title to events - it stays at the top independently
-        
-        # Set the layout engine to handle fixed positions
         dot.engine = 'neato'
 
         with NamedTemporaryFile(delete=False, suffix=f'.{output_format}') as tmp: