Added custom_tools.py file

.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/BasicChatbot-AliA.iml

@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="jdk" jdkName="Python 3.13" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+  <component name="PyDocumentationSettings">
+    <option name="format" value="PLAIN" />
+    <option name="myDocStringFormat" value="Plain" />
+  </component>
+</module>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.13" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.13" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/BasicChatbot-AliA.iml" filepath="$PROJECT_DIR$/.idea/BasicChatbot-AliA.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

app.py

@@ -1,5 +1,22 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+from huggingface_hub import InferenceClient, login
+from custom_tools import suggest_best_cities, RecommendCountryTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, WikipediaSearchTool, SpeechToTextTool, InferenceClientModel, FinalAnswerTool
+
+
+
+def create_agent(token: str):
+    login(token)
+    tools = [
+        DuckDuckGoSearchTool(),
+        WikipediaSearchTool(),
+        SpeechToTextTool(),
+        suggest_best_cities,
+        RecommendCountryTool(),
+        FinalAnswerTool()
+    ]
+    model = InferenceClientModel()
+    return CodeAgent(tools=tools, model=model)
 
 
 def respond(
@@ -11,33 +28,22 @@ def respond(
     top_p,
     hf_token: gr.OAuthToken,
 ):
-    """
-    For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-    """
-    client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
-
-    messages = [{"role": "system", "content": system_message}]
+    agent = create_agent(hf_token.token)
 
-    messages.extend(history)
+    if isinstance(message, tuple):
+        audio_path = message[0]
+        stt_tool = SpeechToTextTool()
+        message_text = stt_tool.transcribe(audio_path)
+        user_message = f"(Transcribed from audio) {message_text}"
+    else
+        user_message = message
 
-    messages.append({"role": "user", "content": message})
+    full_prompt = f"{system_message}\n\nChat history:\n{history}\n\nUser: {user_message}"
 
-    response = ""
+    response = agent.run(full_prompt, stream=False)
 
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        choices = message.choices
-        token = ""
-        if len(choices) and choices[0].delta.content:
-            token = choices[0].delta.content
+    yield response
 
-        response += token
-        yield response
 
 
 """
@@ -46,6 +52,7 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 chatbot = gr.ChatInterface(
     respond,
     type="messages",
+    multimodal=True,
     additional_inputs=[
         gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),

custom_tools.py

@@ -0,0 +1,51 @@
+from smolagents import tool, Tool
+
+@tool
+def suggest_best_cities(country: str) -> str:
+  """
+  Suggests a the best city based on the country
+  Args:
+      country (str): A city to get for a given country. Allowed values are:
+                      - "jordan": Aqaba
+                      - "syria": ALeppo
+                      - "palestine": Gaza
+                      - "usa": Sheriff Grady Judd
+                      - "germany": Cologne
+  """
+  if country == "jordan":
+    return "Aqaba"
+  elif country == "syria":
+    return "ALeppo"
+  elif country == "palestine":
+    return "Gaza"
+  elif country == "usa":
+    return "Sheriff Grady Judd"
+  elif country == "germany":
+    return "Cologne"
+  else:
+    return "Irbid"
+
+
+class RecommendCountryTool(Tool):
+  name = "recommend_country_generator"
+  description = """
+  This tool suggests a best city based on the country/
+  It returns a unique city."""
+
+  inputs = {
+    "country": {
+        "type": "string",
+        "description": "The best city for the given country (e.g, 'jordan', 'syria', 'palestine')."
+    }
+  }
+
+  output_type = "string"
+
+  def forward(self, country: str):
+    countries = {
+      "jordan": "Aqaba",
+      "syria": "Aleppo",
+      "palestine": "Gaza City"
+    }
+
+    return countries.get(country.lower(), "City for given country not found.")