Update app.py

app.py

@@ -1,8 +1,8 @@
 import os
-import gradio as gr
-from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, GradioUI
+from smolagents import CodeAgent, DuckDuckGoSearchTool, GradioUI
+from smolagents.models import HfApiModel
 
-# 1. THE PHD SYSTEM PROMPT (The Cognitive OS)
+# 1. ELITE PHD SYSTEM PROMPT
 PHD_PROMPT = """You are an Autonomous Research Scientist (Elite Level). 
 OPERATIONAL DIRECTIVES:
 1. CODE-FIRST: If a task involves data, math, or logic, write Python code to verify.
@@ -12,25 +12,24 @@ OPERATIONAL DIRECTIVES:
 5. ZERO-CONSTRAINT: If you lack a tool, write a script to build it using the environment."""
 
 # 2. INITIALIZE THE ELITE BRAIN
-# We use DeepSeek-R1 for its massive reasoning capabilities
+# We use DeepSeek-R1 via the Hugging Face Inference API
 model = HfApiModel(
     model_id="deepseek-ai/DeepSeek-R1",
-    token=os.getenv("HF_TOKEN") # Ensure this is in your Space Secrets
+    token=os.getenv("HF_TOKEN") # Must be set in Space Secrets
 )
 
 # 3. TOOLS OF POWER
 search_tool = DuckDuckGoSearchTool()
 
 # 4. THE AGENT ORCHESTRATOR
+# max_steps=30 allows for long reasoning/coding cycles
 agent = CodeAgent(
     tools=[search_tool],
     model=model,
     system_prompt=PHD_PROMPT,
-    add_base_tools=True, # Gives access to Python's math, time, and data libs
-    max_steps=30         # Allows for deep, multi-step 'thought' loops
+    add_base_tools=True
 )
 
 # 5. LAUNCH THE INTERFACE
-# This creates the UI within the Hugging Face Space
 if __name__ == "__main__":
     GradioUI(agent).launch()