Update app.py

app.py

@@ -2,6 +2,7 @@ import spaces
 import json
 import subprocess
 import os
+import requests  # ← Brave Search API 호출 위해 추가
 from llama_cpp import Llama
 from llama_cpp_agent import LlamaCppAgent, MessagesFormatterType
 from llama_cpp_agent.providers import LlamaCppPythonProvider
@@ -10,6 +11,52 @@ from llama_cpp_agent.chat_history.messages import Roles
 import gradio as gr
 from huggingface_hub import hf_hub_download
 
+##############################################################################
+# Brave Web Search 연동용 추가 코드
+##############################################################################
+SERPHOUSE_API_KEY = os.getenv("SERPHOUSE_API_KEY", "")
+
+def do_web_search(query: str) -> str:
+    """
+    Brave Web Search API를 이용하여 query 검색 후,
+    최대 10개 검색 결과를 title/url/description 형태로 마크다운으로 요약 반환.
+    """
+    try:
+        url = "https://api.search.brave.com/res/v1/web/search"
+        params = {
+            "q": query,
+            "count": 10,
+            "search_lang": "en"
+        }
+        headers = {
+            "Accept": "application/json",
+            "Accept-Encoding": "gzip",
+            # Brave API 키 ( SERPHOUSE_API_KEY 를 그대로 사용 )
+            "X-Subscription-Token": SERPHOUSE_API_KEY,
+        }
+        response = requests.get(url, headers=headers, params=params, timeout=30)
+        response.raise_for_status()
+        data = response.json()
+        web_data = data.get("web", {})
+        results = web_data.get("results", [])
+
+        if not results:
+            return "No results from Brave Search."
+
+        lines = []
+        lines.append("## Brave Search Results\n")
+        for i, item in enumerate(results, start=1):
+            title = item.get("title", "Untitled")
+            link = item.get("url", "")
+            snippet = item.get("description", "")
+            lines.append(f"**{i}. {title}**\n\n{snippet}\n\n[{link}]({link})\n\n---\n")
+        return "\n".join(lines)
+    except Exception as e:
+        return f"Brave Search Error: {str(e)}"
+
+##############################################################################
+# 이하 원본 코드
+##############################################################################
 llm = None
 llm_model = None
 
@@ -134,6 +181,17 @@ def respond(
     settings.repeat_penalty = repeat_penalty
     settings.stream = True
 
+    # --------------------------------------------------------------------------------------
+    # 여기서 Brave Web Search를 수행하여 그 결과를 system_message에 추가 (기본 적용)
+    # --------------------------------------------------------------------------------------
+    # 1) 유저 메시지에 대해 Brave 검색
+    search_results = do_web_search(message)
+    
+    # 2) system_message 끝에 검색 결과 추가
+    #    필요에 따라 안내 문구도 덧붙일 수 있음
+    agent.system_prompt += f"\n\n[Brave Search Results for '{message}']\n{search_results}\n"
+    # --------------------------------------------------------------------------------------
+
     messages = BasicChatHistory()
 
     # history의 각 항목이 dict 형식으로 {'user': <user_message>, 'assistant': <assistant_message>} 형태라고 가정