Update app.py

app.py

@@ -203,7 +203,20 @@ class SearchRequest(BaseModel):
 def build_prompt(user_input: str, web_context: Optional[str] = None) -> str:
     """Construye el prompt con contexto web opcional"""
     if web_context:
-        return f"### Contexto de búsqueda web:\n{web_context}\n\n### Instrucción:\n{user_input}\n\n### Respuesta:\n"
+        # Formato especial para búsqueda web con instrucciones claras
+        return f"""### Instrucción del sistema:
+Eres un asistente que busca información en internet. Debes resumir la información encontrada de forma clara y útil.
+
+### Información encontrada en la web:
+{web_context}
+
+### Pregunta del usuario:
+{user_input}
+
+### Respuesta:
+Hola, encontré esto en la web:
+
+"""
     return f"### Instrucción:\n{user_input}\n\n### Respuesta:\n"
 
 def format_search_results(search_data: Dict) -> str:
@@ -212,15 +225,12 @@ def format_search_results(search_data: Dict) -> str:
     
     if search_data.get("wikipedia") and search_data["wikipedia"].get("success"):
         wiki = search_data["wikipedia"]
-        context_parts.append(f"Wikipedia - {wiki['title']}:\n{wiki['summary']}")
+        context_parts.append(f"[Wikipedia - {wiki['title']}]\n{wiki['summary']}\nFuente: {wiki['url']}")
     
     if search_data.get("web_results"):
-        web_snippets = []
-        for i, result in enumerate(search_data["web_results"][:3], 1):
-            web_snippets.append(f"{i}. {result['title']}: {result['snippet'][:200]}")
-        
-        if web_snippets:
-            context_parts.append("Resultados web:\n" + "\n".join(web_snippets))
+        for i, result in enumerate(search_data["web_results"][:4], 1):
+            snippet = result['snippet'][:300].strip()
+            context_parts.append(f"[Resultado {i}: {result['title']}]\n{snippet}\nFuente: {result['url']}")
     
     return "\n\n".join(context_parts) if context_parts else ""
 
@@ -286,11 +296,24 @@ async def generate(req: PromptRequest):
     # Realizar búsqueda web si está activada
     if req.use_web_search:
         try:
-            search_results = await search_engine.search(user_input)
+            # Extraer la consulta de búsqueda del mensaje del usuario
+            search_query = user_input
+            
+            # Si el mensaje es una pregunta larga, extraer palabras clave
+            if len(user_input.split()) > 8:
+                # Usar las primeras palabras más relevantes
+                words = user_input.lower().split()
+                # Filtrar palabras comunes
+                stop_words = {'qué', 'cuál', 'cómo', 'dónde', 'cuándo', 'por', 'para', 'el', 'la', 'los', 'las', 'un', 'una', 'es', 'sobre', 'me', 'puedes', 'explicar', 'decir', 'información'}
+                keywords = [w for w in words if w not in stop_words][:5]
+                search_query = ' '.join(keywords)
+            
+            search_results = await search_engine.search(search_query)
             web_context = format_search_results(search_results)
             
             if web_context:
-                print(f"🔍 Búsqueda web completada. Contexto agregado: {len(web_context)} caracteres")
+                print(f"🔍 Búsqueda web completada para: '{search_query}'")
+                print(f"   Contexto agregado: {len(web_context)} caracteres")
         except Exception as e:
             print(f"Error en búsqueda web: {e}")