Update app.py

app.py

@@ -72,17 +72,23 @@ def replace_encoding(tokens):
                  .replace('Ãļ', 'ý')
             for token in tokens[1:-1]]
 
-def predict(text):
+def predict(user_input_text):
+    explanations_formality = bench.explain(user_input_text, target=0)
+    explanations_sentiment = bench.explain(user_input_text, target=1)
+    explanations_politeness = bench.explain(user_input_text, target=2)
+    explanations_toxicity = bench.explain(user_input_text, target=3)
+
+    prediction_output, keywords, influential_keywords = get_prediction(user_input_text)
+
     greeting_pattern = r"^(Halló|Hæ|Sæl|Góðan dag|Kær kveðja|Daginn|Kvöldið|Ágætis|Elsku)"
     
-    prediction_output, keywords, influential_keywords = get_prediction(text)
     greeting_feedback = ""
     
-    modified_input = text
+    modified_input = user_input_text
     for keyword, _ in keywords:
         modified_input = modified_input.replace(keyword, f"<span style='color:green;'>{keyword}</span>")
     
-    if not re.match(greeting_pattern, text, re.IGNORECASE):
+    if not re.match(greeting_pattern, user_input_text, re.IGNORECASE):
         greeting_feedback = "OTHER FEEDBACK:<br>Heilsaðu dóninn þinn<br>"
 
     response = f"INPUT:<br>{modified_input}<br><br>MY PREDICTION:<br>{prediction_output}<br>{influential_keywords}<br>{greeting_feedback}"