Update app.py

app.py

@@ -1236,9 +1236,64 @@ def create_genesis_animation(prediction_data, enable_animation=True):
                         intensities = [pt['intensity'] for pt in current_track]
                         categories = [pt['category'] for pt in current_track]
                         
-                        # (…rest of historical track, current position, uncertainty cone, genesis marker…)
-                        # identical to before, omitted for brevity
+                        # Get colors based on intensity (Saffir-Simpson-like scale)
+                        colors = []
+                        for intensity in intensities:
+                            if intensity < 34:
+                                colors.append('#80808080')  # Gray - TD
+                            elif intensity < 64:
+                                colors.append('#0000FF80')  # Blue - TS  
+                            elif intensity < 83:
+                                colors.append('#00FF0080')  # Green - Cat 1
+                            elif intensity < 96:
+                                colors.append('#FFFF0080')  # Yellow - Cat 2
+                            elif intensity < 113:
+                                colors.append('#FFA50080')  # Orange - Cat 3
+                            elif intensity < 137:
+                                colors.append('#FF000080')  # Red - Cat 4
+                            else:
+                                colors.append('#8B000080')  # DarkRed - Cat 5
                         
+                        # Historical track line
+                        frame_data.append(
+                            go.Scattergeo(
+                                lat=lats,
+                                lon=lons,
+                                mode='lines',
+                                line=dict(
+                                    width=2,
+                                    color='gray'
+                                ),
+                                name=f"{storm_id} Track",
+                                showlegend=(day_idx == 0),
+                                hoverinfo='skip'
+                            )
+                        )
+                        
+                        # Current position marker
+                        frame_data.append(
+                            go.Scattergeo(
+                                lat=[lats[-1]],
+                                lon=[lons[-1]],
+                                mode='markers',
+                                marker=dict(
+                                    size=10,
+                                    symbol='circle',
+                                    color=colors[-1]
+                                ),
+                                name=f"{storm_id} Position",
+                                showlegend=(day_idx == 0),
+                                hovertemplate=(
+                                    f"{storm_id}<br>"
+                                    f"Lat: {lats[-1]:.1f}°N<br>"
+                                    f"Lon: {lons[-1]:.1f}°E<br>"
+                                    f"Intensity: {intensities[-1]} kt<br>"
+                                    f"Category: {categories[-1]}<br>"
+                                    '<extra></extra>'
+                                )
+                            )
+                        )
+                
                 # Create frame
                 frames.append(go.Frame(
                     data=frame_data,
@@ -1266,8 +1321,50 @@ def create_genesis_animation(prediction_data, enable_animation=True):
             
             fig.frames = frames
             
-            # Set up initial frame, add professional play/pause controls, sliders, etc. (unchanged)
-            (…)
+            # Set up initial frame, add play/pause controls, sliders, etc.
+            slider_steps = [
+                dict(method="animate",
+                     args=[[frame.name], {
+                         "mode": "immediate",
+                         "frame": {"duration": 500, "redraw": True},
+                         "transition": {"duration": 0}
+                     }],
+                     label=frame.name)
+                for frame in frames
+            ]
+            
+            fig.update_layout(
+                updatemenus=[dict(
+                    type="buttons",
+                    showactive=False,
+                    y=0,
+                    x=1.05,
+                    xanchor="right",
+                    yanchor="top",
+                    pad=dict(t=0, r=10),
+                    buttons=[
+                        dict(label="▶️ Play",
+                             method="animate",
+                             args=[None, {
+                                 "frame": {"duration": 500, "redraw": True},
+                                 "fromcurrent": True,
+                                 "transition": {"duration": 0}
+                             }]),
+                        dict(label="⏸️ Pause",
+                             method="animate",
+                             args=[[None], {
+                                 "frame": {"duration": 0, "redraw": False},
+                                 "mode": "immediate",
+                                 "transition": {"duration": 0}
+                             }])
+                    ]
+                )],
+                sliders=[dict(
+                    active=0,
+                    pad=dict(t=50),
+                    steps=slider_steps
+                )]
+            )
         
         else:
             # STATIC VERSION (Final state overview)
@@ -1295,8 +1392,26 @@ def create_genesis_animation(prediction_data, enable_animation=True):
                 )
             )
             
-            # Add complete storm tracks (unchanged)
-            (…)
+            # Add complete storm tracks
+            for storm in storm_predictions:
+                track = storm['track']
+                if track:
+                    lats = [pt['lat'] for pt in track]
+                    lons = [pt['lon'] for pt in track]
+                    fig.add_trace(
+                        go.Scattergeo(
+                            lat=lats,
+                            lon=lons,
+                            mode='lines+markers',
+                            line=dict(width=2, color='gray'),
+                            marker=dict(
+                                size=6,
+                                symbol='circle',
+                                color='red'
+                            ),
+                            name=f"{storm['storm_id']} Full Track"
+                        )
+                    )
         
         # FINAL LAYOUT (Professional meteorological style)
         fig.update_layout(
@@ -1336,7 +1451,7 @@ def create_genesis_animation(prediction_data, enable_animation=True):
         )
         
         return fig
-
+        
     except Exception as e:
         logging.error(f"Error creating professional genesis animation: {e}")
         import traceback