Update app.py

app.py

@@ -1,10 +1,38 @@
 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(emotion):
+    prompt = f"""Give me one verse from the Qur'an or Hadith that is relevant to someone feeling {emotion}.
+Format it like this:
+
+Arabic:
+[Verse Arabic]
+
+English Translation:
+[Translation]
+
+Source:
+[Surah or Hadith Reference]"""
+    try:
+        response = model_gemini.generate_content(prompt)
+        return f"""<div class='notion-card animated'>
+<h3>{emotion.capitalize()}</h3>
+<p style='white-space: pre-wrap;'>{response.text}</p>
+</div>"""
+    except Exception as e:
+        return f"<div class='notion-card'><h3>{emotion.capitalize()}</h3><p>May Allah guide your heart. (AI fallback failed)</p></div>"
+
 model_name = "SamLowe/roberta-base-go_emotions"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
@@ -19,7 +47,17 @@ islamic_advice = {
     "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>“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>",
+    "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>"
 }
 
@@ -42,9 +80,8 @@ def analyze_combined(text, threshold, image):
         img_label = None
 
     final_label = text_label if img_label is None else img_label
-    card = islamic_advice.get(final_label, islamic_advice["neutral"])
+    card = islamic_advice.get(final_label) or get_gemini_advice(final_label)
     return card
-    return f"<h2 style='font-size:2rem;text-align:center;'>{final_label.capitalize()}</h2><p style='text-align:center;font-size:1.2rem;'>{advice}</p>"
 
 def analyze_batch_csv(file):
     df = pd.read_csv(file.name)