Update app.py

app.py

@@ -1,22 +1,49 @@
 import gradio as gr
 from agent import MyHuggingFaceAgent
 
-agent_instance = MyHuggingFaceAgent()
+print("=" * 50)
+print("🤖 AI Agent Framework Demo - Unit 2: smolagents")
+print("=" * 50)
+
+# Initialize agent with mock (for demonstration)
+agent = MyHuggingFaceAgent()
 
 def chat_response(message, history):
-    return agent_instance.run(message)
+    """Handle chat messages with the agent."""
+    try:
+        response = agent.run(message)
+        return response
+    except Exception as e:
+        return f"❌ **System Error:** {str(e)}\n\nPlease try a simpler query or contact support."
 
+# Create Gradio interface
 demo = gr.ChatInterface(
     fn=chat_response,
-    title="🚀 Final Project: Agentic RAG Assistant",
-    description="This agent uses **smolagents** and Agentic RAG to answer questions about the HF course.",
+    title="🤖 Unit 2: AI Agent Framework (smolagents)",
+    description="""This demonstrates the **smolagents framework** from Hugging Face.
+**Features:**
+• Tool-calling agent architecture
+• Multiple specialized tools
+• Framework abstraction layer
+**Available Tools:**
+1. 🧮 Calculator
+2. 🌐 Web Search
+3. 📊 Dataset Info
+4. 🌍 Translator
+""",
     examples=[
-        ["How many bonus marks can I get?"],
-        ["Search for the latest news on AI agents"],
-        ["What is 15% of 450?"],
-        ["How do I submit the final project?"]
-    ]
+        ["calculate 15 * 3 + 10"],
+        ["search for latest AI developments"],
+        ["tell me about the mnist dataset"],
+        ["translate hello to French"],
+        ["what is 45 + 67?"],
+        ["find information about machine learning"],
+    ],
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        share=False,
+        server_name="0.0.0.0",
+        server_port=7860
+    )