Update app.py

app.py

@@ -1,170 +1,45 @@
 import gradio as gr
-import pandas as pd
-import os
-from dotenv import load_dotenv
-import google.generativeai as genai
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, AutoImageProcessor, AutoModelForImageClassification
 from torch.nn.functional import sigmoid
 import torch
 from PIL import Image
 
-load_dotenv()
-genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
-model_gemini = genai.GenerativeModel("gemini-pro")
 
-def get_gemini_advice_from_text(text):
-    prompt = f"""Suggest one Qur'an verse or Hadith to comfort and guide a person who wrote the following:
-
-'{text}'
-
-Please include:
-- Arabic verse or hadith
-- English translation
-- Source (Surah or Hadith reference)
-
-Wrap it in a gentle tone."""
-    try:
-        response = model_gemini.generate_content(prompt)
-        return f"""<div class='notion-card fade-in'>
-<p style='white-space: pre-wrap;'>{response.text}</p>
-</div>"""
-    except Exception as e:
-        return f"<div class='notion-card fade-in'><p>May Allah guide your heart.</p></div>"
 
+# 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 = {
-    "sadness": "<h1 class='emoji'>😢</h1><div class='notion-card animated'><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 animated'><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 animated'><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 animated'><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 animated'><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 animated'><h3>Love</h3><p><strong>“The Most Merciful will appoint for them affection.”</strong><br><span class='arabic'>سَيَجْعَلُ لَهُمُ الرَّحْمَٰنُ وُدًّا</span><br><em>Qur'an, Maryam 19:96</em></p></div>",
-    "gratitude": "<h1 class='emoji'>🙏</h1><div class='notion-card animated'><h3>Gratitude</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>",
-    "disappointment": "<h1 class='emoji'>😞</h1><div class='notion-card animated'><h3>Disappointment</h3><p><strong>“Indeed, with hardship [will be] ease.”</strong><br><span class='arabic'>إِنَّ مَعَ الْعُسْرِ يُسْرًا</span><br><em>Qur'an, Ash-Sharh 94:6</em></p></div>",
-    "nervousness": "<h1 class='emoji'>😬</h1><div class='notion-card animated'><h3>Nervousness</h3><p><strong>“Put your trust in Allah. Indeed, Allah loves those who rely [upon Him].”</strong><br><span class='arabic'>وَتَوَكَّلْ عَلَى اللَّهِ</span><br><em>Qur'an, Ali 'Imran 3:159</em></p></div>",
-    "trust": "<h1 class='emoji'>🤝</h1><div class='notion-card animated'><h3>Trust</h3><p><strong>“And whoever puts his trust in Allah – then He is sufficient for him.”</strong><br><span class='arabic'>وَمَن يَتَوَكَّلْ عَلَى اللَّهِ فَهُوَ حَسْبُهُ</span><br><em>Qur'an, At-Talaq 65:3</em></p></div>",
-    "hope": "<h1 class='emoji'>🌤️</h1><div class='notion-card animated'><h3>Hope</h3><p><strong>“And never give up hope of Allah's mercy.”</strong><br><span class='arabic'>إِنَّهُ لَا يَيْأَسُ مِن رَّوْحِ اللَّهِ إِلَّا الْقَوْمُ الْكَافِرُونَ</span><br><em>Qur'an, Yusuf 12:87</em></p></div>",
-    "remorse": "<h1 class='emoji'>😔</h1><div class='notion-card animated'><h3>Remorse</h3><p><strong>“Indeed, Allah is Forgiving and Merciful.”</strong><br><span class='arabic'>إِنَّ اللَّهَ غَفُورٌ رَّحِيمٌ</span><br><em>Qur'an, Al-Baqarah 2:218</em></p></div>",
-    "caring": "<h1 class='emoji'>🤗</h1><div class='notion-card animated'><h3>Caring</h3><p><strong>“The believers are but brothers, so make peace between your brothers.”</strong><br><span class='arabic'>إِنَّمَا الْمُؤْمِنُونَ إِخْوَةٌ</span><br><em>Qur'an, Al-Hujurat 49:10</em></p></div>",
-    "surprise": "<h1 class='emoji'>😲</h1><div class='notion-card animated'><h3>Surprise</h3><p><strong>“Allah creates what you do not know.”</strong><br><span class='arabic'>وَيَخْلُقُ مَا لَا تَعْلَمُونَ</span><br><em>Qur'an, An-Nahl 16:8</em></p></div>",
-    "embarrassment": "<h1 class='emoji'>😳</h1><div class='notion-card animated'><h3>Embarrassment</h3><p><strong>“Indeed, Allah is ever Accepting of repentance.”</strong><br><span class='arabic'>إِنَّهُ كَانَ تَوَّابًا</span><br><em>Qur'an, An-Nisa 4:16</em></p></div>",
-    "relief": "<h1 class='emoji'>😌</h1><div class='notion-card animated'><h3>Relief</h3><p><strong>“And He it is who sent down tranquility into the hearts of the believers.”</strong><br><span class='arabic'>هُوَ الَّذِي أَنزَلَ السَّكِينَةَ فِي قُلُوبِ الْمُؤْمِنِينَ</span><br><em>Qur'an, Al-Fath 48:4</em></p></div>",
-    "neutral": "<h1 class='emoji'>😐</h1><div class='notion-card animated'><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")
 
-def analyze_combined(text, threshold, image):
-    # Text analysis
-    inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
     with torch.no_grad():
-        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()
-    
-    text_card = islamic_advice.get(text_label)
-    if not text_card:
-        text_card = get_gemini_advice_from_text(text)
-
-    # Image analysis
-    image_card = ""
-    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()
-        image_card = islamic_advice.get(img_label, f"<div class='notion-card fade-in'><h3>{img_label.capitalize()}</h3><p>May Allah guide your expression.</p></div>")
-
-    return f"""
-<div class='notion-card fade-in slide-up glow-card'>
-<h2 style='text-align:center'>📝 From Text Analysis</h2>
-{text_card}
-</div>
-
-{f"<div class='notion-card fade-in slide-up glow-card'><h2 style='text-align:center'>🖼️ From Image Analysis</h2>{image_card}</div>" if image else ""}
-"""
-
-custom_css = """
-@keyframes slideInUp {
-  from {
-    transform: translateY(20px);
-    opacity: 0;
-  }
-  to {
-    transform: translateY(0);
-    opacity: 1;
-  }
-}
-@keyframes glowFadeIn {
-  0% {
-    box-shadow: 0 0 0px rgba(0, 0, 0, 0);
-    opacity: 0;
-  }
-  100% {
-    box-shadow: 0 0 15px rgba(59, 130, 246, 0.3);
-    opacity: 1;
-  }
-}
-.fade-in {
-  animation: fadeInPop 0.6s ease-out both;
-}
-.slide-up {
-  animation: slideInUp 0.6s ease-out both;
-}
-.glow-card {
-  animation: glowFadeIn 1s ease-in-out both;
-}
-@keyframes fadeInPop {
-  0% {
-    opacity: 0;
-    transform: scale(0.95);
-  }
-  100% {
-    opacity: 1;
-    transform: scale(1);
-  }
-}
-.fade-in {
-  animation: fadeInPop 0.6s ease-out both;
-}
-body {
-  background: #f9fafb;
-  font-family: 'Inter', sans-serif;
-  color: #1f2937;
-  padding: 20px;
-}
-.gr-button {
-  background-color: #3b82f6 !important;
-  color: white !important;
-  border-radius: 8px !important;
-  font-weight: 600;
-  padding: 10px 20px;
-  transition: background-color 0.3s ease;
-}
-.gr-button:hover {
-  background-color: #2563eb !important;
-}
-"""
-
-with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown("# 🧠 EmotionLens")
-    gr.Markdown("Analyze your text and optionally a facial photo. Receive emotional insight and reflective Islamic advice.")
-
-    with gr.Tab("Single Input"):
-        text_input = gr.Textbox(lines=4, label="Text Input")
-        threshold_slider = gr.Slider(0.1, 0.9, value=0.3, step=0.05, label="Threshold")
-        image_input = gr.Image(type="pil", label="Upload Face Photo (optional)")
-        btn = gr.Button("Analyze Emotion")
-        result = gr.HTML()
-        btn.click(fn=analyze_combined, inputs=[text_input, threshold_slider, image_input], outputs=result)
-
-    
-demo.launch()
+        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]
+    score = probs[pred_idx].item()
+    return f"{label} ({score:.2f})"
+
+# Emotion label to icon mapping (subset)
+emotion_icons = {
+    inputs=[
+        gr.Textbox(lines=5, placeholder="Write a sentence or a full paragraph...", label="Your Text"),
+        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"]),
+    css=custom_css
+)
+
+demo.launch()