Update app.py

app.py

@@ -27,6 +27,15 @@ BAI_OPTIONS = [
     {"score": 3, "text": "شدید (غیر قابل تحمل بود)", "label": "شدید (۱۰۰٪)", "color": 5}
 ]
 
+# گزینه‌های استاندارد برای آزمون اهمال‌کاری تحصیلی سولومون
+SOLOMON_OPTIONS = [
+    {"score": 1, "text": "هرگز", "label": "۱", "color": 1},
+    {"score": 2, "text": "به ندرت", "label": "۲", "color": 2},
+    {"score": 3, "text": "گهگاهی", "label": "۳", "color": 3},
+    {"score": 4, "text": "اکثر اوقات", "label": "۴", "color": 4},
+    {"score": 5, "text": "همیشه", "label": "۵", "color": 5}
+]
+
 # دیتابیس آزمون اول: ۳۰ سوال MBTI
 MBTI_QUESTIONS = [
     {"id": 1, "text": "در رویدادهای اجتماعی، من معمولاً با افراد جدید زیادی صحبت می‌کنم، نه فقط با کسانی که می‌شناسم.", "dimension": "EI", "direction": 1},
@@ -86,7 +95,7 @@ BAI_QUESTIONS = [
     {"id": 21, "text": "عرق کردن شدید که ناشی از دمای محیط یا فعالیت بدنی نباشد.", "dimension": "somatic"}
 ]
 
-# آزمون سوم: ۲۱ سوال افسردگی بک (BDI-II) - همراه با گزینه‌های بالینی اختصاصی هر سوال [2]
+# آزمون سوم: ۲۱ سوال افسردگی بک (BDI-II)
 BDI_QUESTIONS = [
     {
         "id": 1,
@@ -311,6 +320,37 @@ BDI_QUESTIONS = [
     }
 ]
 
+# آزمون چهارم: ۲۷ سوال اهمال‌کاری تحصیلی سولومون
+SOLOMON_QUESTIONS = [
+    {"id": 1, "text": "هنگام مطالعه برای امتحان مرتباً رویاپردازی می‌کنم و تمرکز کردن برایم دشوار است.", "dimension": "exam"},
+    {"id": 2, "text": "تا لحظه‌ای که امکان داشته باشد، آماده شدن برای امتحانات را به تأخیر نمی‌اندازم.", "dimension": "exam"},
+    {"id": 3, "text": "سازمان دادن مطالب قبل از امتحان برایم دشوار است.", "dimension": "exam"},
+    {"id": 4, "text": "هنگامی که از تاریخ یک امتحان اطلاع پیدا می‌کنم، اجازه نمی‌دهم که وقت تلف شود و فوراً شروع به آماده شدن می‌کنم.", "dimension": "exam"},
+    {"id": 5, "text": "هنگامی که تاریخ یک امتحان نزدیک شود، آماده شدن برای امتحان برایم دشوار است.", "dimension": "exam"},
+    {"id": 6, "text": "هنگامی که می‌بایست برای یک امتحان آماده شوم، بر روی آن متمرکز می‌شوم و دچار حواس‌پرتی نمی‌شوم.", "dimension": "exam"},
+    {"id": 7, "text": "هنگامی که آماده شدن برای امتحانات را به تعویق می‌اندازم، احساس بدی به من دست می‌دهد.", "dimension": "exam"},
+    {"id": 8, "text": "دوست دارم عادت تعلل‌ورزی و امروز و فردا کردن را در خودم تغییر دهم.", "dimension": "exam"},
+    {"id": 9, "text": "تکالیفم را از یک جلسه به جلسه دیگر به تعویق می‌اندازم.", "dimension": "assignment"},
+    {"id": 10, "text": "هنگامی که در حال انجام تکالیف هستم، چیزهای دیگری حواسم را پرت می‌کنند و برایم دشوار است که آن را تا پایان انجام دهم.", "dimension": "assignment"},
+    {"id": 11, "text": "ترجیح می‌دهم که ابتدا تکالیفم را انجام دهم و سپس به چیزهای دیگر بپردازم.", "dimension": "assignment"},
+    {"id": 12, "text": "تمایل دارم که انجام تکالیفم را تا دقیقه آخر به تأخیر بیاندازم.", "dimension": "assignment"},
+    {"id": 13, "text": "وقتی که برای مدرسه رفتن بیدار می‌شوم تقریباً فوراً از رختخواب بلند می‌شوم.", "dimension": "assignment"},
+    {"id": 14, "text": "دیر به کلاس یا مدرسه می‌رسم.", "dimension": "assignment"},
+    {"id": 15, "text": "وقتی که در کلاس چیزی نمی‌فهمم، مطمئنم که به‌زودی آن را خواهم فهمید (مثلاً کتابی می‌خوانم که آن را شرح داده باشد و یا با پرسیدن از دوستم آن را خواهم فهمید).", "dimension": "assignment"},
+    {"id": 16, "text": "پیش از اتمام رسیدن مهلت تحویل تکالیف، آن‌ها را تمام می‌کنم.", "dimension": "assignment"},
+    {"id": 17, "text": "هنگامی که می‌بایست تکالیفم را مرور کنم، آن را به تعویق می‌اندازم.", "dimension": "assignment"},
+    {"id": 18, "text": "وقتی که انجام دادن تکالیف را به تعویق می‌اندازم، احساس بدی به من دست می‌دهد.", "dimension": "assignment"},
+    {"id": 19, "text": "دوست دارم که عادت اهمال‌کاری و امروز و فردا کردن در انجام تکالیف را تغییر دهم.", "dimension": "assignment"},
+    {"id": 20, "text": "هنگامی که مجبور باشم تکلیفی را آماده کنم، مرتباً آن را به تأخیر می‌اندازم.", "dimension": "paper"},
+    {"id": 21, "text": "به محض مشخص شدن زمان انجام تکلیف، شروع به آماده کردن آن می‌کنم و آن را به تأخیر نمی‌اندازم.", "dimension": "paper"},
+    {"id": 22, "text": "هنگامی که تصمیم می‌گیرم بر روی تکلیفی کار کنم، احساس می‌کنم انرژی انجام دادن آن را ندارم.", "dimension": "paper"},
+    {"id": 23, "text": "هنگامی که برای انجام تکلیفی برنامه‌ریزی می‌کنم، دقیقاً به زمان و تاریخ آن پایبند هستم.", "dimension": "paper"},
+    {"id": 24, "text": "هنگامی که بر روی مقاله‌ای کار می‌کنم، نمی‌توانم روی آن تمرکز کنم و مرتباً رویاپردازی می‌کنم.", "dimension": "paper"},
+    {"id": 25, "text": "هنگامی که مجبورم تکلیفی را تا تاریخ معینی ارائه دهم، از چندین روز قبل آن را آماده می‌کنم.", "dimension": "paper"},
+    {"id": 26, "text": "هنگامی که انجام یک تکلیف را به تأخیر می‌اندازم، احساس بدی به من دست می‌دهد.", "dimension": "paper"},
+    {"id": 27, "text": "علاقه‌مند هستم که عادت اهمال‌کاری و امروز و فردا کردن در انجام تکالیف را تغییر دهم.", "dimension": "paper"}
+]
+
 class AnswerItem(BaseModel):
     question_id: int
     score: int
@@ -318,14 +358,15 @@ class AnswerItem(BaseModel):
 class AnalysisRequest(BaseModel):
     first_name: str
     last_name: str
-    test_type: str = Field(description="Type of test taken: 'mbti', 'bai', or 'bdi'")
+    test_type: str = Field(description="Type of test taken: 'mbti', 'bai', 'bdi', or 'solomon'")
     answers: List[AnswerItem]
 
 class UniversalAnalysisSchema(BaseModel):
-    test_type: str = Field(description="The type of test taken ('mbti', 'bai', or 'bdi')")
+    test_type: str = Field(description="The type of test taken ('mbti', 'bai', 'bdi', or 'solomon')")
     mbti_type: str = Field(description="MBTI 4-letter type. Leave blank if not MBTI.")
     bai_level: str = Field(description="BAI anxiety severity level. Leave blank if not BAI.")
     bdi_level: str = Field(description="BDI depression severity level. Leave blank if not BDI.")
+    solomon_level: str = Field(description="Solomon procrastination severity level. Leave blank if not Solomon.")
     personality_title: str = Field(description="A creative Persian title representing their state/personality")
     introversion_score: int = Field(description="Percentage of Introversion (0-100)")
     extraversion_score: int = Field(description="Percentage of Extraversion (0-100)")
@@ -335,11 +376,11 @@ class UniversalAnalysisSchema(BaseModel):
     feeling_score: int = Field(description="Percentage of Feeling (0-100)")
     judging_score: int = Field(description="Percentage of Judging (0-100)")
     perceiving_score: int = Field(description="Percentage of Perceiving (0-100)")
-    somatic_score: int = Field(description="Somatic/Physiological symptoms percentage (0-100)")
-    cognitive_score: int = Field(description="Cognitive/Subjective symptoms percentage (0-100)")
-    panic_score: int = Field(description="BAI: Panic/Fear symptoms percentage. Leave 0 if BDI.")
-    affective_score: int = Field(description="BDI: Affective/Emotional symptoms percentage. Leave 0 if BAI.")
-    total_score_raw: int = Field(description="Total raw score sum of BAI or BDI")
+    somatic_score: int = Field(description="Somatic symptoms / Exam procrastination percentage (0-100)")
+    cognitive_score: int = Field(description="Cognitive symptoms / Assignment procrastination percentage (0-100)")
+    panic_score: int = Field(description="BAI Panic / Solomon Paper procrastination percentage (0-100)")
+    affective_score: int = Field(description="BDI Affective percentage (0-100)")
+    total_score_raw: int = Field(description="Total raw score sum of BAI, BDI, or Solomon")
     summary: str = Field(description="A short Persian summary of the results (2-3 sentences)")
     strengths: List[str] = Field(description="Coping mechanisms, strengths or psychological resources (4-5 items in Persian)")
     weaknesses: List[str] = Field(description="Triggers, weaknesses, risk situations or symptoms to monitor (4-5 items in Persian)")
@@ -365,6 +406,13 @@ BDI_SYSTEM_INSTRUCTION = """
 لحن متن فوق‌العاده همدلانه، مهربان و امیدبخش باشد. بخش detailed_analysis را با تیترهای مارک‌داون جدا کنید.
 """
 
+SOLOMON_SYSTEM_INSTRUCTION = """
+شما یک روان‌شناس بالینی ارشد متخصص در حوزه روان‌شناسی تربیتی و تحصیلی و ارزیابی عادات مطالعه و تعلل‌ورزی بر اساس مقیاس اهمال‌کاری تحصیلی سولومون و راث‌بلوم (PASS) هستید.
+وظیفه شما تحلیل دقیق پاسخ‌های کاربر به پرسشنامه ۲۷ سواله اهمال‌کاری تحصیلی سولومون و تولید یک گزارش روان‌شناختی بسیار عمیق، علمی، کاربردی و ارائه‌دهنده راهکارهای عملی مدیریت زمان و غلبه بر تنبلی تحصیلی به زبان فارسی است.
+سه مؤلفه اصلی: آماده شدن برای امتحانات (exam)، آماده شدن برای تکالیف (assignment) و آماده شدن برای مقاله‌های پایان‌ترم (paper) را تحلیل کرده و راهکارهای درمانی معتبر مبتنی بر شناخت‌درمانی (CBT)، تکنیک‌های مدیریت تمرکز و برنامه‌ریزی تحصیلی ارائه دهید.
+بخش detailed_analysis را با تیترهای مارک‌داون تفکیک و تشریح کنید.
+"""
+
 def calculate_mbti_raw(answers: List[AnswerItem]):
     scores = {"EI": 0, "SN": 0, "TF": 0, "JP": 0}
     counts = {"EI": 0, "SN": 0, "TF": 0, "JP": 0}
@@ -445,6 +493,54 @@ def calculate_bdi_raw(answers: List[AnswerItem]):
     
     return {"level": level, "total": total_score, "cognitive": cognitive_pct, "affective": affective_pct, "somatic": somatic_pct}
 
+def calculate_solomon_raw(answers: List[AnswerItem]):
+    # گویه‌های نیازمند نمره‌گذاری معکوس طبق پرسشنامه: ۲ - ۴ - ۶ - ۱۱ - ۱۳ - ۱۵ - ۱۶ و ۲۱ - ۲۳ - ۲۵
+    reverse_ids = {2, 4, 6, 11, 13, 15, 16, 21, 23, 25}
+    
+    subscales = {
+        "exam": {"score": 0, "max": 0},
+        "assignment": {"score": 0, "max": 0},
+        "paper": {"score": 0, "max": 0}
+    }
+    total_score = 0
+    
+    q_map = {q["id"]: q for q in SOLOMON_QUESTIONS}
+    for ans in answers:
+        q_id = ans.question_id
+        if q_id in q_map:
+            q = q_map[q_id]
+            dim = q["dimension"]
+            
+            raw_val = max(1, min(5, ans.score))
+            if q_id in reverse_ids:
+                actual_score = 6 - raw_val  # نمره‌گذاری معکوس: ۵ به ۱، ۴ به ۲، ۳ به ۳، ۲ به ۴، ۱ به ۵
+            else:
+                actual_score = raw_val
+                
+            subscales[dim]["score"] += actual_score
+            subscales[dim]["max"] += 5
+            total_score += actual_score
+            
+    # سطح‌بندی بر اساس نمره کل بین ۲۷ تا ۱۳۵
+    if total_score <= 67:
+        level = "اهمال‌کاری ضعیف (مطلوب)"
+    elif total_score <= 101:
+        level = "اهمال‌کاری متوسط"
+    else:
+        level = "اهمال‌کاری شدید"
+        
+    exam_pct = round((subscales["exam"]["score"] / subscales["exam"]["max"]) * 100) if subscales["exam"]["max"] > 0 else 0
+    assignment_pct = round((subscales["assignment"]["score"] / subscales["assignment"]["max"]) * 100) if subscales["assignment"]["max"] > 0 else 0
+    paper_pct = round((subscales["paper"]["score"] / subscales["paper"]["max"]) * 100) if subscales["paper"]["max"] > 0 else 0
+    
+    return {
+        "level": level,
+        "total": total_score,
+        "exam": exam_pct,
+        "assignment": assignment_pct,
+        "paper": paper_pct
+    }
+
 @app.get("/", response_class=HTMLResponse)
 async def get_index():
     try:
@@ -456,7 +552,6 @@ async def get_index():
 @app.get("/api/questions")
 async def get_questions(test_type: str = "mbti"):
     if test_type == "bai":
-        # ارسال سوالات اضطراب بک همراه با گزینه‌های پیش‌فرض ۴ گزینه‌ای استاندارد
         return [
             {
                 "id": q["id"],
@@ -465,7 +560,6 @@ async def get_questions(test_type: str = "mbti"):
             } for q in BAI_QUESTIONS
         ]
     elif test_type == "bdi":
-        # ارسال سوالات افسردگی بک همراه با گزینه‌های بالینی و کاملا تخصصی منحصر به فرد خودش
         return [
             {
                 "id": q["id"],
@@ -473,8 +567,15 @@ async def get_questions(test_type: str = "mbti"):
                 "options": q["options"]
             } for q in BDI_QUESTIONS
         ]
+    elif test_type == "solomon":
+        return [
+            {
+                "id": q["id"],
+                "text": q["text"],
+                "options": SOLOMON_OPTIONS
+            } for q in SOLOMON_QUESTIONS
+        ]
     
-    # ارسال سوالات تیپ‌شناسی شخصیت MBTI همراه با گزینه‌های استاندارد ۵ گزینه‌ای
     return [
         {
             "id": q["id"],
@@ -529,7 +630,6 @@ async def analyze_personality(payload: AnalysisRequest):
         for ans in payload.answers:
             q = q_map.get(ans.question_id)
             if q:
-                # استخراج متن پاسخ اختصاصی ثبت شده توسط کاربر بر اساس نمره‌اش
                 selected_opt = next((o["text"] for o in q["options"] if o["score"] == ans.score), "نامشخص")
                 answers_text_list.append(f"سوال {ans.question_id}: '{q['text']}' -> پاسخ کاربر: {selected_opt} (نمره: {ans.score})")
         answers_summary = "\n".join(answers_text_list)
@@ -544,6 +644,34 @@ async def analyze_personality(payload: AnalysisRequest):
         - درصد علائم عاطفی/خلقی: {calc['affective']}%
         - درصد علائم بدنی/فیزیولوژیکی: {calc['somatic']}%
         
+        پاسخ‌های تفصیلی کاربر:
+        {answers_summary}
+        """
+    elif payload.test_type == "solomon":
+        calc = calculate_solomon_raw(payload.answers)
+        system_instruction = SOLOMON_SYSTEM_INSTRUCTION
+        q_map = {q["id"]: q for q in SOLOMON_QUESTIONS}
+        answers_text_list = []
+        for ans in payload.answers:
+            q = q_map.get(ans.question_id)
+            if q:
+                selected_opt = next((o["text"] for o in SOLOMON_OPTIONS if o["score"] == ans.score), "نامشخص")
+                answers_text_list.append(f"سوال {ans.question_id}: '{q['text']}' -> پاسخ: {selected_opt} (نمره خام: {ans.score})")
+        answers_summary = "\n".join(answers_text_list)
+        prompt_data = f"""
+        کاربر ارزیابی‌شونده: {payload.first_name} {payload.last_name}
+        آزمون انجام شده: سنجش علمی اهمال‌کاری تحصیلی سولومون
+        
+        محاسبات ریاضی پایه (از این محاسبات دقیق در خروجی JSON استفاده کنید):
+        - فیلد test_type: مقدار "solomon" قرار گیرد.
+        - فیلد solomon_level: مقدار "{calc['level']}" قرار گیرد.
+        - فیلد somatic_score (درصد امتحانات): مقدار {calc['exam']} قرار گیرد.
+        - فیلد cognitive_score (درصد تکالیف): مقدار {calc['assignment']} قرار گیرد.
+        - فیلد panic_score (درصد مقالات): مقدار {calc['paper']} قرار گیرد.
+        - فیلد total_score_raw (نمره کل خام): مقدار {calc['total']} قرار گیرد.
+        - سایر فیلدهای عددی مربوط به MBTI یا BAI/BDI که در این آزمون کاربرد ندارند را برابر 0 قرار دهید.
+        - فیلدهای mbti_type، bai_level و bdi_level را خالی یا تهی بگذارید.
+        
         پاسخ‌های تفصیلی کاربر:
         {answers_summary}
         """