Update app.py

app.py

@@ -1,11 +1,22 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
 import yaml
+import threading
+import logging
 from tools.final_answer import FinalAnswerTool
+from tools.web_search import DuckDuckGoSearchTool
 
 from Gradio_UI import GradioUI
+from telegram_bot import start_telegram_bot
+
+# Setup logging
+logging.basicConfig(
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    level=logging.INFO
+)
+logger = logging.getLogger(__name__)
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -53,9 +64,12 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
+# Initialize the web search tool
+web_search = DuckDuckGoSearchTool()
+
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[web_search, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -65,5 +79,16 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
+# Start Telegram bot in a separate daemon thread
+logger.info("Starting Telegram bot in background thread...")
+telegram_thread = threading.Thread(
+    target=start_telegram_bot,
+    args=(agent,),
+    daemon=True  # Daemon thread will exit when main program exits
+)
+telegram_thread.start()
+logger.info("Telegram bot thread started!")
 
+# Launch Gradio UI in main thread (this blocks)
+logger.info("Starting Gradio UI...")
 GradioUI(agent).launch()