Update app.py

app.py

@@ -1,18 +1,16 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
 import requests
-import json
 import torch
 
-# مدل شما (merged version)
+# مدل merged شما
 MODEL_REPO = "OmidSakaki/mental-health-deberta"
 
-print("در حال لود مدل و توکنایزر...")
+print("Loading model and tokenizer...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_REPO)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL_REPO)
 
-# ساخت pipeline برای طبقه‌بندی
-# device مدیریت می‌شود: اگر GPU باشد از 0 استفاده می‌کند، در غیر این صورت CPU (-1)
+# Pipeline برای طبقه‌بندی
 classifier = pipeline(
     "text-classification",
     model=model,
@@ -20,96 +18,127 @@ classifier = pipeline(
     device=0 if torch.cuda.is_available() else -1
 )
 
-# API ترجمه فارسی → انگلیسی (رایگان - MyMemory)
+# API ترجمه فارسی به انگلیسی (MyMemory - رایگان)
 TRANSLATE_API_URL = "https://api.mymemory.translated.net/get"
 
 def translate_to_english(text_fa: str) -> str:
-    """ترجمه متن فارسی به انگلیسی"""
-    if not text_fa or not text_fa.strip():
+    if not text_fa.strip():
         return "متن وارد نشده است"
     
     params = {
         "q": text_fa,
         "langpair": "fa|en"
     }
-    
     try:
         response = requests.get(TRANSLATE_API_URL, params=params, timeout=12)
         response.raise_for_status()
         data = response.json()
         translated = data.get("responseData", {}).get("translatedText", "")
-        if not translated:
-            return "ترجمه انجام نشد (پاسخ نامعتبر)"
-        return translated
+        return translated if translated else "ترجمه انجام نشد"
     except Exception as e:
         return f"خطا در ترجمه: {str(e)}"
 
 def classify_mental_health(text_fa: str) -> str:
-    # مرحله ۱: ترجمه
+    if not text_fa.strip():
+        return "لطفاً متن فارسی وارد کنید."
+
+    # ترجمه به انگلیسی
     text_en = translate_to_english(text_fa)
     
-    # مرحله ۲: طبقه‌بندی
     try:
+        # طبقه‌بندی
         result = classifier(text_en, truncation=True, max_length=512)[0]
         label_en = result['label']
         score = result['score'] * 100
         
-        # نقشه‌برداری لیبل انگلیسی به فارسی
+        # نقشه لیبل به فارسی
         label_map = {
             'Normal': 'نرمال / عادی',
             'Depression': 'افسردگی',
             'Anxiety': 'اضطراب',
-            'Suicidal': 'افکار خودکشی / خودکشی'
+            'Suicidal': 'افکار خودکشی'
         }
-        label_fa = label_map.get(label_en, f"ناشناخته ({label_en})")
+        label_fa = label_map.get(label_en, label_en)
+        
+        # توصیه‌های حمایتی کوتاه (فقط برای کلاس‌های حساس)
+        advice = ""
+        if label_en == 'Depression':
+            advice = (
+                "احساس افسردگی خیلی سنگین است، اما شما تنها نیستید و این وضعیت قابل بهبود است.\n"
+                "امروز با یک دوست یا عضو خانواده مورد اعتماد صحبت کنید.\n"
+                "حتی قدم‌های کوچک مثل پیاده‌روی کوتاه یا نوشتن احساسات کمک‌کننده است.\n"
+                "خواب و تغذیه منظم را جدی بگیرید.\n"
+                "حتماً به روانشناس یا روانپزشک مراجعه کنید – درمان‌های مؤثری وجود دارد.\n"
+                "شما ارزشمند هستید و شرایط می‌تواند بهتر شود."
+            )
+        
+        elif label_en == 'Anxiety':
+            advice = (
+                "اضطراب شدید می‌تواند خیلی آزاردهنده باشد، اما راه‌هایی برای کنترل آن وجود دارد.\n"
+                "چند نفس عمیق و آهسته بکشید (۴ ثانیه دم، ۴ ثانیه نگه‌داشتن، ۶ ثانیه بازدم).\n"
+                "به ۵ چیز اطراف خود که می‌بینید، ۴ چیز که لمس می‌کنید، ۳ چیز که می‌شنوید توجه کنید.\n"
+                "از کافئین و اخبار استرس‌زا دوری کنید.\n"
+                "با یک فرد نزدیک یا مشاور صحبت کنید.\n"
+                "این احساس موقتی است و شما قادر به عبور از آن هستید."
+            )
         
-        # خروجی خوانا
-        output = f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
+        elif label_en == 'Suicidal':
+            advice = (
+                "اگر افکار خودکشی دارید، لطفاً همین حال�� کمک بگیرید – شما مهم هستید.\n"
+                "فوراً با خطوط حمایتی تماس بگیرید: در ایران شماره ۱۲۳ (اورژانس اجتماعی) یا ۱۴۸۰ (مشاوره بهزیستی) را بگیرید.\n"
+                "وسایل خطرناک را از دسترس دور کنید (حتی برای مدتی کوتاه).\n"
+                "با یک فرد مورد اعتماد صحبت کنید یا به نزدیک‌ترین مرکز درمانی بروید.\n"
+                "این افکار گذرا هستند و کمک واقعی وجود دارد – لطفاً یک قدم بردارید.\n"
+                "زندگی شما ارزشمند است و شرایط می‌تواند تغییر کند."
+            )
+        
+        # خروجی نهایی
+        #output = f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
         output += f"وضعیت تشخیص داده شده: **{label_fa}**\n"
         output += f"اطمینان مدل: {score:.1f}%\n\n"
-        output += "(این فقط یک تحلیل اولیه بر اساس مدل است و جایگزین مشاوره حرفه‌ای نمی‌شود)"
+        
+        if advice:
+            output += "**توصیه حمایتی کوتاه:**\n" + advice + "\n\n"
+        
+        output += "⚠️ **توجه مهم:** این فقط یک تحلیل اولیه توسط هوش مصنوعی است و جایگزین نظر متخصص روانشناس یا روانپزشک نمی‌شود. اگر احساس خطر یا بحران دارید، فوراً با خطوط حمایتی (۱۲۳ یا ۱۴۸۰ در ایران) تماس بگیرید یا به مرکز درمانی مراجعه کنید."
         
         return output
     
     except Exception as e:
-        return f"خطا در پردازش مدل: {str(e)}"
+        return f"خطا در پردازش: {str(e)}\nلطفاً دوباره امتحان کنید یا با متخصص مشورت کنید."
 
-# رابط کاربری Gradio
+# رابط Gradio
 demo = gr.Interface(
     fn=classify_mental_health,
     inputs=gr.Textbox(
         label="متن فارسی خود را وارد کنید",
-        placeholder="مثال: احساس می‌کنم دیگه هیچ امیدی ندارم و خسته‌ام...",
-        lines=6,
-        max_lines=12
+        placeholder="مثال: احساس می‌کنم هیچ امیدی ندارم و خسته‌ام...",
+        lines=7,
+        max_lines=15
     ),
     outputs=gr.Textbox(
-        label="نتیجه تحلیل",
-        lines=8
+        label="نتیجه تحلیل و توصیه",
+        lines=12
     ),
-    title="تشخیص وضعیت سلامت روان از روی متن (فارسی)",
+    title="تحلیل وضعیت سلامت روان از روی متن فارسی",
     description=(
-        "۱. متن فارسی خود را بنویسید\n"
-        "۲. به طور خودکار به انگلیسی ترجمه می‌شود\n"
-        "۳. توسط مدل تخصصی تحلیل می‌شود\n"
-        "۴. نتیجه به فارسی نمایش داده می‌شود\n\n"
-        "کلاس‌های تشخیص: نرمال، افسردگی، اضطراب، افکار خودکشی\n\n"
-        "توجه: این ابزار فقط جنبه آزمایشی و آموزشی دارد و جایگزین نظر متخصص نیست."
+        "متن فارسی بنویسید → توسط مدل تحلیل می‌شود → نتیجه و توصیه حمایتی به فارسی نمایش داده می‌شود.\n\n"
+        "کلاس‌ها: نرمال، افسردگی، اضطراب، افکار خودکشی\n\n"
+        "این ابزار آزمایشی است و جایگزین مشاوره حرفه‌ای نیست. در موارد بحران فوراً با ۱۲۳ یا ۱۴۸۰ تماس بگیرید."
     ),
     examples=[
-        ["امروز روز خیلی خوبی داشتم و انرژی بالایی دارم"],
-        ["مدام نگرانم و نمی‌تونم آرام بگیرم، قلبم تند می‌زنه"],
-        ["احساس می‌کنم هیچی دیگه ارزش ادامه دادن نداره"],
-        ["همه چیز عادیه، نه خوشحالم نه ناراحت"]
+        ["امروز روز عالی بود و خیلی خوشحالم"],
+        ["مدام نگرانم و نمی‌تونم آرام باشم"],
+        ["دیگه هیچی برام مهم نیست، خسته شدم از همه چیز"],
+        ["احساس می‌کنم زندگی ارزش ادامه دادن نداره"]
     ],
     cache_examples=False,
-    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray")
+    theme=gr.themes.Soft(primary_hue="blue")
 )
 
 if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        theme=gr.themes.Soft(),
         debug=False
     )