Update Backend.py

Backend.py

@@ -155,7 +155,7 @@ def generate_plot(dropout_risk):
     ax.set_title('Dropout Risk Prediction')
 
     plt.savefig("dropout_plot.png")  # Save before using in PDF
-    return fig
+    return fig ,labels[index]
 
 def remove_emojis(text):
     """Removes all emojis and unsupported Unicode characters from text."""
@@ -167,10 +167,9 @@ def process_and_generate_report(**input_values):
     """Processes input values and generates risk prediction along with a report."""
     
     dropout_risk,risk_factor = predict_dropout(**input_values)
-    insights = generate_dropout_insights(input_values,risk_factor)
+    fig, label = generate_plot(dropout_risk)
+    insights = generate_dropout_insights(input_values,risk_factor,label)
     cleaned_insights = remove_emojis(insights)
-    fig = generate_plot(dropout_risk)
     pdf_filename = generate_pdf(input_values, dropout_risk, cleaned_insights)
     
-
     return fig, insights, pdf_filename