Update volume_agent.py

volume_agent.py

@@ -1,9 +1,25 @@
 class VolumeSurgeAgent:
+    """
+    Detects volume surges compared to recent average volume.
+    Returns a structured object with surge status and ratio.
+    """
+
     def run(self, df, threshold=1.8):
+        if len(df) < 21:
+            return {
+                "volume_surge": False,
+                "ratio": 0,
+                "status": "⚠️ Insufficient data"
+            }
+
         recent_avg = df['Volume'].iloc[-20:-1].mean()
         latest_volume = df['Volume'].iloc[-1]
+        ratio = latest_volume / recent_avg
+
+        surge = ratio > threshold
 
-        if latest_volume > threshold * recent_avg:
-            return {"volume_surge": True, "ratio": round(latest_volume / recent_avg, 2)}
-        else:
-            return {"volume_surge": False, "ratio": round(latest_volume / recent_avg, 2)}
+        return {
+            "volume_surge": surge,
+            "ratio": round(ratio, 2),
+            "status": "✅ Spike" if surge else "Normal"
+        }