Update app/utils/predict_afteruse.py

app/utils/predict_afteruse.py

@@ -17,17 +17,18 @@ afteruse_labels = [
     "redness reducing", "skin texture", "soothing", "unknown", "whitening"
 ]
 
-afteruse_descriptions_en = {
-    "acne fighting": "helps fight acne",
-    "acne trigger": "may trigger acne",
-    "anti aging": "reduces signs of aging",
-    "brightening": "brightens the skin",
-    "moisturizing": "moisturizes the skin",
-    "redness reducing": "reduces redness",
-    "skin texture": "improves skin texture",
-    "soothing": "soothes the skin",
-    "unknown": "has unknown effects",
-    "whitening": "whitens the skin"
+# Deskripsi efek after use dalam bahasa Indonesia
+afteruse_descriptions_id = {
+    "acne fighting": "membantu melawan jerawat",
+    "acne trigger": "dapat memicu jerawat",
+    "anti aging": "mengurangi tanda-tanda penuaan",
+    "brightening": "mencerahkan kulit",
+    "moisturizing": "melembapkan kulit",
+    "redness reducing": "mengurangi kemerahan pada kulit",
+    "skin texture": "memperbaiki tekstur kulit",
+    "soothing": "menenangkan kulit",
+    "unknown": "memiliki efek yang belum diketahui",
+    "whitening": "memutihkan kulit"
 }
 
 def predict_after_use(input_ingredients):
@@ -60,19 +61,18 @@ def predict_after_use(input_ingredients):
 
     return predicted_labels
 
-def generate_afteruse_sentence_en(predicted_labels):
+def generate_afteruse_sentence_id(predicted_labels):
     if not predicted_labels:
-        return "No effects were detected based on the provided ingredients."
+        return "Tidak ada efek yang terdeteksi berdasarkan bahan yang diberikan."
 
-    descriptions = [afteruse_descriptions_en.get(label, label) for label in predicted_labels]
+    descriptions = [afteruse_descriptions_id.get(label, label) for label in predicted_labels]
 
     if len(descriptions) == 1:
-        return f"This product {descriptions[0]}."
+        return f"Produk ini {descriptions[0]}."
     elif len(descriptions) == 2:
-        return f"This product {descriptions[0]} and {descriptions[1]}."
+        return f"Produk ini {descriptions[0]} dan {descriptions[1]}."
     else:
-        return f"This product {', '.join(descriptions[:-1])}, and {descriptions[-1]}."
-    
+        return f"Produk ini {', '.join(descriptions[:-1])}, dan {descriptions[-1]}."
 
 def predict_after_use_with_probs(input_ingredients):
     if not input_ingredients: