Update app.py

app.py

@@ -39,12 +39,20 @@ def translate_to_english(text_fa: str) -> str:
         return f"خطا در ترجمه: {str(e)}"
 
 def classify_mental_health(text_fa: str) -> str:
-    if not text_fa.strip():
+    if not text_fa or not text_fa.strip():
         return "لطفاً متن فارسی وارد کنید."
 
     # ترجمه به انگلیسی
     text_en = translate_to_english(text_fa)
     
+    # مقدار پیش‌فرض برای خروجی (در صورت خطا استفاده می‌شود)
+    output = (
+        f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
+        "خطا در پردازش مدل رخ داد.\n"
+        "لطفاً دوباره امتحان کنید یا متن دیگری وارد کنید.\n"
+        "⚠️ اگر احساس بحران دارید، فوراً با ۱۲۳ (اورژانس اجتماعی) یا ۱۴۸۰ (مشاوره بهزیستی) تماس بگیرید."
+    )
+    
     try:
         # طبقه‌بندی
         result = classifier(text_en, truncation=True, max_length=512)[0]
@@ -60,7 +68,7 @@ def classify_mental_health(text_fa: str) -> str:
         }
         label_fa = label_map.get(label_en, label_en)
         
-        # توصیه‌های حمایتی کوتاه (فقط برای کلاس‌های حساس)
+        # توصیه‌های حمایتی
         advice = ""
         if label_en == 'Depression':
             advice = (
@@ -71,7 +79,6 @@ def classify_mental_health(text_fa: str) -> str:
                 "حتماً به روانشناس یا روانپزشک مراجعه کنید – درمان‌های مؤثری وجود دارد.\n"
                 "شما ارزشمند هستید و شرایط می‌تواند بهتر شود."
             )
-        
         elif label_en == 'Anxiety':
             advice = (
                 "اضطراب شدید می‌تواند خیلی آزاردهنده باشد، اما راه‌هایی برای کنترل آن وجود دارد.\n"
@@ -81,19 +88,18 @@ def classify_mental_health(text_fa: str) -> str:
                 "با یک فرد نزدیک یا مشاور صحبت کنید.\n"
                 "این احساس موقتی است و شما قادر به عبور از آن هستید."
             )
-        
         elif label_en == 'Suicidal':
             advice = (
                 "اگر افکار خودکشی دارید، لطفاً همین حالا کمک بگیرید – شما مهم هستید.\n"
-                "فوراً با خطوط حمایتی تماس بگیرید: در ایران شماره ۱۲۳ (اورژانس اجتماعی) یا ۱۴۸۰ (مشاوره بهزیستی) را بگیرید.\n"
+                "فوراً با خطوط حمایتی تماس بگیرید: در ایران شماره ۱۲۳ (اورژانس اجتماعی) یا ۱۴۸۰ (مشاوره بهزیستی).\n"
                 "وسایل خطرناک را از دسترس دور کنید (حتی برای مدتی کوتاه).\n"
                 "با یک فرد مورد اعتماد صحبت کنید یا به نزدیک‌ترین مرکز درمانی بروید.\n"
                 "این افکار گذرا هستند و کمک واقعی وجود دارد – لطفاً یک قدم بردارید.\n"
                 "زندگی شما ارزشمند است و شرایط می‌تواند تغییر کند."
             )
         
-        # خروجی نهایی
-        #output = f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
+        # ساخت خروجی نهایی
+        output = f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
         output += f"وضعیت تشخیص داده شده: **{label_fa}**\n"
         output += f"اطمینان مدل: {score:.1f}%\n\n"
         
@@ -102,10 +108,14 @@ def classify_mental_health(text_fa: str) -> str:
         
         output += "⚠️ **توجه مهم:** این فقط یک تحلیل اولیه توسط هوش مصنوعی است و جایگزین نظر متخصص روانشناس یا روانپزشک نمی‌شود. اگر احساس خطر یا بحران دارید، فوراً با خطوط حمایتی (۱۲۳ یا ۱۴۸۰ در ایران) تماس بگیرید یا به مرکز درمانی مراجعه کنید."
         
-        return output
-    
     except Exception as e:
-        return f"خطا در پردازش: {str(e)}\nلطفاً دوباره امتحان کنید یا با متخصص مشورت کنید."
+        # در صورت خطا، خروجی پیش‌فرض را با جزئیات خطا غنی‌تر می‌کنیم
+        #output = f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
+        output += f"خطا در پردازش مدل: {str(e)}\n"
+        output += "لطفاً دوباره امتحان کنید یا متن کوتاه‌تری وارد کنید.\n"
+        output += "⚠️ اگر احساس بحران دارید، فوراً با ۱۲۳ یا ۱۴۸۰ تماس بگیرید."
+    
+    return output
 
 # راب�� Gradio
 demo = gr.Interface(