Update app.py

app.py

@@ -7,27 +7,67 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
+# 검색 tool
+
 @tool
-def search_articles(query: str) -> str:
+def search_news(topic: str) -> str:
     """
-    DuckDuckGo 검색을 사용하여 기사를 검색하고 결과를 반환합니다.
-
+    주어진 주제에 대한 최신 뉴스를 검색하는 도구입니다.
     Args:
-        query: 검색어.
+        topic: 검색할 뉴스 주제
     Returns:
-        검색 결과 문자열.
+        검색된 뉴스 정보
     """
     try:
-        results = ddg(query, max_results=5) # max_results는 검색 결과의 수를 지정합니다.
-        if results:
-            formatted_results = "\n".join([f"Title: {r['title']}\nLink: {r['href']}\nSnippet: {r['body']}" for r in results])
-            return formatted_results
-        else:
-            return "검색 결과가 없습니다."
+        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}")
     except Exception as e:
-        return f"오류 발생: {str(e)}"
+        traceback.print_exc()
+        return f"검색 중 오류 발생: {str(e)}"
+
+# 기자 tool
+@tool
+def write_article(topic: str, search_results: str) -> str:
+    """
+    검색 결과를 바탕으로 새로운 기사를 작성하는 도구입니다.
+    Args:
+        topic: 기사 주제
+        search_results: 검색된 뉴스 정보
+    Returns:
+        작성된 기사
+    """
+    try:
+        system_prompt = """
+        당신은 전문 기자입니다. 주어진 주제에 대해 검색된 정보를 바탕으로 새로운 기사를 한국어로 작성해야 합니다.
+        다음 형식으로 기사를 작성해주세요:
+
+        1. 제목 (흥미롭고 눈에 띄는 제목)
+        2. 요약 (핵심 내용을 2-3문장으로 요약)
+        3. 본문 (상세한 내용을 단락으로 구분하여 작성)
+        4. 결론 (기사의 의의나 향후 전망)
+
+        기사는 객관적이고 사실에 기반하여 작성해야 하며, 검색된 정보를 재구성하여 새로운 시각으로 작성해주세요.
+        """
+        
+        prompt = f"""
+        다음 주제와 검색 결과를 바탕으로 새로운 기사를 한국어로 작성해주세요:
 
+        주제: {topic}
+
+        검색 결과:
+        {search_results}
+
+        위 정보를 바탕으로 새롭고 통찰력 있는 기사를 작성해주세요.
+        """
+        return model.generate(system_prompt + prompt)
+    except Exception as e:
+        traceback.print_exc()
+        return f"기사 작성 중 오류 발생: {str(e)}"
 
 
 final_answer = FinalAnswerTool()
@@ -39,15 +79,15 @@ 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_articles,final_answer], ## add your tools here (don't remove final answer)
+    tools=[search_news, write_article, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,