From ce8127cadbdfcd8c1cbc151f0b7cb9552b84e846 Mon Sep 17 00:00:00 2001
From: NY <nnn@qq.com>
Date: Fri, 14 Mar 2025 15:16:53 +0800
Subject: [PATCH] +probability(unfinished)

---
 pkg/dash/func/info_func.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/pkg/dash/func/info_func.py b/pkg/dash/func/info_func.py
index 7e9f40a..d397c7d 100644
--- a/pkg/dash/func/info_func.py
+++ b/pkg/dash/func/info_func.py
@@ -210,9 +210,21 @@ def calculate_tweet_probability(tweet_count, days_to_next_friday, prob_start, pr
 
     a = prob_start - tweet_count
     b = prob_end - tweet_count
+
+    if tweet_count > prob_end:
+        return "0.0000 - 0.0000"
+
     if a < 0:
-        return "0.0000 - 0.0000" if tweet_count > prob_end else "1.0000 - 1.0000"
+        a = 0
+    if b < 0:
+        return "0.0000 - 0.0000"
 
     prob_low = norm.cdf((b - mu_low) / sigma_low) - norm.cdf((a - mu_low) / sigma_low)
     prob_high = norm.cdf((b - mu_high) / sigma_high) - norm.cdf((a - mu_high) / sigma_high)
-    return f"{prob_low:.4f} - {prob_high:.4f}"
\ No newline at end of file
+
+    prob_low = max(0.0, min(1.0, prob_low))
+    prob_high = max(0.0, min(1.0, prob_high))
+
+    prob_min = min(prob_low, prob_high)
+    prob_max = max(prob_low, prob_high)
+    return f"{prob_min:.4f} - {prob_max:.4f}"
\ No newline at end of file