Update app.py

app.py

@@ -36,21 +36,29 @@ def analyze_text(icelandic_text):
     formality_label, formality_score = analyze_formality(icelandic_text)
     sentiment_label, sentiment_score = analyze_sentiment(icelandic_text)
 
+    # Convert sentiment label
+    sentiment_label = sentiment_label.replace("LABEL_", "")
+
     translated_text = translate_text(icelandic_text)
 
     toxicity_results = analyze_toxicity(translated_text)
     if isinstance(toxicity_results, list):
         toxicity_results = toxicity_results[0]
+    
+    # Determine toxicity label based on score
+    toxicity_label = '1' if toxicity_results['score'] >= 0.5 else '0'
 
     politeness_label, politeness_score = analyze_politeness(translated_text)
+    # Convert politeness label to binary
+    politeness_label = '1' if politeness_label.lower() == 'polite' else '0'
 
-    analysis_results = f"""
-    Translated Text: {translated_text}
-    Sentiment: Label: {sentiment_label}, Score: {round(sentiment_score, 2)}
-    Formality: Label: {formality_label}, Score: {round(formality_score, 2)}
-    Toxicity: Score: {round(toxicity_results['score'], 2)}
-    Politeness: Label: {politeness_label}, Score: {round(politeness_score, 2)}
-    """
+    analysis_results = (
+        f"Translated Text: {translated_text}\n\n"
+        f"Sentiment: Label: {sentiment_label}, Score: {round(sentiment_score, 2)}\n"
+        f"Formality: Label: {round(formality_score, 2)}, Score: {round(formality_score, 2)}\n"
+        f"Toxicity: Label: {toxicity_label}, Score: {round(toxicity_results['score'], 2)}\n"
+        f"Politeness: Label: {politeness_label}, Score: {round(politeness_score, 2)}"
+    )
     return analysis_results
 
 demo = gr.Interface(fn=analyze_text,