Update app.py

Multilingual sentiment analysis for multiple sentences.
Output in graph

app.py

@@ -1,42 +1,72 @@
 import gradio as gr
+import matplotlib.pyplot as plt
 from transformers import pipeline
 
-# Laad het vertaalmodel (meertalig → Engels) en het sentimentanalyse-model (Engels)
+# Laad de modellen
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en")
-classifier = pipeline("sentiment-analysis")
+classifier = pipeline("sentiment-analysis", model="cardiffnlp/twitter-roberta-base-sentiment")
 
-# Functie om tekst te vertalen en sentimentanalyse uit te voeren
-def analyze_multilingual_sentiment(text):
+# Functie om meerdere zinnen te vertalen en analyseren
+def analyze_multilingual_sentences(text):
     if not text.strip():
-        return "⚠️ Voer een tekst in voor analyse."
+        return "⚠️ Voer een tekst in voor analyse.", None
 
-    # Stap 1: Vertaal de tekst naar Engels
-    translated_text = translator(text)[0]['translation_text']
+    # Splits de input op nieuwe regels (elke regel is een aparte zin)
+    sentences = [s.strip() for s in text.split("\n") if s.strip()]
+    
+    # Vertaal elke zin naar Engels
+    translated_sentences = [translator(sentence)[0]['translation_text'] for sentence in sentences]
 
-    # Stap 2: Voer sentimentanalyse uit op de vertaalde tekst
-    result = classifier(translated_text)[0]
+    # Voer sentimentanalyse uit op elke vertaalde zin
+    results = classifier(translated_sentences)
 
-    # Stap 3: Geef de output terug met oorspronkelijke tekst, vertaling en sentimentanalyse
-    output = f"""
-    🌍 **Originele tekst:** {text}
-    🔄 **Vertaling (Engels):** {translated_text}
-    📊 **Sentimentanalyse:** {result['label']} ({result['score']:.2f})
-    """
-    return output
+    # Tel het aantal positieve, negatieve en neutrale resultaten
+    positive_count = sum(1 for r in results if r['label'] == "LABEL_2")  # POSITIVE
+    negative_count = sum(1 for r in results if r['label'] == "LABEL_0")  # NEGATIVE
+    neutral_count = len(results) - (positive_count + negative_count)  # NEUTRAL
 
-# Maak een Gradio-interface
+    # Maak een overzichtelijke output
+    output = "**🌍 Meertalige Sentimentanalyse Resultaten:**\n\n"
+    for original, translated, result in zip(sentences, translated_sentences, results):
+        sentiment_label = result['label']
+        score = result['score']
+
+        # Bepaal sentiment en emoji
+        if sentiment_label == "LABEL_2":
+            sentiment = "POSITIVE"
+            emoji = "😊"
+        elif sentiment_label == "LABEL_0":
+            sentiment = "NEGATIVE"
+            emoji = "😡"
+        else:
+            sentiment = "NEUTRAL"
+            emoji = "🟦"
+
+        output += f"📌 **'{original}'**\n🔄 **Vertaling:** {translated}\n📊 **Sentiment:** {emoji} {sentiment} ({score:.2f})\n\n"
+
+    # Maak een grafiek met de sentimentverdeling
+    labels = ["Positief", "Neutraal", "Negatief"]
+    values = [positive_count, neutral_count, negative_count]
+    colors = ["#4CAF50", "#2196F3", "#F44336"]  # Groen, Blauw, Rood
+
+    fig, ax = plt.subplots()
+    ax.pie(values, labels=labels, autopct='%1.1f%%', startangle=90, colors=colors)
+    ax.axis("equal")  # Gelijke assen voor een ronde taartdiagram
+
+    return output, fig
+
+# Gradio-interface met tekstanalyse en een grafiek
 demo = gr.Interface(
-    fn=analyze_multilingual_sentiment, 
-    inputs=gr.Textbox(lines=3, placeholder="Voer hier een tekst in in elke taal..."), 
-    outputs="text",
-    title="🌍 Meertalige Sentimentanalyse",
-    description="Voer een tekst in in elke taal. De app vertaalt de tekst naar Engels en voert sentimentanalyse uit.",
+    fn=analyze_multilingual_sentences, 
+    inputs=gr.Textbox(lines=5, placeholder="Voer meerdere zinnen in, gescheiden door een nieuwe regel..."), 
+    outputs=["text", "plot"],
+    title="🌍 Meertalige Sentimentanalyse (Met Neutraal Sentiment en Grafiek)",
+    description="Voer meerdere zinnen in in elke taal. De app vertaalt elke zin naar Engels en voert sentimentanalyse uit. Het resultaat wordt weergegeven met een overzicht en een grafiek.",
     examples=[
-        ["Ik hou van programmeren!"],
-        ["Je suis très fatigué aujourd'hui."],
-        ["Hoy hace mucho frío y estoy triste."],
-        ["这部电影太棒了！"],
-        ["Das Wetter ist schrecklich, ich hasse es."]
+        ["Ik hou van programmeren!\nHet is vandaag een moeilijke dag.\nDeze film was oké."],
+        ["Je suis fatigué aujourd'hui.\nLe temps est magnifique!\nC'était une expérience moyenne."],
+        ["Hoy hace mucho frío.\nEstoy muy feliz con mi nuevo trabajo.\nNo estoy seguro de cómo me siento hoy."],
+        ["Das Essen war fantastisch!\nIch habe slecht geslapen.\nDer Film war ganz normal."]
     ]
 )
 
@@ -44,4 +74,3 @@ demo = gr.Interface(
 demo.launch()
 
 
-