Update app.py

app.py

@@ -4,66 +4,46 @@ from torch.nn.functional import sigmoid
 import torch
 from PIL import Image
 
-# Load text emotion model
 model_name = "SamLowe/roberta-base-go_emotions"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
-# Load image emotion model
 image_model_name = "Celal11/resnet-50-finetuned-FER2013-0.001"
 image_processor = AutoImageProcessor.from_pretrained(image_model_name)
 image_model = AutoModelForImageClassification.from_pretrained(image_model_name)
 
-# Islamic advice mapping based on emotion
 islamic_advice = {
-    "sadness": "<div class='notion-card'><h3>\u201cDo not grieve; indeed Allah is with us.\u201d</h3><p><strong>Arabic:</strong> <span style='font-size: 1.2em;'>لَا تَحْزَنْ إِنَّ اللَّهَ مَعَنَا</span><br><em>Qur'an, At-Tawbah 9:40</em></p></div>",
-    "joy": "<div class='notion-card'><h3>\u201cIf you are grateful, I will surely increase your favor.\u201d</h3><p><strong>Arabic:</strong> <span style='font-size: 1.2em;'>لَئِن شَكَرْتُمْ لَأَزِيدَنَّكُمْ</span><br><em>Qur'an, Ibrahim 14:7</em></p></div>",
-    "fear": "<div class='notion-card'><h3>\u201cSufficient for us is Allah, and [He is] the best Disposer of affairs.\u201d</h3><p><strong>Arabic:</strong> <span style='font-size: 1.2em;'>حَسْبُنَا اللَّهُ وَنِعْمَ الْوَكِيلُ</span><br><em>Qur'an, Al Imran 3:173</em></p></div>",
-    "anger": "<div class='notion-card'><h3>\u201cThe strong is not the one who overcomes people by his strength, but the strong is the one who controls himself while in anger.\u201d</h3><p><em>Hadith, Sahih al-Bukhari 6114</em></p></div>",
-    "confusion": "<div class='notion-card'><h3>\u201cSeek help through patience and prayer.\u201d</h3><p><strong>Arabic:</strong> <span style='font-size: 1.2em;'>وَاسْتَعِينُوا بِالصَّبْرِ وَالصَّلَاةِ</span><br><em>Qur'an, Al-Baqarah 2:45</em></p></div>",
-    "love": "<div class='notion-card'><h3>\u201cIndeed, those who have believed and done righteous deeds - the Most Merciful will appoint for them affection.\u201d</h3><p><strong>Arabic:</strong> <span style='font-size: 1.2em;'>سَيَجْعَلُ لَهُمُ الرَّحْمَٰنُ وُدًّا</span><br><em>Qur'an, Maryam 19:96</em></p></div>",
-    "neutral": "<div class='notion-card'><h3>May Allah always guide your heart in every situation.</h3></div>"
+    "sadness": "<h1 class='emoji'>😢</h1><div class='notion-card'><h3>Sadness</h3><p><strong>“Do not grieve; indeed Allah is with us.”</strong><br><span class='arabic'>لَا تَحْزَنْ إِنَّ اللَّهَ مَعَنَا</span><br><em>Qur'an, At-Tawbah 9:40</em></p></div>",
+    "joy": "<h1 class='emoji'>😃</h1><div class='notion-card'><h3>Joy</h3><p><strong>“If you are grateful, I will surely increase your favor.”</strong><br><span class='arabic'>لَئِن شَكَرْتُمْ لَأَزِيدَنَّكُمْ</span><br><em>Qur'an, Ibrahim 14:7</em></p></div>",
+    "fear": "<h1 class='emoji'>😱</h1><div class='notion-card'><h3>Fear</h3><p><strong>“Sufficient for us is Allah, and [He is] the best Disposer of affairs.”</strong><br><span class='arabic'>حَسْبُنَا اللَّهُ وَنِعْمَ الْوَكِيلُ</span><br><em>Qur'an, Al Imran 3:173</em></p></div>",
+    "anger": "<h1 class='emoji'>😡</h1><div class='notion-card'><h3>Anger</h3><p><strong>“The strong is not the one who overcomes people by his strength, but the strong is the one who controls himself while in anger.”</strong><br><em>Hadith, Sahih al-Bukhari 6114</em></p></div>",
+    "confusion": "<h1 class='emoji'>🤔</h1><div class='notion-card'><h3>Confusion</h3><p><strong>“Seek help through patience and prayer.”</strong><br><span class='arabic'>وَاسْتَعِينُوا بِالصَّبْرِ وَالصَّلَاةِ</span><br><em>Qur'an, Al-Baqarah 2:45</em></p></div>",
+    "love": "<h1 class='emoji'>❤️</h1><div class='notion-card'><h3>Love</h3><p><strong>“Indeed, those who have believed and done righteous deeds - the Most Merciful will appoint for them affection.”</strong><br><span class='arabic'>سَيَجْعَلُ لَهُمُ الرَّحْمَٰنُ وُدًّا</span><br><em>Qur'an, Maryam 19:96</em></p></div>",
+    "neutral": "<h1 class='emoji'>😐</h1><div class='notion-card'><h3>Neutral</h3><p><strong>May Allah always guide your heart in every situation.</strong></p></div>"
 }
 
-# Analyze image emotion using processor and model
-def analyze_image_emotion(image):
-    if image is None:
-        return "No image provided.", ""
-    inputs = image_processor(images=image, return_tensors="pt")
-    with torch.no_grad():
-        logits = image_model(**inputs).logits
-    probs = torch.nn.functional.softmax(logits, dim=1)[0]
-    pred_idx = torch.argmax(probs).item()
-    label = image_model.config.id2label[pred_idx].lower()
-    score = probs[pred_idx].item()
-    islamic = islamic_advice.get(label, islamic_advice["neutral"])
-    return f"{label.capitalize()} ({score:.2f})", islamic
-
-# Analyze text emotion
-def get_emotions(text, threshold):
+def analyze_combined(text, threshold, image):
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
     with torch.no_grad():
-        logits = model(**inputs).logits
-    probs = sigmoid(logits)[0]
-
-    top_idx = torch.argmax(probs).item()
-    top_label = model.config.id2label[top_idx].lower()
-    top_score = probs[top_idx].item()
-    islamic = islamic_advice.get(top_label, islamic_advice["neutral"])
+        logits_text = model(**inputs).logits
+    probs_text = sigmoid(logits_text)[0]
+    text_idx = torch.argmax(probs_text).item()
+    text_label = model.config.id2label[text_idx].lower()
 
-    labels = [model.config.id2label[i] for i, p in enumerate(probs) if p > threshold]
-    icons = [f"{label.capitalize()} ({probs[i]:.2f})" for i, label in enumerate(labels)]
+    if image:
+        inputs_image = image_processor(images=image, return_tensors="pt")
+        with torch.no_grad():
+            logits_img = image_model(**inputs_image).logits
+        probs_img = torch.nn.functional.softmax(logits_img, dim=1)[0]
+        img_idx = torch.argmax(probs_img).item()
+        img_label = image_model.config.id2label[img_idx].lower()
+    else:
+        img_label = None
 
-    result = ", ".join(icons) if icons else "No strong emotion detected."
-    return result, islamic
+    final_label = text_label if img_label is None else img_label
+    card = islamic_advice.get(final_label, islamic_advice["neutral"])
+    return card
 
-# Combined analysis
-def analyze_combined(text, threshold, image):
-    text_result, text_tip = get_emotions(text, threshold)
-    image_result, image_tip = analyze_image_emotion(image)
-    return text_result, text_tip, image_result, image_tip
-
-# Gradio UI
 custom_css = """
 body {
   background: #f9fafb;
@@ -81,16 +61,6 @@ body {
 .gr-button:hover {
   background-color: #2563eb !important;
 }
-.gr-textbox, .output-textbox {
-  border-radius: 12px !important;
-  border: 1px solid #d1d5db !important;
-  padding: 12px !important;
-  font-size: 1rem;
-}
-.output-textbox {
-  background-color: #ffffff;
-  border: 1px solid #60a5fa;
-}
 .notion-card {
   background: #ffffff;
   border-radius: 12px;
@@ -98,18 +68,37 @@ body {
   padding: 16px;
   margin-top: 12px;
   box-shadow: 0 2px 6px rgba(0,0,0,0.04);
+  max-width: 600px;
+  margin-left: auto;
+  margin-right: auto;
 }
 .notion-card h3 {
   margin-top: 0;
   color: #111827;
-  font-size: 1.1rem;
-  margin-bottom: 6px;
+  font-size: 1.25rem;
+  margin-bottom: 8px;
+  font-weight: 600;
 }
 .notion-card p {
-  font-size: 0.95rem;
+  font-size: 1rem;
   color: #374151;
   margin: 0;
 }
+.emoji {
+  text-align: center;
+  font-size: 3rem;
+  margin: 0;
+  padding: 0;
+}
+.arabic {
+  font-size: 1.3rem;
+  direction: rtl;
+  display: block;
+  margin-top: 4px;
+  margin-bottom: 4px;
+  text-align: right;
+  font-family: 'Scheherazade', serif;
+}
 """
 
 demo = gr.Interface(
@@ -119,16 +108,11 @@ demo = gr.Interface(
         gr.Slider(minimum=0.1, maximum=0.9, value=0.3, step=0.05, label="Threshold"),
         gr.Image(type="pil", label="Upload Face Photo")
     ],
-    outputs=[
-        gr.Textbox(label="Detected Text Emotions", elem_classes=["output-textbox"], lines=1),
-        gr.HTML(label="Quranic/Hadith Advice (Text)"),
-        gr.Textbox(label="Detected Photo Emotion", elem_classes=["output-textbox"], lines=1),
-        gr.HTML(label="Quranic/Hadith Advice (Image)")
-    ],
+    outputs=gr.HTML(label="Detected Emotion & Islamic Insight"),
     title="🧠 EmotionLens: Understand & Reflect Emotionally",
-    description="An AI-powered emotional intelligence assistant with Islamic wisdom. Analyze how you feel from text and facial expressions, and receive gentle spiritual insight.",
+    description="Analyze how you feel through text and facial expression, and receive inspirational Islamic wisdom in a beautifully styled reflection.",
     theme="default",
     css=custom_css
 )
 
-demo.launch()
+demo.launch()