Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import json
 import httpx
+from urllib.parse import urlparse
 from fastapi import FastAPI, Request
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
@@ -11,6 +12,13 @@ app = FastAPI()
 class ChatRequest(BaseModel):
     query: str
 
+def get_favicon_url(url: str) -> str:
+    try:
+        domain = urlparse(url).netloc
+        return f"https://www.google.com/s2/favicons?domain={domain}&sz=64"
+    except Exception:
+        return ""
+
 @app.post("/api/search")
 async def search_api(request: ChatRequest, req: Request):
     async def generate():
@@ -25,8 +33,14 @@ async def search_api(request: ChatRequest, req: Request):
         context_parts = []
 
         for idx, r in enumerate(results, 1):
-            sources.append({"title": r.get("title", ""), "url": r.get("url", "")})
-            context_parts.append(f"[{idx}] Title: {r.get('title')}\nURL: {r.get('url')}\nContent: {r.get('content')}\n")
+            url = r.get("url", "")
+            favicon = get_favicon_url(url)
+            sources.append({
+                "title": r.get("title", ""),
+                "url": url,
+                "favicon": favicon
+            })
+            context_parts.append(f"[{idx}] Title: {r.get('title')}\nURL: {url}\nContent: {r.get('content')}\n")
 
         context_str = "\n".join(context_parts)