Update my_functions.py

my_functions.py

@@ -1,6 +1,7 @@
 import yfinance as yf
+# youtube-search-python 추가 임포트
+from youtubesearchpython import VideosSearch
 
-# 예시 제품 카탈로그
 product_catalog = {
     "807ZPKBL9V": "SuperWidget",
     "1234567890": "MegaGadget"
@@ -22,27 +23,28 @@ def get_stock_price(ticker: str) -> float:
         return data['Close'].iloc[-1]
     return float('nan')
 
-# ==============================
-# (New) YouTube 동영상 검색 함수
-# ==============================
 def search_youtube(query: str) -> str:
     """
-    Searches YouTube for the given query and returns a short list of video titles/links.
-    Actual implementation could use YouTube Data API or 'youtube-search-python'.
-    Here we mock the result for demonstration.
+    Searches YouTube for the given query and returns top 3 video titles & links.
     """
-    # 실전 예시 (youtube-search-python 사용 시):
-    # from youtubesearchpython import VideosSearch
-    # videosSearch = VideosSearch(query, limit=3)
-    # results = videosSearch.result()['result']
-    # formatted = []
-    # for item in results:
-    #     title = item['title']
-    #     link = item['link']
-    #     formatted.append(f"{title} - {link}")
-    # return "\n".join(formatted)
+    # youtube-search-python 이용
+    try:
+        videosSearch = VideosSearch(query, limit=3)
+        results = videosSearch.result()["result"]  # 실제 검색 결과
+        
+        if not results:
+            return "No videos found for your query."
 
-    # -- 데모용 더미 결과 --
-    return (
-        "[Demo Video](https://www.youtube.com/watch?v=JzjR5LGOvqg)\n"
-    )
+        # 결과에서 타이틀과 링크를 추출
+        formatted_list = []
+        for idx, item in enumerate(results, start=1):
+            title = item.get("title", "No Title")
+            link = item.get("link", "#")
+            formatted_list.append(f"{idx}) [{title}]({link})")
+
+        # 여러 줄로 묶어서 반환
+        return "\n".join(formatted_list)
+
+    except Exception as e:
+        # 어떤 에러든 발생 시 안전하게 메시지 반환
+        return f"Error searching YouTube: {str(e)}"