Update app.py

app.py

@@ -1,80 +1,105 @@
-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 smolagents import CodeAgent, ToolCallingAgent, DuckDuckGoSearchTool, LiteLLMModel, tool
+from typing import Optional
+import os
+from dotenv import load_dotenv
+import gradio as gr
 
-from Gradio_UI import GradioUI
+# .env 파일에서 환경 변수를 로드
+load_dotenv()
+
+# Gemini AI 모델 초기화
+model = LiteLLMModel(
+    model_id="gemini/gemini-2.0-flash-exp",
+    api_key=os.getenv("GOOGLE_API_KEY")
+)
 
 @tool
-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'
+def search_news(topic: str) -> str:
+    """
+    주어진 주제에 대한 최신 뉴스를 검색하는 도구입니다.
     Args:
-        stock_ticker: The stock ticker/symbol for the stock listed in NSE. The exact string input needs to be provided to the function.
+        topic: 검색할 뉴스 주제
     Returns:
-        result_string: A string containing the stock_ticker along with the amount in Rupees.
-        example: "RELIANCE NS: 1237.22 Rupees."
+        검색된 뉴스 정보
     """
-    try:
-        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 "INVALID REQUEST"
+    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}")
 
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
+def write_article(topic: str, search_results: str) -> str:
+    """
+    검색 결과를 바탕으로 새로운 기사를 작성하는 도구입니다.
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        topic: 기사 주제
+        search_results: 검색된 뉴스 정보
+    Returns:
+        작성된 기사
     """
-    try:
-        # 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:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+    NEWS_WRITER_PROMPT = """
+    당신은 전문 기자입니다. 주어진 주제에 대해 검색된 정보를 바탕으로 새로운 기사를 한국어로 작성해야 합니다.
+    다음 형식으로 기사를 작성해주세요:
 
+    1. 제목 (흥미롭고 눈에 띄는 제목)
+    2. 요약 (핵심 내용을 2-3문장으로 요약)
+    3. 본문 (상세한 내용을 단락으로 구분하여 작성)
+    4. 결론 (기사의 의의나 향후 전망)
 
-final_answer = FinalAnswerTool()
-model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',
-custom_role_conversions=None,
-)
+    기사는 객관적이고 사실에 기반하여 작성해야 하며, 검색된 정보를 재구성하여 새로운 시각으로 작성해주세요.
+    """
+    
+    writer_agent = ToolCallingAgent(
+        model=model,
+        system_prompt=NEWS_WRITER_PROMPT
+    )
+    
+    prompt = f"""
+    다음 주제와 검색 결과를 바탕으로 새로운 기사를 한국어로 작성해주세요:
+
+    주제: {topic}
 
+    검색 결과:
+    {search_results}
 
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+    위 정보를 바탕으로 새롭고 통찰력 있는 기사를 한국어로 작성해주세요.
+    """
+    return writer_agent.run(prompt)
 
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
+class NewsWriterGradioUI:
+    def __init__(self):
+        self.agent = CodeAgent(
+            tools=[search_news, write_article],
+            model=model
+        )
     
-agent = CodeAgent(
-    model=model,
-    tools=[nse_stock_price_tool, final_answer], ## add your tools here (don't remove final answer)
-    max_steps=6,
-    verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
-    prompt_templates=prompt_templates
-)
+    def generate_article(self, topic: str) -> str:
+        try:
+            prompt = f"""
+            1. '{topic}' 주제에 대해 search_news 도구를 사용하여 최신 뉴스를 검색하세요.
+            2. 검색된 결과를 바탕으로 write_article 도구를 사용하여 새로운 기사를 작성하세요.
+            """
+            result = self.agent.run(prompt)
+            return result
+        except Exception as e:
+            return f"에러가 발생했습니다: {str(e)}"
+    
+    def launch(self):
+        interface = gr.Interface(
+            fn=self.generate_article,
+            inputs=gr.Textbox(label="뉴스 주제를 입력하세요"),
+            outputs=gr.Textbox(label="생성된 기사"),
+            title="AI 기자",
+            description="주제를 입력하면 관련 뉴스를 검색하고 새로운 기사를 작성합니다."
+        )
+        interface.launch()
 
+def main():
+    ui = NewsWriterGradioUI()
+    ui.launch()
 
-GradioUI(agent).launch()
+if __name__ == "__main__":
+    main()