| |
| from transformers import pipeline |
|
|
| MBTI_MODEL = "f3nsmart/MBTIclassifier" |
| mbti_pipe = pipeline("text-classification", model=MBTI_MODEL, return_all_scores=True) |
|
|
| def analyze_mbti(user_text: str): |
| """Пошаговый MBTI-анализ (генератор для стриминга).""" |
| yield "⏳ Analyzing personality traits..." |
| try: |
| res = mbti_pipe(user_text)[0] |
| res_sorted = sorted(res, key=lambda x: x["score"], reverse=True) |
| mbti_text = "\n".join([f"{r['label']} → {r['score']:.3f}" for r in res_sorted[:3]]) |
| yield mbti_text |
| except Exception as e: |
| yield f"❌ Error: {e}" |
|
|
| def compute_dominant_axis(results): |
| """ |
| Преобразует оценки классификатора в MBTI-код. |
| """ |
| axes = { |
| "IE": ("Introversion", "Extroversion"), |
| "SN": ("Sensing", "Intuition"), |
| "TF": ("Thinking", "Feeling"), |
| "JP": ("Judging", "Perceiving") |
| } |
| mbti_code = "" |
| for pair in axes.values(): |
| left, right = pair |
| left_score = next((r["score"] for r in results if r["label"] == left), 0) |
| right_score = next((r["score"] for r in results if r["label"] == right), 0) |
| mbti_code += left[0] if left_score >= right_score else right[0] |
| return mbti_code |
|
|
