Update app.py

app.py

@@ -310,7 +310,8 @@ def create_dual_bar_chart(tasks, bar_width, task_colors, common_x_range, font_pr
 
     # Customize legend with selected position
     ax.legend(fontsize=font_properties['legend_size'], loc=legend_loc_map[legend_position])
-
+    # Add just before returning the figure in this function
+    ax.legend(loc=legend_loc_map[legend_position], bbox_to_anchor=legend_bbox_map[legend_position])
     plt.tight_layout()
     return fig
 
@@ -371,6 +372,9 @@ def plot_timeline_chart(tasks, bar_width, font_properties, progress_text_positio
     # Add legend with selected position
     if show_milestones or show_today_line:
         ax.legend(loc=legend_loc_map[legend_position], fontsize=font_properties['legend_size'], frameon=True)
+
+    # Add just before returning the figure in this function
+    ax.legend(loc=legend_loc_map[legend_position], bbox_to_anchor=legend_bbox_map[legend_position])
     
     plt.tight_layout()
     return fig
@@ -512,9 +516,13 @@ def create_dual_donut_chart(tasks, font_properties,
         ax.axis('equal')  # Keep the pie chart circular
         ax.text(0, 0, "Project Progress", fontsize=font_properties['title_size'], 
                fontname='Times New Roman', ha='center')
+    # Add this right before the return fig statement
+    ax.legend(loc=legend_loc_map[legend_position], bbox_to_anchor=legend_bbox_map[legend_position])
     
     plt.tight_layout()
     return fig
+
+    
 # Function to create an interactive gantt chart with plotly
 def create_gantt_chart(tasks):
     df = []