Update app.py

app.py

@@ -100,14 +100,24 @@ def remove_outliers(df, epsilon):
     return df.loc[mask].reset_index(drop=True)
 
 # Advanced features
-def calculate_rsi(d, window=14):
-    delta = d.diff()
-    gain = np.where(delta > 0, delta, 0)
-    loss = np.where(delta < 0, -delta, 0)
-    avg_gain = pd.Series(gain).rolling(window=window).mean()
-    avg_loss = pd.Series(loss).rolling(window=window).mean()
+def calculate_rsi(series, window=14):
+    # 1) compute deltas
+    delta = series.diff()
+
+    # 2) separate gains and losses as Series
+    gain = delta.clip(lower=0)
+    loss = -delta.clip(upper=0)
+
+    # 3) rolling averages
+    avg_gain = gain.rolling(window=window).mean()
+    avg_loss = loss.rolling(window=window).mean()
+
+    # 4) compute RS and RSI
     rs = avg_gain / avg_loss
-    return 100 - (100 / (1 + rs))
+    rsi = 100 - (100 / (1 + rs))
+
+    return rsi
+
 
 def generate_advanced_features(d, other_data=None):
     d = d.copy()