Update analytics/ai_assistant.py

analytics/ai_assistant.py

@@ -126,6 +126,8 @@ def _get_inference_client():
             "HUGGINGFACE_API_TOKEN is required for AI assistant. "
             "Set it in your environment before running the app."
         )
+    else:
+        print("Inference set up successful.")
     return InferenceClient(token=HF_TOKEN)
 
 
@@ -141,6 +143,7 @@ def generate_ai_answer(question: str, df: pd.DataFrame, as_of_month: str = None,
     )
 
     client = _get_inference_client()
+    print("Inference called successfully")
     response = client.text_generation(
         model=HF_MODEL_ID, 
         prompt = prompt,
@@ -149,6 +152,7 @@ def generate_ai_answer(question: str, df: pd.DataFrame, as_of_month: str = None,
         top_k= 50,
         top_p= 0.95
     )
+    print(response)
 
     if isinstance(response, dict):
         return response.get("generated_text") or response.get("text") or str(response)