Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-
+import requests
+import re
 # ==== Константы ====
 MODEL_NAME = "openai/gpt-oss-20b"
 SYSTEM_MESSAGE = "Ты Pok.Bot, ты используешь открытую локальную модель GPT-OSS от OpenAI, не GPT-4 (наверно😆). Тебя создал POKilondron. Используй емодзи 😄."
@@ -9,26 +10,62 @@ TEMPERATURE = 0.7
 TOP_P = 0.95
 
 
+(DuckDuckGo API) ====
+def search_web(query: str, max_results: int = 3):
+    """Ищет информацию в интернете через DuckDuckGo API"""
+    url = f"https://api.duckduckgo.com/?q={query}&format=json"
+    try:
+        r = requests.get(url, timeout=10)
+        r.raise_for_status()
+        data = r.json()
+    except Exception as e:
+        return f"⚠️ Ошибка поиска: {e}"
+
+    results = []
+    if "RelatedTopics" in data:
+        for i, item in enumerate(data["RelatedTopics"][:max_results]):
+            if "Text" in item and "FirstURL" in item:
+                results.append(f"- {item['Text']} ({item['FirstURL']})")
+
+    return "\n".join(results) if results else "Ничего не найдено."
+
+
+# ==== Детектор необходимости поиска ====
+def needs_search(message: str) -> bool:
+    """Решает, нужно ли использовать интернет"""
+    keywords = [
+        r"\bсейчас\b", r"\bсегодня\b", r"\bновости\b",
+        r"\bкурс\b", r"\bпогода\b", r"\bцена\b",
+        r"\b202[3-9]\b",  # годы 2023–2029
+        r"\bпоследн"
+    ]
+    return any(re.search(pat, message.lower()) for pat in keywords)
+
+
+# ==== Основная функция ====
 def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
-    """
-    Chat handler using Hugging Face Inference API
-    """
     client = InferenceClient(token=hf_token.token, model=MODEL_NAME)
 
+    web_context = ""
+    if needs_search(message):
+        web_context = search_web(message)
+
     messages = [{"role": "system", "content": SYSTEM_MESSAGE}]
+    if web_context:
+        messages.append({"role": "system", "content": f"Web context:\n{web_context}"})
+
     messages.extend(history)
     messages.append({"role": "user", "content": message})
 
     response = ""
-
-    for message in client.chat_completion(
+    for msg in client.chat_completion(
         messages,
         max_tokens=MAX_TOKENS,
         stream=True,
         temperature=TEMPERATURE,
         top_p=TOP_P,
     ):
-        choices = message.choices
+        choices = msg.choices
         token = ""
         if len(choices) and choices[0].delta.content:
             token = choices[0].delta.content
@@ -41,7 +78,7 @@ def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
 chatbot = gr.ChatInterface(
     respond,
     type="messages",
-    additional_inputs=[],  # настройки отключены
+    additional_inputs=[],
 )
 
 with gr.Blocks() as demo:
@@ -50,6 +87,4 @@ with gr.Blocks() as demo:
     chatbot.render()
 
 if __name__ == "__main__":
-    demo.launch()
-
-
+    demo.launch()