Update app.py

app.py

@@ -53,71 +53,67 @@ def analyze_messages(input_text):
     with torch.no_grad():
         outputs = model(**inputs)
 
-    # Squeeze out batch dimension: shape should be [17]
+    # Squeeze out batch dimension
     logits = outputs.logits.squeeze(0)
 
-    # Convert logits to probabilities
+    # Convert to probabilities
     scores = torch.sigmoid(logits).numpy()
     print("Scores:", scores)
     print("Danger Scores:", scores[14:])  # suicidal, physical, extreme
 
-    # Debug printing (remove once you're confident everything works)
-    print("Scores:", scores)
-
     pattern_count = 0
     danger_flag_count = 0
 
-for i, (label, score) in enumerate(zip(LABELS, scores)):
-    if score > THRESHOLDS[label]:
-        if i < 14:
-            pattern_count += 1
-        else:
-            danger_flag_count += 1
-    # (Optional) Print label-by-label for debugging
+    for i, (label, score) in enumerate(zip(LABELS, scores)):
+        if score > THRESHOLDS[label]:
+            if i < 14:
+                pattern_count += 1
+            else:
+                danger_flag_count += 1
+
+    # Optional debug print
     for i, s in enumerate(scores):
         print(LABELS[i], "=", round(s, 3))
-    
-   
+
     danger_assessment = (
-    "High" if danger_flag_count >= 2 else
-    "Moderate" if danger_flag_count == 1 else
-    "Low"
-)
-    # Customize resource links based on Danger Assessment Score (with additional niche support)
+        "High" if danger_flag_count >= 2 else
+        "Moderate" if danger_flag_count == 1 else
+        "Low"
+    )
+
+    # Set resources
     if danger_assessment == "High":
         resources = (
             "**Immediate Help:** If you are in immediate danger, please call 911.\n\n"
-            "**Crisis Support:** National DV Hotline – Safety Planning: [thehotline.org/plan-for-safety](https://www.thehotline.org/plan-for-safety/)\n"
-            "**Legal Assistance:** WomensLaw – Legal Help for Survivors: [womenslaw.org](https://www.womenslaw.org/)\n"
-            "**Specialized Support:** For LGBTQ+, immigrants, and neurodivergent survivors, please consult local specialized services or visit RAINN: [rainn.org](https://www.rainn.org/)"
+            "**Crisis Support:** National DV Hotline – Safety Planning: [thehotline.org/plan-for-safety](https://www.thehotline.org/plan-for-safety)\n"
+            "**Legal Assistance:** WomensLaw – Legal Help for Survivors: [womenslaw.org](https://www.womenslaw.org)\n"
+            "**Specialized Support:** For LGBTQ+, immigrants, and neurodivergent survivors, please consult local services."
         )
     elif danger_assessment == "Moderate":
         resources = (
-            "**Safety Planning:** The Hotline – What Is Emotional Abuse?: [thehotline.org/resources](https://www.thehotline.org/resources/what-is-emotional-abuse/)\n"
-            "**Relationship Health:** One Love Foundation – Digital Relationship Health: [joinonelove.org](https://www.joinonelove.org/)\n"
-            "**Support Chat:** National Domestic Violence Hotline Chat: [thehotline.org](https://www.thehotline.org/)\n"
+            "**Safety Planning:** The Hotline – What Is Emotional Abuse?: [thehotline.org/resources](https://www.thehotline.org/resources)\n"
+            "**Relationship Health:** One Love Foundation – Digital Relationship Health: [joinonelove.org](https://www.joinonelove.org)\n"
+            "**Support Chat:** National Domestic Violence Hotline Chat: [thehotline.org](https://www.thehotline.org)\n"
             "**Specialized Groups:** Look for support groups tailored for LGBTQ+, immigrant, and neurodivergent communities."
         )
-    else:  # Low risk
+    else:
         resources = (
-            "**Educational Resources:** Love Is Respect – Healthy Relationships: [loveisrespect.org](https://www.loveisrespect.org/)\n"
-            "**Therapy Finder:** Psychology Today – Find a Therapist: [psychologytoday.com](https://www.psychologytoday.com/us/therapists)\n"
-            "**Relationship Tools:** Relate – Relationship Health Tools: [relate.org.uk](https://www.relate.org.uk/)\n"
+            "**Educational Resources:** Love Is Respect – Healthy Relationships: [loveisrespect.org](https://www.loveisrespect.org)\n"
+            "**Therapy Finder:** Psychology Today – Find a Therapist: [psychologytoday.com](https://www.psychologytoday.com)\n"
+            "**Relationship Tools:** Relate – Relationship Health Tools: [relate.org.uk](https://www.relate.org.uk)\n"
             "**Community Support:** Consider community-based and online support groups, especially those focused on LGBTQ+, immigrant, and neurodivergent survivors."
         )
-    
-   
-    # Prepare the output result with just pattern count and dynamic resources
+
+    # Output
     result_md = (
         f"**Abuse Pattern Count:** {pattern_count}\n\n"
         f"**Support Resources:**\n{resources}"
     )
- 
-    # Save the result to a temporary text file for download
+
     with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w") as f:
         f.write(result_md)
         report_path = f.name
-    
+
     return result_md, report_path
 
 # Build the Gradio interface