Update huggingfaceModel.py

huggingfaceModel.py

@@ -1,7 +1,7 @@
 import os
 from huggingface_hub import InferenceClient
 
-def generate_dropout_insights(input_values, risk_factors):
+def generate_dropout_insights(input_values, risk_factors, label):
     """
     Generates structured dropout risk insights using AI, ensuring Unicode-safe text for PDFs.
     """
@@ -20,29 +20,29 @@ def generate_dropout_insights(input_values, risk_factors):
         {
             "role": "user",
             "content": f"""
-            Generate a **detailed dropout risk analysis** report based on the following data:
+            Generate a detailed dropout risk analysis report based on the following data:
             
             {formatted_input}
             
             Format the response as:
             
-            ## **Analysis of Student Dropout Risk**
+            ## Analysis of Student Dropout Risk
             
-            **Overall Risk Level: {{risk_level}}**
+            Overall Risk Level: {label}
             
-            ### **🟢 Strengths:**
+            ### 🟢 Strengths:
             List areas where the student is performing well:
             {{strengths}}
             
-            ### **🔴 Concerns:**
+            ### 🔴 Concerns:
             Identify factors that increase dropout risk:
             {{concerns}}
             
-            ### **📌 Action Plan for Improvement:**
+            ### 📌 Action Plan for Improvement:
             Provide targeted **solutions** based on the student's **weak areas**:
             {{recommendations}}
             
-            ## **📝 Final Recommendation:**
+            ## 📝 Final Recommendation:
             {{overall_recommendation}}
             """
         }