Update app.py

app.py

@@ -1,17 +1,14 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 from langdetect import detect
-from googletrans import Translator
 
-# Multilingual sentiment model
+# Multilingual 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)
 
-translator = Translator()
-
-# Map stars (1–5) to emotion labels with emojis
+# Emotion labels
 STAR_EMOJIS = {
     1: "😡 Very Negative",
     2: "☹️ Negative",
@@ -20,17 +17,89 @@ STAR_EMOJIS = {
     5: "🤩 Very Positive"
 }
 
-# Suggested actions in English
+# Predefined actions for multiple languages
 ACTIONS = {
-    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.",
-    4: "Share your positive experience and stay motivated!",
-    5: "Celebrate and spread your joy; keep up the enthusiasm!"
+    '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.",
+        4: "Share your positive experience and stay motivated!",
+        5: "Celebrate and spread your joy; keep up the 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.",
+        4: "Partagez votre expérience positive et restez motivé !",
+        5: "Célébrez et partagez votre joie ; continuez 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.",
+        4: "Teilen Sie Ihre positive Erfahrung und bleiben Sie motiviert!",
+        5: "Feiern Sie und verbreiten Sie Ihre Freude; bleiben Sie begeistert!"
+    },
+    '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.",
+        4: "Comparte tu experiencia positiva y mantente motivado!",
+        5: "Celebra y comparte tu alegría; continú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.",
+        4: "Condividi la tua esperienza positiva e rimani motivato!",
+        5: "Festeggia e diffondi la tua gioia; continua 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.",
+        4: "Deel je positieve ervaring en blijf gemotiveerd!",
+        5: "Vier en verspreid je vreugde; blijf 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.",
+        4: "Compartilhe sua experiência positiva e mantenha-se motivado!",
+        5: "Celebre e espalhe sua alegria; continue com entusiasmo!"
+    },
+    'ru': {
+        1: "Сделайте перерыв, обдумайте ситуацию или обратитесь за поддержкой.",
+        2: "Подумайте, что вас беспокоит, и попытайтесь спокойно решить проблему.",
+        3: "Сохраняйте баланс; вы чувствуете себя нейтрально, продолжайте как обычно.",
+        4: "Поделитесь положительным опытом и оставайтесь мотивированными!",
+        5: "Празднуйте и распространяйте радость; продолжайте с энтузиазмом!"
+    },
+    'ar': {
+        1: "خذ استراحة، تأمل في الوضع، أو اطلب الدعم.",
+        2: "فكر فيما يزعجك وحاول التعامل معه بهدوء.",
+        3: "حافظ على توازنك؛ أنت محايد، استمر كالمعتاد.",
+        4: "شارك تجربتك الإيجا��ية وابقَ متحمسًا!",
+        5: "احتفل وانشر فرحك؛ استمر بحماسة!"
+    },
+    'ja': {
+        1: "休憩を取り、状況を振り返るかサポートを求めてください。",
+        2: "何が不満なのかを考え、冷静に対処してください。",
+        3: "バランスを保ち、中立的な気持ちで通常通り続けてください。",
+        4: "ポジティブな体験を共有し、モチベーションを維持してください！",
+        5: "祝って喜びを広め、熱意を持って続けましょう！"
+    },
+    'yo': {  # Yoruba
+        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à.",
+        4: "Pín ìrírí rẹ tó dáa kí o sì máa ní ìmọ̀lára rere!",
+        5: "Ṣe ayẹyẹ, tàn ìdùnnú rẹ kaakiri, tẹ̀síwájú pẹ̀lú ìfẹ́!"
+    }
 }
 
 def analyze_sentiment(text):
-    # Sentiment analysis
+    # Sentiment
     result = sentiment_model(text)[0]
     stars = int(result["label"][0])
     sentiment = STAR_EMOJIS.get(stars, result["label"])
@@ -39,27 +108,21 @@ def analyze_sentiment(text):
     # Detect language
     try:
         lang = detect(text)
+        lang = lang if lang in ACTIONS else 'en'
     except:
-        lang = "en"
-
-    # Translate action to detected language
-    action_en = ACTIONS.get(stars, "")
-    if lang != "en":
-        try:
-            action_translated = translator.translate(action_en, dest=lang).text
-        except:
-            action_translated = action_en
-    else:
-        action_translated = action_en
+        lang = 'en'
 
-    return [[sentiment, confidence, action_translated]]
+    action = ACTIONS[lang].get(stars, "")
+    return [[sentiment, confidence, action]]
 
-# Example texts including Yoruba
+# Examples
 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
 ]
 
 # Gradio UI
@@ -74,8 +137,9 @@ demo = gr.Interface(
     examples=examples,
     title="🌍 Multilingual Emotion & Action Analyzer",
     description=(
-        "Supports multiple languages including English, Yoruba, French, German, Spanish, etc. "
-        "Detects emotion (1–5 stars) and provides suggested actions in the same language as input."
+        "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."
     ),
 )