Upload 2 files

agent (2).py

@@ -0,0 +1,10 @@
+from smol_agent import SmolAgent
+
+# Define the agent with a set of useful tools
+tools = ["search", "calculator", "python", "code"]
+agent = SmolAgent(tools=tools)
+
+def run_agent(question: str) -> str:
+    prompt = f"You are a helpful AI assistant. Answer the following question as clearly and accurately as possible:\n\n{question}"
+    response = agent.run(prompt)
+    return response

main.py

@@ -0,0 +1,22 @@
+import gradio as gr
+from agent import run_agent
+
+examples = [
+    "What's the capital of France?",
+    "Solve: 345 * 12",
+    "Write a Python function to reverse a list.",
+    "What's the square root of 169?",
+    "Who discovered penicillin?"
+]
+
+def ask_agent(input_text):
+    return run_agent(input_text)
+
+iface = gr.Interface(fn=ask_agent, 
+                     inputs="text", 
+                     outputs="text", 
+                     title="Final Assignment Agent",
+                     description="Enter your question and the agent will answer it.",
+                     examples=examples)
+
+iface.launch()