Update app.py

Updated version

app.py

@@ -2,13 +2,13 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 from langdetect import detect
 
-# Multilingual model
+# Multilingual sentiment model
 MODEL = "nlptown/bert-base-multilingual-uncased-sentiment"
 tokenizer = AutoTokenizer.from_pretrained(MODEL)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL)
 sentiment_model = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
 
-# Emotion labels
+# Star emojis for sentiment
 STAR_EMOJIS = {
     1: "😡 Very Negative",
     2: "☹️ Negative",
@@ -17,79 +17,79 @@ STAR_EMOJIS = {
     5: "🤩 Very Positive"
 }
 
-# Predefined actions for multiple languages
+# Predefined "What to do" actions per language
 ACTIONS = {
     'en': {
-        1: "Take a break, reflect on the situation, or seek support.",
-        2: "Consider what’s bothering you and try to address it calmly.",
-        3: "Maintain balance; you’re feeling neutral, continue as usual.",
+        1: "Take a break, reflect, or seek support.",
+        2: "Consider what's bothering you and address it calmly.",
+        3: "Maintain balance; continue as usual.",
         4: "Share your positive experience and stay motivated!",
-        5: "Celebrate and spread your joy; keep up the enthusiasm!"
+        5: "Celebrate and spread your joy; keep up enthusiasm!"
     },
     'fr': {
-        1: "Faites une pause, réfléchissez à la situation ou demandez de l'aide.",
-        2: "Considérez ce qui vous dérange et essayez d'y remédier calmement.",
-        3: "Restez équilibré; vous vous sentez neutre, continuez comme d'habitude.",
+        1: "Faites une pause, réfléchissez ou demandez de l'aide.",
+        2: "Réfléchissez à ce qui vous dérange et agissez calmement.",
+        3: "Restez équilibré; continuez comme d'habitude.",
         4: "Partagez votre expérience positive et restez motivé !",
-        5: "Célébrez et partagez votre joie ; continuez avec enthousiasme !"
+        5: "Célébrez et partagez votre joie avec enthousiasme !"
     },
     'de': {
-        1: "Machen Sie eine Pause, reflektieren Sie die Situation oder suchen Sie Unterstützung.",
-        2: "Überlegen Sie, was Sie stört, und versuchen Sie, es ruhig zu lösen.",
-        3: "Halten Sie das Gleichgewicht; Sie fühlen sich neutral, machen Sie wie gewohnt weiter.",
+        1: "Machen Sie eine Pause, reflektieren Sie oder suchen Sie Unterstützung.",
+        2: "Überlegen Sie, was Sie stört, und lösen Sie es ruhig.",
+        3: "Halten Sie das Gleichgewicht; fahren Sie fort wie gewohnt.",
         4: "Teilen Sie Ihre positive Erfahrung und bleiben Sie motiviert!",
-        5: "Feiern Sie und verbreiten Sie Ihre Freude; bleiben Sie begeistert!"
+        5: "Feiern Sie und verbreiten Sie Ihre Freude enthusiastisch!"
     },
     'es': {
-        1: "Tómate un descanso, reflexiona sobre la situación o busca apoyo.",
-        2: "Considera lo que te molesta y trata de abordarlo con calma.",
-        3: "Mantén el equilibrio; te sientes neutral, continúa como de costumbre.",
+        1: "Tómate un descanso, reflexiona o busca apoyo.",
+        2: "Considera lo que te molesta y actúa con calma.",
+        3: "Mantén el equilibrio; continúa como de costumbre.",
         4: "Comparte tu experiencia positiva y mantente motivado!",
-        5: "Celebra y comparte tu alegría; continúa con entusiasmo!"
+        5: "Celebra y comparte tu alegría con entusiasmo!"
     },
     'it': {
-        1: "Fai una pausa, rifletti sulla situazione o cerca supporto.",
-        2: "Considera cosa ti disturba e cerca di affrontarlo con calma.",
-        3: "Mantieni l'equilibrio; ti senti neutrale, continua come al solito.",
+        1: "Fai una pausa, rifletti o cerca supporto.",
+        2: "Considera ciò che ti infastidisce e affrontalo con calma.",
+        3: "Mantieni l'equilibrio; continua come al solito.",
         4: "Condividi la tua esperienza positiva e rimani motivato!",
-        5: "Festeggia e diffondi la tua gioia; continua con entusiasmo!"
+        5: "Festeggia e diffondi la tua gioia con entusiasmo!"
     },
     'nl': {
-        1: "Neem een pauze, denk na over de situatie of zoek ondersteuning.",
-        2: "Overweeg wat je stoort en probeer het rustig aan te pakken.",
-        3: "Behoud je balans; je voelt je neutraal, ga door zoals gewoonlijk.",
+        1: "Neem een pauze, reflecteer of zoek ondersteuning.",
+        2: "Overweeg wat je stoort en handel rustig.",
+        3: "Behoud je balans; ga door zoals gewoonlijk.",
         4: "Deel je positieve ervaring en blijf gemotiveerd!",
-        5: "Vier en verspreid je vreugde; blijf enthousiast!"
+        5: "Vier en verspreid je vreugde enthousiast!"
     },
     'pt': {
-        1: "Faça uma pausa, reflita sobre a situação ou busque apoio.",
-        2: "Considere o que está incomodando e tente resolver calmamente.",
-        3: "Mantenha o equilíbrio; você se sente neutro, continue como de costume.",
+        1: "Faça uma pausa, reflita ou busque apoio.",
+        2: "Considere o que está incomodando e resolva calmamente.",
+        3: "Mantenha o equilíbrio; continue normalmente.",
         4: "Compartilhe sua experiência positiva e mantenha-se motivado!",
-        5: "Celebre e espalhe sua alegria; continue com entusiasmo!"
+        5: "Celebre e espalhe sua alegria com entusiasmo!"
     },
     'ru': {
         1: "Сделайте перерыв, обдумайте ситуацию или обратитесь за поддержкой.",
-        2: "Подумайте, что вас беспокоит, и попытайтесь спокойно решить проблему.",
-        3: "Сохраняйте баланс; вы чувствуете себя нейтрально, продолжайте как обычно.",
+        2: "Подумайте, что вас беспокоит, и решайте спокойно.",
+        3: "Сохраняйте баланс; продолжайте как обычно.",
         4: "Поделитесь положительным опытом и оставайтесь мотивированными!",
-        5: "Празднуйте и распространяйте радость; продолжайте с энтузиазмом!"
+        5: "Празднуйте и распространяйте радость с энтузиазмом!"
     },
     'ar': {
-        1: "خذ استراحة، تأمل في الوضع، أو اطلب الدعم.",
-        2: "فكر فيما يزعجك وحاول التعامل معه بهدوء.",
-        3: "حافظ على توازنك؛ أنت محايد، استمر كالمعتاد.",
+        1: "خذ استراحة، تأمل، أو اطلب الدعم.",
+        2: "فكر فيما يزعجك وتعامل معه بهدوء.",
+        3: "حافظ على توازنك؛ استمر كالمعتاد.",
         4: "شارك تجربتك الإيجابية وابقَ متحمسًا!",
-        5: "احتفل وانشر فرحك؛ استمر بحماسة!"
+        5: "احتفل وانشر فرحك بحماس!"
     },
     'ja': {
         1: "休憩を取り、状況を振り返るかサポートを求めてください。",
-        2: "何が不満なのかを考え、冷静に対処してください。",
-        3: "バランスを保ち、中立的な気持ちで通常通り続けてください。",
+        2: "何が不満か考え、冷静に対処してください。",
+        3: "バランスを保ち、通常通り続けてください。",
         4: "ポジティブな体験を共有し、モチベーションを維持してください！",
-        5: "祝って喜びを広め、熱意を持って続けましょう！"
+        5: "喜びを祝福し、熱意をもって広めましょう！"
     },
-    'yo': {  # Yoruba
+    'yo': {
         1: "Sinmi, ronú lórí ohun tó ṣẹlẹ̀, tàbí wa ìrànlọ́wọ́.",
         2: "Ronú nípa ohun tó ń dá ọ lẹ́rù, kí o sì gbìmọ̀ láti ṣe atunṣe.",
         3: "Dá a lára, o wà lórí ìṣàkóso, tẹ̀síwájú bí ó ṣe wà.",
@@ -99,15 +99,16 @@ ACTIONS = {
 }
 
 def analyze_sentiment(text):
-    # Sentiment
+    # Sentiment analysis
     result = sentiment_model(text)[0]
     stars = int(result["label"][0])
     sentiment = STAR_EMOJIS.get(stars, result["label"])
     confidence = f"{result['score']:.2f}"
 
-    # Detect language
+    # Language detection with Yoruba mapping
     try:
         lang = detect(text)
+        lang = 'yo' if lang == 'yo' else lang
         lang = lang if lang in ACTIONS else 'en'
     except:
         lang = 'en'
@@ -115,14 +116,13 @@ def analyze_sentiment(text):
     action = ACTIONS[lang].get(stars, "")
     return [[sentiment, confidence, action]]
 
-# Examples
+# Examples for testing
 examples = [
-    ["I absolutely love this new phone, the camera is stunning!"],       # English
-    ["Mo nifẹ́ fíìmù yìí gan-an!"],                                      # Yoruba Positive
-    ["Mo bínú gan-an sí ìṣẹ̀lẹ̀ náà."],                                   # Yoruba Negative
-    ["Je déteste quand cette application plante sans cesse."],           # French
-    ["Das Essen in diesem Restaurant war fantastisch!"],                 # German
-    ["Este producto es muy malo y no funciona."],                        # Spanish
+    ["I absolutely love this new phone!"],         # English
+    ["Mo nifẹ́ fíìmù yìí gan-an!"],                # Yoruba
+    ["Je déteste quand cette application plante."], # French
+    ["Das Essen in diesem Restaurant war fantastisch!"], # German
+    ["Este producto es muy malo."],               # Spanish
 ]
 
 # Gradio UI
@@ -137,9 +137,7 @@ demo = gr.Interface(
     examples=examples,
     title="🌍 Multilingual Emotion & Action Analyzer",
     description=(
-        "Supports 11+ languages including English, French, German, Spanish, Italian, Dutch, "
-        "Portuguese, Russian, Arabic, Japanese, Yoruba. Detects emotion (1–5 stars) and provides "
-        "suggested actions in the input language."
+        "Supports 11+ languages. Detects emotion (1–5 stars) and provides suggested actions in the input language."
     ),
 )