Update app.py

app.py

@@ -1,29 +1,26 @@
-import os
 import gradio as gr
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
+from smolagents.agents import CodeAgent
+from smolagents.tools import CalculatorTool
+from smolagents.models import HfApiModel
 
-# Optional: if you add HF_TOKEN in secrets, it can use it
-os.environ["HF_TOKEN"] = os.getenv("HF_TOKEN")
-
-# Build the agent
+# Build agent
 agent = CodeAgent(
-    tools=[DuckDuckGoSearchTool()],
-    model=HfApiModel(model="HuggingFaceH4/zephyr-7b-beta")
+    tools=[CalculatorTool()],
+    model=HfApiModel(model="HuggingFaceH4/zephyr-7b-alpha")  # Small model
 )
 
-# Simple app function
-def who_is_bot(person_name):
-    prompt = f"Who is {person_name}? Give a short 1-2 sentence summary."
-    result = agent.run(prompt)
+# Function
+def ask_bot(question):
+    result = agent.run(question)
     return result.strip()
 
 # Gradio UI
 iface = gr.Interface(
-    fn=who_is_bot,
-    inputs=gr.Textbox(label="Person Name"),
-    outputs=gr.Textbox(label="Summary"),
-    title="Who Is Bot",
-    description="Enter a name — get a 1-2 sentence summary."
+    fn=ask_bot,
+    inputs=gr.Textbox(label="Ask a math question"),
+    outputs=gr.Textbox(label="Bot Answer"),
+    title="Simple Math Helper Bot",
+    description="Ask math or unit conversion questions — runs offline on CPU Space."
 )
 
 if __name__ == "__main__":