Update app.py

app.py

@@ -786,6 +786,7 @@ with tabs[4]:
                 for rec in analysis_data['recommendations']:
                     st.write(f"• {rec}")
 
+# Tab 6: OPTIMIZED Future Risk Predictor
 # Tab 6: OPTIMIZED Future Risk Predictor
 # Tab 6: OPTIMIZED Future Risk Predictor
 with tabs[5]:
@@ -814,21 +815,28 @@ with tabs[5]:
                 st.success(f"Risk analysis completed in {end_time - start_time:.1f} seconds")
                 st.info(summary)
                 
-                # Yearly risk metrics - FIXED: Remove delta_color parameter
+                # Yearly risk metrics
                 st.subheader("Risk Timeline")
                 cols = st.columns(5)
                 for i, year_data in enumerate(timeline):
                     with cols[i]:
-                        # Simple metric without delta_color
+                        # Determine badge color based on risk level
+                        if year_data["risk_level"] == "High":
+                            badge_color = "🔴"
+                        elif year_data["risk_level"] == "Medium":
+                            badge_color = "🟡"
+                        else:
+                            badge_color = "🟢"
+                            
                         st.metric(
-                            f"Year {year_data['year']}", 
+                            f"Year {year_data['year']} {badge_color}", 
                             f"{year_data['risk_score_0_100']}%",
                             year_data["risk_level"]
                         )
                 
-                # Detailed view
+                # Detailed view - FIXED: Remove 'key' parameter from expander
                 for year_data in timeline:
-                    with st.expander(f"Year {year_data['year']} Details", key=f"year_{year_data['year']}"):
+                    with st.expander(f"Year {year_data['year']} Details"):
                         col1, col2 = st.columns(2)
                         with col1:
                             st.write("**Risks:**")