update

app.py

@@ -2,12 +2,13 @@ import inspect
 import json
 import os
 from io import BytesIO
+from typing import Optional
 
 import gradio as gr
 import pandas as pd
 import requests
 import whisper
-from bs4 import BeautifulSoup
+from bs4 import BeautifulSoup, NavigableString, Tag
 from PIL import Image
 from smolagents import (
     CodeAgent,
@@ -190,7 +191,9 @@ def get_wikipedia_page_url_by_year(wikipedia_page_name: str, year: int) -> str:
 
 
 @tool
-def get_wikipedia_section(section_name: str, soup_object: BeautifulSoup):
+def get_wikipedia_section(
+    section_name: str, soup_object: BeautifulSoup
+) -> Tag | NavigableString | None:
     """
     A tool that extracts a specific section from a Wikipedia page using BeautifulSoup.