Update app.py

app.py

@@ -7,7 +7,7 @@ from typing import List
 
 from langchain.text_splitter import CharacterTextSplitter
 from tools.final_answer import FinalAnswerTool
-from smolagents import CodeAgent, DuckDuckGoSearchTool, load_tool, tool, from_config
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 from Gradio_UI import GradioUI
 
 # === TOOLS ===
@@ -22,7 +22,6 @@ def web_search(query: str) -> str:
     results = search_tool(query)
     return "\n".join(results)
 
-
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """Fetches the current local time in a specified timezone.
@@ -36,58 +35,65 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-
 @tool
 def visit_webpage(url: str) -> str:
     """Fetches raw HTML content of a web page.
     Args:
-        url: The url of the webpage
+        url: The url of the webpage.
     """
     try:
         response = requests.get(url, timeout=5)
-        return response.text[:5000]
+        return response.text[:5000]  # Limit length
     except Exception as e:
         return f"[ERROR fetching {url}]: {str(e)}"
 
-
 @tool
 def text_splitter(text: str) -> List[str]:
     """Splits text into chunks using LangChain's CharacterTextSplitter.
     Args:
         text: A string of text to split.
     """
-    c_splitter = CharacterTextSplitter(chunk_size=450, chunk_overlap=10)
-    return c_splitter.split_text(text)
+    splitter = CharacterTextSplitter(chunk_size=450, chunk_overlap=10)
+    return splitter.split_text(text)
 
 # === FINAL ANSWER TOOL ===
-
 final_answer = FinalAnswerTool()
 
-# === LOAD MODEL FROM agent.json + INJECT TOKEN SECURELY ===
+# === LOAD PROMPT TEMPLATES ===
+with open("prompts.yaml", "r") as stream:
+    prompt_templates = yaml.safe_load(stream)
 
+# === LOAD agent.json CONFIG ===
 with open("agent.json", "r") as f:
     agent_config = yaml.safe_load(f)
 
-# Inject HF token securely
-agent_config["model"]["data"]["token"] = os.getenv("HF_TOKEN")
-
-# Construct the model safely
-model = from_config(agent_config["model"])
-
-# === IMPORT PROMPT TEMPLATES ===
-
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-
-# === IMPORT EXTRA TOOL FROM HUB ===
+model_config = agent_config["model"]["data"]
+
+# === BUILD MODEL MANUALLY (SAFE FALLBACK) ===
+model = HfApiModel(
+    model_id=model_config["model_id"],
+    token=os.getenv("HF_TOKEN"),  # Injected securely from Space secret
+    temperature=model_config.get("temperature", 0.5),
+    max_tokens=model_config.get("max_tokens", 2048),
+    last_input_token_count=model_config.get("last_input_token_count", 0),
+    last_output_token_count=model_config.get("last_output_token_count", 0),
+    custom_role_conversions=model_config.get("custom_role_conversions", None),
+)
 
+# === IMPORT TOOL FROM HUB ===
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# === AGENT ===
-
+# === BUILD AGENT ===
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, web_search, get_current_time_in_timezone, visit_webpage, text_splitter, image_generation_tool],
+    tools=[
+        final_answer,
+        web_search,
+        get_current_time_in_timezone,
+        visit_webpage,
+        text_splitter,
+        image_generation_tool
+    ],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -98,5 +104,4 @@ agent = CodeAgent(
 )
 
 # === LAUNCH UI ===
-
 GradioUI(agent).launch()