fix dv

app.py

@@ -248,16 +248,22 @@ if predict_button:
         st.markdown("---")
         st.markdown("### 📊 Drug Recommendations Visualization")
         df_drugs = pd.DataFrame({"Drug": top_drugs, "Rank": range(len(top_drugs), 0, -1)})  # Reverse ranks
+        color_map = {}
+        if len(top_drugs) >= 1:
+            color_map[top_drugs[0]] = "#b4befe"  # Lavender
+        if len(top_drugs) >= 2:
+            color_map[top_drugs[1]] = "#a6e3a1"  # Green
+        if len(top_drugs) >= 3:
+            color_map[top_drugs[2]] = "#f38ba8"  # Red
+        if len(top_drugs) >= 4:
+            color_map[top_drugs[3]] = "#f2cdcd"  # Flamingo
+            
         fig = px.bar(df_drugs, x="Rank", y="Drug", title="Top Recommended Drugs", 
                      labels={"Drug": "Drug Name", "Rank": "Rank"},
                      orientation='h',  # Horizontal bar chart
                      color="Drug",  # Color by drug name
-                     color_discrete_map={
-                         top_drugs[0]: "#b4befe",  # Lavender
-                         top_drugs[1]: "#a6e3a1",  # Green
-                         top_drugs[2]: "#f38ba8",  # Red
-                         top_drugs[3]: "#f2cdcd"   # Flamingo
-                     })  # Map colors to drugs
+                     color_discrete_map=color_map)  # Map colors to drugs
+        
         fig.update_layout(yaxis={'categoryorder':'total ascending'})  # Sort by rank
         st.plotly_chart(fig, use_container_width=True)