Update app.py

app.py

@@ -14,19 +14,20 @@ def search_web(query: str, max_results: int = 3):
     results = []
     with DDGS() as ddg:
         for r in ddg.text(query, max_results=max_results):
-            results.append(f"- {r['title']} ({r['href']})\n{r['body']}")
-    return "\n\n".join(results) if results else "Ничего не найдено."
+            results.append(f"{r['title']}: {r['body']}")
+    return "\n".join(results) if results else "Ничего не найдено."
 
 
 # ==== Основная функция ====
 def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
     client = InferenceClient(token=hf_token.token, model=MODEL_NAME)
 
+    # 1-й прогон: что отвечает модель
     messages = [{"role": "system", "content": SYSTEM_MESSAGE}]
     messages.extend(history)
     messages.append({"role": "user", "content": message})
 
-    response = ""
+    raw_response = ""
     for msg in client.chat_completion(
         messages,
         max_tokens=MAX_TOKENS,
@@ -38,14 +39,38 @@ def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
             continue
         delta = msg.choices[0].delta.content
         if delta:
-            response += delta
-            yield response
+            raw_response += delta
 
-    # Если модель вернула команду поиска
-    if response.strip().startswith("#search"):
-        query = response.replace("#search", "").strip()
+    # === Проверка на команду поиска ===
+    if raw_response.strip().startswith("#search"):
+        query = raw_response.replace("#search", "").strip()
         web_context = search_web(query)
-        yield f"🔎 Поиск по запросу: **{query}**\n\n{web_context}"
+
+        # 2-й прогон: модель формирует финальный ответ с учётом поиска
+        followup_messages = [
+            {"role": "system", "content": SYSTEM_MESSAGE},
+            {"role": "system", "content": f"Web context:\n{web_context}"},
+        ]
+        followup_messages.extend(history)
+        followup_messages.append({"role": "user", "content": message})
+
+        final_response = ""
+        for msg in client.chat_completion(
+            followup_messages,
+            max_tokens=MAX_TOKENS,
+            stream=True,
+            temperature=TEMPERATURE,
+            top_p=TOP_P,
+        ):
+            if not msg.choices:
+                continue
+            delta = msg.choices[0].delta.content
+            if delta:
+                final_response += delta
+                yield final_response
+    else:
+        # Если поиска не нужно — выдаём обычный ответ
+        yield raw_response
 
 
 # ==== Интерфейс ====