Update app.py

app.py

@@ -1,4 +1,37 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+# 1) Import the real model class
+from smolagents.models import HfApiModel
+
+# 2) Keep a reference to the original __call__
+_original_hfapi_call = HfApiModel.__call__
+
+# 3) Define a safe wrapper that never blows up on None usage
+def _safe_hfapi_call(self, *args, **kwargs):
+    # 3a) Invoke the original method
+    response = _original_hfapi_call(self, *args, **kwargs)
+
+    # 3b) Now defensively pull out token‐usage
+    try:
+        usage = response.usage or type("U", (), {})()
+        # fallback to 0 if either is None or missing
+        self.last_input_token_count  = getattr(usage, "prompt_tokens", 0) or 0
+        self.last_output_token_count = getattr(usage, "completion_tokens", 0) or 0
+    except Exception:
+        self.last_input_token_count  = 0
+        self.last_output_token_count = 0
+
+    # 3c) Return the same ChatMessage the original returned
+    return response.choices[0].message
+
+# 4) Monkey-patch the method in place
+HfApiModel.__call__ = _safe_hfapi_call
+
+# -----------------------------------------------------------------------------
+# Now proceed with your normal imports and setup:
+from smolagents import CodeAgent, DuckDuckGoSearchTool, load_tool, tool
+
+
+
+#from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz