Update app.py

app.py

@@ -1,93 +1,72 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool, LiteLLMModel
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import traceback
 
 from Gradio_UI import GradioUI
 
-# 검색 tool
-
 @tool
-def search_news(topic: str) -> str:
-    """
-    주어진 주제에 대한 최신 뉴스를 검색하는 도구입니다.
+def nse_stock_price_tool(stock_ticker:str)-> str: 
+    """A tool that is used to retrieve the latest closing price of a stock based on the stock ticker from the Indian Stock Exchange (NSE).
+    example: 'RELIANCE.NS'
     Args:
-        topic: 검색할 뉴스 주제
+        stock_ticker: The stock ticker/symbol for the stock listed in NSE. The exact string input needs to be provided to the function.
     Returns:
-        검색된 뉴스 정보
+        result_string: A string containing the stock_ticker along with the amount in Rupees.
+        example: "RELIANCE NS: 1237.22 Rupees."
     """
     try:
-        search_agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool()],
-            model=model,
-            additional_authorized_imports=["requests", "bs4"]
-        )
-        search_query = f"최신 뉴스 {topic}"
-        return search_agent.run(f"Search for: {search_query}")
+        import math
+        import yfinance as yf
+        stock_symbol = stock_ticker
+        data = yf.download(tickers=stock_symbol, period='1d', interval='1m')
+        if len(data):
+            latest_data = data.tail(1)
+            latest_price = latest_data['Close'].values[0]   
+            return (f"{stock_symbol}: {round(latest_price, 2)} Rupees.")
+        else:
+            return "INVALID TICKER/STOCK"
     except Exception as e:
         traceback.print_exc()
-        return f"검색 중 오류 발생: {str(e)}"
+        return "INVALID REQUEST"
 
-# 기자 tool
 @tool
-def write_article(topic: str, search_results: str) -> str:
-    """
-    검색 결과를 바탕으로 새로운 기사를 작성하는 도구입니다.
+def get_current_time_in_timezone(timezone: str) -> str:
+    """A tool that fetches the current local time in a specified timezone.
     Args:
-        topic: 기사 주제
-        search_results: 검색된 뉴스 정보
-    Returns:
-        작성된 기사
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
     """
     try:
-        system_prompt = """
-        당신은 전문 기자입니다. 주어진 주제에 대해 검색된 정보를 바탕으로 새로운 기사를 한국어로 작성해야 합니다.
-        다음 형식으로 기사를 작성해주세요:
-
-        1. 제목 (흥미롭고 눈에 띄는 제목)
-        2. 요약 (핵심 내용을 2-3문장으로 요약)
-        3. 본문 (상세한 내용을 단락으로 구분하여 작성)
-        4. 결론 (기사의 의의나 향후 전망)
-
-        기사는 객관적이고 사실에 기반하여 작성해야 하며, 검색된 정보를 재구성하여 새로운 시각으로 작성해주세요.
-        """
-        
-        prompt = f"""
-        다음 주제와 검색 결과를 바탕으로 새로운 기사를 한국어로 작성해주세요:
-
-        주제: {topic}
-
-        검색 결과:
-        {search_results}
-
-        위 정보를 바탕으로 새롭고 통찰력 있는 기사를 작성해주세요.
-        """
-        return model.generate(system_prompt + prompt)
+        # Create timezone object
+        tz = pytz.timezone(timezone)
+        # Get current time in that timezone
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
-        traceback.print_exc()
-        return f"기사 작성 중 오류 발생: {str(e)}"
+        return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
 final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 custom_role_conversions=None,
 )
 
 
-# # Import tool from Hub
-# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[search_news, write_article, final_answer], ## add your tools here (don't remove final answer)
+    tools=[nse_stock_price_tool, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,