Update app.py

app.py

@@ -44,7 +44,21 @@ from smolagents.agent_types import AgentText, AgentImage, AgentAudio
 from smolagents.gradio_ui import pull_messages_from_step, handle_agent_output_types
 
 from smolagents import Tool
+from huggingface_hub import InferenceClient
+def hf_chat(api_key, model, text):
+    client = InferenceClient(api_key=api_key)
+    messages = [
+        {
+            "role": "user",
+            "content": text,
+        }
+    ]
+
+    stream = client.chat.completions.create(
+        model=model, messages=messages, max_tokens=6000, stream=False
+    )
 
+    return stream.choices[0].message.content
 
 class GoogleSearchTool(Tool):
     name = "web_search"