Update utils.py

utils.py

@@ -3,7 +3,6 @@ from dotenv import load_dotenv
 import os
 import pandas as pd
 
-# Load .env variables
 load_dotenv()
 
 def summarize_dataframe(df: pd.DataFrame, max_rows: int = 30) -> str:
@@ -29,7 +28,6 @@ def query_agent(df: pd.DataFrame, query: str) -> str:
         print("Direct analysis failed:", e)
 
     data_text = summarize_dataframe(df)
-
     prompt = f"""
 You are a data analysis assistant with expertise in statistics and data interpretation.
 Analyze the dataset sample below and answer the user's question in a clear, detailed, and well-explained way.
@@ -44,11 +42,13 @@ Question:
 Answer (with explanation):
 """
 
-    # Use Hugging Face InferenceClient directly
+    # ----------------- Option 2 starts here -----------------
     client = InferenceClient(token=os.environ.get("HUGGINGFACE_API_KEY"))
-    response = client.text_generation(prompt, model="google/gemma-2b-it")
+    # Use a model that supports text-generation (TGI-compatible)
+    response = client.text_generation(prompt, model="mistralai/Mistral-7B-Instruct-v0.3")
 
-    # Extract the generated text from the response
+    # Extract the text
     answer = response[0]["generated_text"] if isinstance(response, list) and "generated_text" in response[0] else str(response)
+    # ----------------- Option 2 ends here -----------------
 
     return answer