Update app.py

app.py

@@ -15,9 +15,7 @@ import os
 
 
 # 🔑 Set your HF API key
-def set_hf_token(token):
-    os.environ["HF_TOKEN"] = token.strip()
-    return "✅ Token saved successfully! You can now upload your CSV file."
+agent = None
 
 
 # ————————————————————————————————
@@ -383,19 +381,40 @@ def AutoML(data: dict, task_hint: str = None) -> dict:
 # 🧠 Initialize the AI Agent
 # ————————————————————————————————
 
-agent = CodeAgent(
-    tools=[LoadData, CleanData, EDA, AutoML],
-    model=InferenceClientModel(
-        model_id="Qwen/Qwen2.5-Coder-1.5B-Instruct",
-        token=os.environ["HF_TOKEN"], 
-        provider="Featherless AI",
-        max_tokens=4048
-    ),
-    additional_authorized_imports=[
-        "pandas", "matplotlib.pyplot", "seaborn", "PIL", "sklearn", "io", "os","joblib","tempfile"
-    ],
-    max_steps=10,
-)
+#agent = CodeAgent(
+    #tools=[LoadData, CleanData, EDA, AutoML],
+    #model=InferenceClientModel(
+        #model_id="Qwen/Qwen2.5-Coder-1.5B-Instruct",
+        #token=os.environ["HF_TOKEN"], 
+        #provider="Featherless AI",
+        #max_tokens=4048
+    #),
+    #additional_authorized_imports=[
+        #"pandas", "matplotlib.pyplot", "seaborn", "PIL", "sklearn", "io", "os","joblib","tempfile"
+    #],
+    #max_steps=10,
+#)
+
+def set_hf_token(token):
+    global agent
+    os.environ["HF_TOKEN"] = token.strip()
+
+    # ✅ Initialize the agent *only now* when token is available
+    agent = CodeAgent(
+        tools=[LoadData, CleanData, EDA, AutoML],
+        model=InferenceClientModel(
+            model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+            token=os.environ["HF_TOKEN"],
+            provider="together",
+            max_tokens=4048
+        ),
+        additional_authorized_imports=[
+            "pandas", "matplotlib.pyplot", "seaborn", "PIL", "sklearn", "io", "os", "joblib", "tempfile"
+        ],
+        max_steps=10,
+    )
+
+    return "✅ Token saved and agent initialized successfully! You can now upload your CSV file."
 
 
 # ————————————————————————————————