Update app.py

app.py

@@ -8,35 +8,60 @@ formality_classifier = pipeline("text-classification", model="svanhvit/formality
 detoxify_pipeline = pipeline('text-classification', model='unitary/toxic-bert', tokenizer='bert-base-uncased', function_to_apply='sigmoid', return_all_scores=True)
 politeness_classifier = pipeline("text-classification", model="Genius1237/xlm-roberta-large-tydip")
 
-def analyze_icelandic_text(icelandic_text):
-    analysis_result = analyze_text(icelandic_text)
-
-    sentiment_label = "Positive" if analysis_result["sentiment"]["label"] == "LABEL_1" else "Negative"
-    formality_label = "Formal" if analysis_result["formality"]["label"] == "formal" else "Informal"
-    toxic_label = "Toxic" if analysis_result["toxicity"]["score"] > 0.5 else "Not Toxic"
-    politeness_label = "Polite" if analysis_result["politeness"]["label"] == "polite" else "Impolite"
-    
-    return {
-        "Translated Text": analysis_result["translated_text"],
-        "Sentiment": f"{sentiment_label} (Score: {round(analysis_result['sentiment']['score'], 2)})",
-        "Formality": f"{formality_label} (Score: {round(analysis_result['formality']['score'], 2)})",
-        "Toxicity": f"{toxic_label} (Score: {round(analysis_result['toxicity']['score'], 2)})",
-        "Politeness": f"{politeness_label} (Score: {round(analysis_result['politeness']['score'], 2)})"
+def translate_text(text):
+    translation = translator(text, max_length=512)
+    return translation[0]['translation_text']
+
+def analyze_toxicity(text):
+    toxicity_results = detoxify_pipeline(text)
+    return toxicity_results[0]
+
+def analyze_politeness(text):
+    politeness_result = politeness_classifier(text)
+    return politeness_result[0]['label'], politeness_result[0]['score']
+
+def analyze_formality(text):
+    formality_result = formality_classifier(text)
+    formality_label = formality_result[0]['label']
+    formality_score = formality_result[0]['score']
+    return formality_label, formality_score
+
+def analyze_sentiment(text):
+    sentiment_result = sentiment_classifier(text)
+    sentiment_label = sentiment_result[0]['label']
+    sentiment_score = sentiment_result[0]['score']
+    return sentiment_label, sentiment_score
+
+def analyze_text(icelandic_text):
+    formality_label, formality_score = analyze_formality(icelandic_text)
+    sentiment_label, sentiment_score = analyze_sentiment(icelandic_text)
+
+    translated_text = translate_text(icelandic_text)
+
+    toxicity_results = analyze_toxicity(translated_text)
+    if isinstance(toxicity_results, list):
+        toxicity_results = toxicity_results[0]
+
+    politeness_label, politeness_score = analyze_politeness(translated_text)
+
+    analysis_results = {
+        "Translated Text": translated_text,
+        "Sentiment": f"Label: {sentiment_label}, Score: {round(sentiment_score, 2)}",
+        "Formality": f"Label: {formality_label}, Score: {round(formality_score, 2)}",
+        "Toxicity": f"Score: {round(toxicity_results['score'], 2)}",
+        "Politeness": f"Label: {politeness_label}, Score: {round(politeness_score, 2)}"
     }
+    return analysis_results
 
-iface = gr.Interface(
-    fn=analyze_icelandic_text,
-    inputs=gr.inputs.Textbox(lines=2, placeholder="Enter an Icelandic sentence here..."),
-    outputs=[
-        gr.outputs.Textbox(label="Translated Text"),
-        gr.outputs.Textbox(label="Sentiment"),
-        gr.outputs.Textbox(label="Formality"),
-        gr.outputs.Textbox(label="Toxicity"),
-        gr.outputs.Textbox(label="Politeness"),
-    ],
-    title="Icelandic Text Analysis",
-    description="This app translates Icelandic text to English and analyzes its sentiment, formality, toxicity, and politeness."
-)
+demo = gr.Interface(fn=analyze_text, 
+                     inputs="text", 
+                     outputs=[gr.outputs.Textbox(label="Translated Text"),
+                              gr.outputs.Textbox(label="Sentiment"),
+                              gr.outputs.Textbox(label="Formality"),
+                              gr.outputs.Textbox(label="Toxicity"),
+                              gr.outputs.Textbox(label="Politeness")],
+                     title="Icelandic Text Analysis",
+                     description="This app translates Icelandic text to English and performs sentiment, formality, toxicity, and politeness analysis.")
 
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()