Update app.py

app.py

@@ -1,60 +1,40 @@
 import gradio as gr
-from smolagents import CodeAgent, DuckDuckGoSearchTool, Tool, LiteLLMModel
-from llama_cpp import Llama
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
 from datetime import datetime
-from huggingface_hub import hf_hub_download
 
-# --- 1. SETUP THE GGUF MODEL ---
-# We use a 3B model that fits well in standard HF Space RAM
-model_path = hf_hub_download(
-    repo_id="bartowski/Llama-3.2-3B-Instruct-GGUF",
-    filename="Llama-3.2-3B-Instruct-Q4_K_M.gguf"
-)
-
-# Initialize the local LLM engine
-llm = Llama(model_path=model_path, n_ctx=2048, verbose=False)
-
-# Custom wrapper to make llama-cpp compatible with smolagents
-class LocalGGUFModel:
-    def __call__(self, messages, stop_sequences=None):
-        # Convert messages to a string prompt
-        prompt = ""
-        for msg in messages:
-            prompt += f"{msg['role']}: {msg['content']}\n"
-        prompt += "assistant: "
-        
-        response = llm(
-            prompt, 
-            stop=["user:", "assistant:"] + (stop_sequences or []),
-            max_tokens=500
-        )
-        return response["choices"][0]["text"]
+# 1. Setup the "Brain" 
+# This uses the HF API so you don't have to build llama-cpp. 
+# It's faster and handles tools much better.
+model = HfApiModel(model_id="meta-llama/Llama-3.2-3B-Instruct")
 
-# --- 2. DEFINE CUSTOM TOOLS ---
-def get_current_time_tool():
-    """Returns the current date and time."""
-    return f"Today is {datetime.now().strftime('%A, %B %d, %Y')}. The current time is {datetime.now().strftime('%H:%M:%S')}."
+# 2. Define Custom Tools
+def get_current_info():
+    """A tool that fetches the exact current date and time."""
+    now = datetime.now()
+    return f"Current Date: {now.strftime('%A, %B %d, %Y')}\nTime: {now.strftime('%H:%M:%S')}"
 
-# --- 3. INITIALIZE THE AGENT ---
-# DuckDuckGoSearchTool handles "visiting websites" and "news"
+# 3. Create the Agent
 agent = CodeAgent(
-    tools=[DuckDuckGoSearchTool(), get_current_time_tool],
-    model=LocalGGUFModel(),
+    tools=[
+        DuckDuckGoSearchTool(), # For News and Website visiting
+        get_current_info        # For Current Date
+    ],
+    model=model,
     add_base_tools=True
 )
 
-# --- 4. GRADIO INTERFACE ---
-def chat_fn(history, message):
-    # Process the user request through the agent
-    response = agent.run(message)
-    history.append({"role": "user", "content": message})
-    history.append({"role": "assistant", "content": str(response)})
-    return history, ""
-
-with gr.Blocks() as demo:
-    gr.Markdown("# 🤖 GGUF Smart Agent")
-    chatbot = gr.Chatbot(type="messages")
-    msg = gr.Textbox(placeholder="Ask me for news or the date...")
-    msg.submit(chat_fn, [chatbot, msg], [chatbot, msg])
+# 4. Gradio Interface Setup
+def interact(message, history):
+    # This runs the agentic logic: Search -> Think -> Answer
+    result = agent.run(message)
+    return str(result)
+
+demo = gr.ChatInterface(
+    fn=interact,
+    title="🚀 Super-Smart AI Agent",
+    description="I can visit websites, check the news, and I know exactly what day it is.",
+    examples=["What's the latest news in tech today?", "What is the date today?", "Visit google.com and tell me what's trending."]
+)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()