Update app.py

app.py

@@ -23,42 +23,20 @@ label_encoder = joblib.load(hf_hub_download(
 ))
 
 print("Loading Sinhala model...")
-sinhala_pipe = pipeline(
-    "text-classification",
-    model="E-motionAssistant/Sinhala_Text_Emotion_Model_Retrained"
-)
+sinhala_pipe = pipeline("text-classification",
+                        model="E-motionAssistant/Sinhala_Text_Emotion_Model_Retrained")
 
 print("Loading Tamil model...")
-tamil_pipe = pipeline(
-    "text-classification",
-    model="E-motionAssistant/Tamil_Emotion_Recognition_Model"
-)
+tamil_pipe = pipeline("text-classification",
+                      model="E-motionAssistant/Tamil_Emotion_Recognition_Model")
 
 print("All models loaded successfully!")
 
-# Sinhala label mapping (replace these names with your actual classes)
-# You need to fill this list with the correct Sinhala emotion names in order
-SINHALA_LABELS = [
-    "neutral",       # LABEL_0
-    "සතුට",         # LABEL_1
-    "දුක",           # LABEL_2
-    "කෝපය",         # LABEL_3
-    "බය",            # LABEL_4
-    "ආදරය",         # LABEL_5
-    "අප්‍රසාදය",     # LABEL_6
-    "අපහාසය",       # LABEL_7
-    "කලකිරීම",      # LABEL_8
-    "අනුකම්පාව",    # LABEL_9
-    "අපේක්ෂාව",     # LABEL_10
-    "සැහැල්ලුව",     # LABEL_11
-    "වෙනත්"          # LABEL_12  ← change this if LABEL_12 is something else
-]
-
 @app.route("/", methods=["GET", "POST"])
 def home():
     english_result = ""
     sinhala_result = ""
-    tamil_result = ""
+    tamil_result   = ""
 
     if request.method == "POST":
         action = request.form.get("action")
@@ -77,32 +55,17 @@ def home():
             text = request.form.get("sinhala_text", "").strip()
             if text:
                 res = sinhala_pipe(text)[0]
-                label = res['label']
-                score = res['score']
-
-                # Convert LABEL_XX → real Sinhala emotion name
-                if label.startswith("LABEL_"):
-                    try:
-                        num = int(label.replace("LABEL_", ""))
-                        if 0 <= num < len(SINHALA_LABELS):
-                            label = SINHALA_LABELS[num]
-                        else:
-                            label = f"නොදන්නා ({num})"
-                    except:
-                        pass  # keep original if conversion fails
-
-                sinhala_result = f"හැඟීම: {label} ({score:.3f})"
+                sinhala_result = f"Emotion: {res['label']})"
             else:
-                sinhala_result = "කරුණාකර යමක් ලියන්න"
+                sinhala_result = "Please write something"
 
         elif action == "predict_tamil":
             text = request.form.get("tamil_text", "").strip()
             if text:
                 res = tamil_pipe(text)[0]
-                # Tamil remains unchanged (as per your request)
-                tamil_result = f"உணர்வு: {res['label']} ({res['score']:.3f})"
+                tamil_result = f"Emotion: {res['label']})"
             else:
-                tamil_result = "கருணையுடன் உரையை உள்ளிடவும்"
+                tamil_result = "Please write something"
 
     return render_template(
         "index.html",
@@ -111,6 +74,5 @@ def home():
         tamil_result=tamil_result
     )
 
-
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=False)