Create app.py

app.py

@@ -0,0 +1,83 @@
+import gradio as gr
+from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
+import requests
+import json
+
+# مدلت رو از repo خودت لود کن
+MODEL_REPO = "OmidSakaki/mental-health-deberta"  # یا هر نامی که push کردی
+
+tokenizer = AutoTokenizer.from_pretrained(MODEL_REPO)
+model = AutoModelForSequenceClassification.from_pretrained(MODEL_REPO)
+
+# pipeline برای classification
+classifier = pipeline("text-classification", model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)
+
+# API ترجمه فارسی به انگلیسی (رایگان – می‌تونی عوض کنی)
+TRANSLATE_API_URL = "https://api.mymemory.translated.net/get"  # یا DeepL / Google Translate API اگر کلید داری
+
+def translate_to_english(text_fa):
+    """ترجمه متن فارسی به انگلیسی با API رایگان"""
+    if not text_fa.strip():
+        return "متن وارد نشده است"
+    
+    params = {
+        "q": text_fa,
+        "langpair": "fa|en"
+    }
+    try:
+        response = requests.get(TRANSLATE_API_URL, params=params, timeout=10)
+        data = response.json()
+        translated = data["responseData"]["translatedText"]
+        return translated
+    except Exception as e:
+        return f"خطا در ترجمه: {str(e)}"
+
+def classify_mental_health(text_fa):
+    # مرحله ۱: ترجمه به انگلیسی
+    text_en = translate_to_english(text_fa)
+    
+    # مرحله ۲: طبقه‌بندی با مدل
+    result = classifier(text_en)[0]
+    label = result['label']          # مثلاً 'Suicidal' یا 'Depression'
+    score = result['score'] * 100    # درصد اطمینان
+    
+    # مرحله ۳: تبدیل لیبل به فارسی برای خروجی کاربرپسند
+    label_fa_map = {
+        'Normal': 'نرمال / عادی',
+        'Depression': 'افسردگی',
+        'Anxiety': 'اضطراب',
+        'Suicidal': 'افکار خودکشی'
+    }
+    label_fa = label_fa_map.get(label, label)  # اگر لیبل ناشناخته بود همون انگلیسی بمونه
+    
+    # خروجی نهایی
+    output = f"متن ترجمه‌شده به انگلیسی:\n{text_en}\n\n"
+    output += f"وضعیت تشخیص داده شده: **{label_fa}**\n"
+    output += f"اطمینان مدل: {score:.1f}%"
+    
+    return output
+
+# ساخت رابط Gradio
+demo = gr.Interface(
+    fn=classify_mental_health,
+    inputs=gr.Textbox(
+        label="متن فارسی خود را وارد کنید",
+        placeholder="مثال: احساس می‌کنم دیگه نمی‌تونم ادامه بدم...",
+        lines=5
+    ),
+    outputs=gr.Textbox(label="نتیجه تشخیص"),
+    title="تشخیص وضعیت سلامت روان از روی متن فارسی",
+    description="""متن فارسی وارد کنید → به انگلیسی ترجمه می‌شود → توسط مدل تشخیص داده می‌شود → نتیجه به فارسی نمایش داده می‌شود.
+    
+    کلاس‌ها: نرمال، افسردگی، اضطراب، افکار خودکشی""",
+    examples=[
+        ["امروز خیلی خوب بودم و خوشحالم"],
+        ["دیگه نمی‌تونم تحمل کنم، همه چیز تموم بشه بهتره"],
+        ["قلبم تند می‌زنه و نمی‌تونم آرام باشم"]
+    ],
+    theme=gr.themes.Soft(),
+    allow_flagging="never"
+)
+
+if __name__ == "__main__":
+    demo.launch()