Update app.py

app.py

@@ -89,28 +89,46 @@ from huggingface_hub import login
 from rich.console import Console
 from rich.logging import RichHandler
 from smolagents import FinalAnswerTool  # smolagents
-from smolagents import (CodeAgent, GoogleSearchTool, HfApiModel, LiteLLMModel,
-                        OpenAIServerModel, Tool, TransformersModel)
-from smolagents.agent_types import AgentText
-from smolagents.gradio_ui import (handle_agent_output_types,
-                                  pull_messages_from_step)
+from smolagents import (
+    CodeAgent,
+    GoogleSearchTool,
+    HfApiModel,
+    LiteLLMModel,
+    OpenAIServerModel,
+    Tool,
+    TransformersModel,
+)
+from smolagents.agent_types import AgentText, handle_agent_output_types
+from smolagents.gradio_ui import pull_messages_from_step
 
 # local
-from scripts.finance_tools import (DataVisualizationTool,
-                                   FinancialCalculatorTool, TrendAnalysisTool,
-                                   get_balance_sheet_data, get_cash_flow_data,
-                                   get_company_overview_data,
-                                   get_earnings_data,
-                                   get_income_statement_data,
-                                   get_market_news_sentiment,
-                                   get_stock_quote_data, get_time_series_daily,
-                                   search_symbols)
+from scripts.finance_tools import (
+    DataVisualizationTool,
+    FinancialCalculatorTool,
+    TrendAnalysisTool,
+    get_balance_sheet_data,
+    get_cash_flow_data,
+    get_company_overview_data,
+    get_earnings_data,
+    get_income_statement_data,
+    get_market_news_sentiment,
+    get_stock_quote_data,
+    get_time_series_daily,
+    search_symbols,
+)
 from scripts.flux_lora_tool import FluxLoRATool
 from scripts.text_cleaner_tool import TextCleanerTool
 from scripts.text_inspector_tool import TextInspectorTool
-from scripts.text_web_browser import (ArchiveSearchTool, DownloadTool,
-                                      FinderTool, FindNextTool, PageDownTool,
-                                      PageUpTool, SimpleTextBrowser, VisitTool)
+from scripts.text_web_browser import (
+    ArchiveSearchTool,
+    DownloadTool,
+    FinderTool,
+    FindNextTool,
+    PageDownTool,
+    PageUpTool,
+    SimpleTextBrowser,
+    VisitTool,
+)
 from scripts.time_tools import get_temporal_context
 from scripts.visual_qa import visualizer
 
@@ -135,7 +153,7 @@ AUTHORIZED_IMPORTS = [
     "pubchempy",  # Accessing PubChem chemical database
     "yaml",
     "xml",  # XML processing
-    "yahoo_finance",  # Fetching stock datauv 
+    "yahoo_finance",  # Fetching stock datauv
     "Bio",  # Bioinformatics tools (e.g., sequence analysis)
     "sklearn",  # Scikit-learn for machine learning
     "scipy",  # Scientific computing (stats, optimization)