Update app.py

app.py

@@ -3,14 +3,14 @@ from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassifica
 import requests
 import torch
 
-# مدل merged شما
+# مدل شما (merged version که push کردی)
 MODEL_REPO = "OmidSakaki/mental-health-deberta"
 
-print("Loading model and tokenizer...")
+print("در حال بارگذاری مدل و توکنایزر...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_REPO)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL_REPO)
 
-# Pipeline برای طبقه‌بندی
+# ساخت pipeline برای طبقه‌بندی
 classifier = pipeline(
     "text-classification",
     model=model,
@@ -22,13 +22,14 @@ classifier = pipeline(
 TRANSLATE_API_URL = "https://api.mymemory.translated.net/get"
 
 def translate_to_english(text_fa: str) -> str:
-    if not text_fa.strip():
+    if not text_fa or 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()
@@ -38,28 +39,28 @@ def translate_to_english(text_fa: str) -> str:
     except Exception as e:
         return f"خطا در ترجمه: {str(e)}"
 
+
 def classify_mental_health(text_fa: str) -> str:
     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"
-        "⚠️ اگر احساس بحران دارید، فوراً با ۱۲۳ (اورژانس اجتماعی) یا ۱۴۸۰ (مشاوره بهزیستی) تماس بگیرید."
+
+    # خروجی پایه (در صورت خطا استفاده می‌شود)
+    result_text = (
+        #f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
+        "متأسفانه در پردازش مدل مشکلی پیش آمد.\n"
+        "لطفاً متن کوتاه‌تر یا واضح‌تری وارد کنید و دوباره امتحان کنید.\n"
+        "⚠️ اگر احساس بحران یا خطر دارید، فوراً با ۱۲۳ (اورژانس اجتماعی) یا ۱۴۸۰ (مشاوره بهزیستی) تماس بگیرید."
     )
-    
+
     try:
         # طبقه‌بندی
         result = classifier(text_en, truncation=True, max_length=512)[0]
         label_en = result['label']
         score = result['score'] * 100
-        
-        # نقشه لیبل به فارسی
+
         label_map = {
             'Normal': 'نرمال / عادی',
             'Depression': 'افسردگی',
@@ -67,62 +68,63 @@ def classify_mental_health(text_fa: str) -> str:
             'Suicidal': 'افکار خودکشی'
         }
         label_fa = label_map.get(label_en, label_en)
-        
-        # توصیه‌های حمایتی
+
+        # توصیه حمایتی
         advice = ""
         if label_en == 'Depression':
             advice = (
-                "احساس افسردگی خیلی سنگین است، اما شما تنها نیستید و این وضعیت قابل بهبود است.\n"
-                "امروز با یک دوست یا عضو خانواده مورد اعتماد صحبت کنید.\n"
-                "حتی قدم‌های کوچک مثل پیاده‌روی کوتاه یا نوشتن احساسات کمک‌کننده است.\n"
+                "احساس افسردگی سنگین است، اما شما تنها نیستید و کمک وجود دارد.\n"
+                "امروز با یک فرد مورد اعتماد صحبت کنید.\n"
+                "فعالیت‌های کوچک لذت‌بخش را امتحان کنید.\n"
                 "خواب و تغذیه منظم را جدی بگیرید.\n"
-                "حتماً به روانشناس یا روانپزشک مراجعه کنید – درمان‌های مؤثری وجود دارد.\n"
+                "حتماً به روانشناس یا روانپزشک مراجعه کنید – درمان مؤثر است.\n"
                 "شما ارزشمند هستید و شرایط می‌تواند بهتر شود."
             )
         elif label_en == 'Anxiety':
             advice = (
-                "اضطراب شدید می‌تواند خیلی آزاردهنده باشد، اما راه‌هایی برای کنترل آن وجود دارد.\n"
-                "چند نفس عمیق و آهسته بکشید (۴ ثانیه دم، ۴ ثانیه نگه‌داشتن، ۶ ثانیه بازدم).\n"
-                "به ۵ چیز اطراف خود که می‌بینید، ۴ چیز که لمس می‌کنید، ۳ چیز که می‌شنوید توجه کنید.\n"
-                "از کافئین و اخبار استرس‌زا دوری کنید.\n"
-                "با یک فرد نزدیک یا مشاور صحبت کنید.\n"
-                "این احساس موقتی است و شما قادر به عبور از آن هستید."
+                "اضطراب زیاد آزاردهنده است، اما قابل کنترل است.\n"
+                "نفس عمیق بکشید (۴ ثانیه دم، ۴ نگه، ۶ بازدم).\n"
+                "به اطرافتان توجه کنید (۵ چیز ببینید، ۴ چیز لمس کنید).\n"
+                "از محرک‌ها (کافئین، اخبار) دوری کنید.\n"
+                "با کسی صحبت کنید.\n"
+                "این احساس می‌گذرد و شما قوی هستید."
             )
         elif label_en == 'Suicidal':
             advice = (
-                "اگر افکار خودکشی دارید، لطفاً همین حالا کمک بگیرید – شما مهم هستید.\n"
-                "فوراً با خطوط حمایتی تماس بگیرید: در ایران شماره ۱۲۳ (اورژانس اجتماعی) یا ۱۴۸۰ (مشاوره بهزیستی).\n"
-                "وسایل خطرناک را از دسترس دور کنید (حتی برای مدتی کوتاه).\n"
-                "با یک فرد مورد اعتماد صحبت کنید یا به نزدیک‌ترین مرکز درمانی بروید.\n"
-                "این افکار گذرا هستند و کمک واقعی وجود دارد – لطفاً یک قدم بردارید.\n"
-                "زندگی شما ارزشمند است و شرایط می‌تواند تغییر کند."
+                "اگر افکار خودکشی دارید، لطفاً همین حالا کمک بگیرید.\n"
+                "فوراً با ۱۲۳ (اورژانس اجتماعی) یا ۱۴۸۰ (مشاوره بهزیستی) تماس بگیرید.\n"
+                "وسایل خطرناک را دور کنید.\n"
+                "با یک فرد نزدیک صحبت کنید.\n"
+                "به مرکز درمانی بروید.\n"
+                "زندگی شما ارزشمند است – کمک وجود دارد."
             )
-        
-        # ساخت خروجی نهایی
-        #output = f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
-        output += f"وضعیت تشخیص داده شده: **{label_fa}**\n"
-        output += f"اطمینان مدل: {score:.1f}%\n\n"
-        
+
+        # خروجی موفق
+        #result_text = f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
+        result_text += f"وضعیت تشخیص داده شده: **{label_fa}**\n"
+        result_text += f"اطمینان مدل: {score:.1f}%\n\n"
+
         if advice:
-            output += "**توصیه حمایتی کوتاه:**\n" + advice + "\n\n"
-        
-        output += "⚠️ **توجه مهم:** این فقط یک تحلیل اولیه توسط هوش مصنوعی است و جایگزین نظر متخصص روانشناس یا روانپزشک نمی‌شود. اگر احساس خطر یا بحران دارید، فوراً با خطوط حمایتی (۱۲۳ یا ۱۴۸۰ در ایران) تماس بگیرید یا به مرکز درمانی مراجعه کنید."
-        
+            result_text += "**توصیه حمایتی کوتاه:**\n" + advice + "\n\n"
+
+        result_text += (
+            "⚠️ **توجه مهم:**\n"
+            "این فقط یک تحلیل اولیه توسط هوش مصنوعی است و جایگزین نظر متخصص روانشناس یا روانپزشک نمی‌شود.\n"
+            "در موارد بحران یا احساس خطر فوراً با خطوط حمایتی (۱۲۳ یا ۱۴۸۰ در ایران) تماس بگیرید یا به نزدیک‌ترین مرکز درمانی مراجعه کنید."
+        )
+
     except Exception as e:
-        # در صورت خطا، خروجی پیش‌فرض را با جزئیات خطا غنی‌تر می‌کنیم
-        #output = f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
-        output += f"خطا در پردازش مدل: {str(e)}\n"
-        output += "لطفاً دوباره امتحان کنید یا متن کوتاه‌تری وارد کنید.\n"
-        output += "⚠️ اگر احساس بحران دارید، فوراً با ۱۲۳ یا ۱۴۸۰ تماس بگیرید."
-    
-    return output
+        result_text += f"\n\nجزئیات خطا: {str(e)}"
+
+    return result_text
+
 
-# رابط Gradio
+# رابط کاربری Gradio
 demo = gr.Interface(
     fn=classify_mental_health,
     inputs=gr.Textbox(
         label="متن فارسی خود را وارد کنید",
-        placeholder="مثال: احساس می‌کنم هیچ امیدی ندارم و خسته‌ام...",
+        placeholder="مثال: احساس می��کنم دیگه هیچ امیدی ندارم...",
         lines=7,
         max_lines=15
     ),
@@ -132,13 +134,15 @@ demo = gr.Interface(
     ),
     title="تحلیل وضعیت سلامت روان از روی متن فارسی",
     description=(
-        "متن فارسی بنویسید → توسط مدل تحلیل می‌شود → نتیجه و توصیه حمایتی به فارسی نمایش داده می‌شود.\n\n"
-        "کلاس‌ها: نرمال، افسردگی، اضطراب، افکار خودکشی\n\n"
-        "این ابزار آزمایشی است و جایگزین مشاوره حرفه‌ای نیست. در موارد بحران فوراً با ۱۲۳ یا ۱۴۸۰ تماس بگیرید."
+        "متن فارسی بنویسید → توسط مدل تحلیل می‌شود → "
+        "نتیجه و توصیه حمایتی به فارسی نمایش داده می‌شود.\n\n"
+        "کلاس‌های تشخیص: نرمال، افسردگی، اضطراب، افکار خودکشی\n\n"
+        "این ابزار فقط جنبه آزمایشی و آموزشی دارد و جایگزین مشاوره حرفه‌ای نیست.\n"
+        "در موارد بحران فوراً با ۱۲۳ یا ۱۴۸۰ تماس بگیرید."
     ),
     examples=[
         ["مدام نگرانم و نمی‌تونم آرام باشم"],
-        ["دیگه هیچی برام مهم نیست، خسته شدم از همه چیز"],
+        ["دیگه هیچی برام مهم نیست، خسته شدم"],
         ["احساس می‌کنم زندگی ارزش ادامه دادن نداره"]
     ],
     cache_examples=False,
@@ -149,6 +153,5 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=True,
         debug=False
     )