diff --git a/backend/main.py b/backend/main.py
index cc538e6..96a9418 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -40,10 +40,10 @@ async def chat(request: Request):
 
     async def stream():
         async for token in stream_response(agent, messages):
-            chunk = json.dumps({"type": "text", "delta": token})
+            chunk = json.dumps({"type": "text-delta", "textDelta": token})
             yield f"data: {chunk}\n\n"
         yield "data: [DONE]\n\n"
-
+        
     return StreamingResponse(
         stream(),
         media_type="text/event-stream",
diff --git a/backend/mcp_server/__init__.py b/backend/mcp_server/__init__.py
deleted file mode 100644
index e69de29..0000000
diff --git a/backend/mcp_server/tools/judges.py b/backend/mcp_server/tools/judges.py
index d5ad172..0ecf6ad 100644
--- a/backend/mcp_server/tools/judges.py
+++ b/backend/mcp_server/tools/judges.py
@@ -3,7 +3,6 @@ import httpx
 import json
 from fastmcp import FastMCP
 from typing import Optional
-from pydantic import Field
 
 JUSTICE_API = "https://obcan.justice.sk/pilot/api/ress-isu-service"
 HEADERS = {
@@ -29,7 +28,7 @@ def register_judge_tools(mcp: FastMCP):
     async def judge_autocomplete(
         query: str,
         court_id: Optional[str] = None,
-        limit: int = Field(default=10, description="Maximálny počet výsledkov"),
+        limit: int = 10,
     ) -> str:
         """
         Autocomplete pre mená sudcov — použiť AKO PRVÝ krok
@@ -48,8 +47,8 @@ def register_judge_tools(mcp: FastMCP):
         kraj: Optional[str] = None,
         court_id: Optional[str] = None,
         status: Optional[str] = None,
-        page: int = Field(default=0, description="Číslo stránky (začína od 0)"),
-        size: int = Field(default=20, description="Počet výsledkov na stránku"),
+        page: int = 0,
+        size: int = 20,
     ) -> str:
         """
         Vyhľadávanie sudcov s filtrami.
diff --git a/compose.yaml b/compose.yaml
index 2f5e22e..1fc00b8 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -16,7 +16,7 @@ services:
   backend:
     build: 
       context: .
-      dockerfile: /backend/Dockerfile
+      dockerfile: backend/Dockerfile
     restart: unless-stopped
     ports:
       - "8000:8000"
@@ -40,9 +40,6 @@ services:
       - "4000:4000"
     env_file:
       - backend/.env
-    environment:
-      - GROQ_API_KEY=${GROQ_API_KEY}
-      - GEMINI_API_KEY=${GEMINI_API_KEY}
     volumes:
       - ./backend/config.yaml:/app/config.yaml:ro
     command: ['--config', '/app/config.yaml', '--port', '4000']
@@ -50,7 +47,7 @@ services:
   mcp:
     build:
       context: .
-      dockerfile: /backend/mcp_server/Dockerfile
+      dockerfile: backend/mcp_server/Dockerfile
     restart: unless-stopped
     ports:
       - "8001:8001"
diff --git a/frontend/app/api/chat/route.ts b/frontend/app/api/chat/route.ts
deleted file mode 100644
index 578c895..0000000
--- a/frontend/app/api/chat/route.ts
+++ /dev/null
@@ -1,19 +0,0 @@
-export async function POST(req: Request) {
-  const body = await req.json();
-
-  const response = await fetch("http://backend:8000/api/chat", {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-    },
-    body: JSON.stringify(body),
-  });
-
-  return new Response(response.body, {
-    headers: {
-      "Content-Type": "text/event-stream",
-      "Cache-Control": "no-cache",
-      "X-Accel-Buffering": "no",
-    },
-  });
-}
\ No newline at end of file
diff --git a/frontend/app/globals.css b/frontend/app/globals.css
index 04c459a..d053cc2 100644
--- a/frontend/app/globals.css
+++ b/frontend/app/globals.css
@@ -67,31 +67,7 @@
   --ring: oklch(0.5 0.15 240);
 }
 
-.dark {
-  --background: oklch(0.12 0.04 240);
-  --foreground: oklch(0.95 0.01 240);
-  --card: oklch(0.17 0.05 240);
-  --card-foreground: oklch(0.95 0.01 240);
-  --popover: oklch(0.17 0.05 240);
-  --popover-foreground: oklch(0.95 0.01 240);
-  --primary: oklch(0.6 0.18 240);
-  --primary-foreground: oklch(0.98 0 0);
-  --secondary: oklch(0.2 0.05 240);
-  --secondary-foreground: oklch(0.95 0.01 240);
-  --muted: oklch(0.2 0.05 240);
-  --muted-foreground: oklch(0.65 0.05 240);
-  --accent: oklch(0.22 0.06 240);
-  --accent-foreground: oklch(0.95 0.01 240);
-  --destructive: oklch(0.704 0.191 22.216);
-  --border: oklch(1 0 0 / 8%);
-  --input: oklch(1 0 0 / 12%);
-  --ring: oklch(0.5 0.15 240);
-  --chart-1: oklch(0.488 0.243 264.376);
-  --chart-2: oklch(0.696 0.17 162.48);
-  --chart-3: oklch(0.769 0.188 70.08);
-  --chart-4: oklch(0.627 0.265 303.9);
-  --chart-5: oklch(0.645 0.246 16.439);
-}
+
 
 @layer base {
   * {
diff --git a/frontend/app/layout.tsx b/frontend/app/layout.tsx
index 3813dcc..e73c9c4 100644
--- a/frontend/app/layout.tsx
+++ b/frontend/app/layout.tsx
@@ -14,8 +14,7 @@ const geistMono = Geist_Mono({
 });
 
 export const metadata: Metadata = {
-  title: "assistant-ui starter app",
-  description: "Generated by create-assistant-ui",
+  title: "Legal AI Assistant",
 };
 
 export default function RootLayout({
@@ -24,7 +23,7 @@ export default function RootLayout({
   children: React.ReactNode;
 }>) {
   return (
-    <html lang="en" className="dark">
+    <html lang="en" className="root">
       <body
         className={`${geistSans.variable} ${geistMono.variable} antialiased`}
       >
diff --git a/frontend/components/assistant-ui/header.tsx b/frontend/components/assistant-ui/header.tsx
index c20d319..a514b15 100644
--- a/frontend/components/assistant-ui/header.tsx
+++ b/frontend/components/assistant-ui/header.tsx
@@ -16,10 +16,6 @@ export const ThreadHeader: FC = () => {
         </span>
         <ChevronDownIcon className="size-4 text-muted-foreground" />
       </div>
-
-      <div className="ml-auto p-1.5">
-        <ScaleIcon className="size-5 text-blue-400" />
-      </div>
     </div>
   );
 };
\ No newline at end of file
diff --git a/frontend/components/assistant-ui/thread.tsx b/frontend/components/assistant-ui/thread.tsx
index b218eee..6d970bb 100644
--- a/frontend/components/assistant-ui/thread.tsx
+++ b/frontend/components/assistant-ui/thread.tsx
@@ -57,9 +57,6 @@ export const Thread: FC = () => {
         <ThreadPrimitive.ViewportFooter className="sticky bottom-0 mx-auto mt-auto flex w-full max-w-(--thread-max-width) flex-col gap-4 overflow-visible rounded-t-2xl bg-background pb-4 md:pb-6 pt-2">
           <ThreadScrollToBottom />
           <ThreadComposer />
-          <p className="text-center text-xs text-muted-foreground/50 pb-1">
-            Právny AI Asistent môže robiť chyby. Overte dôležité informácie.
-          </p>
         </ThreadPrimitive.ViewportFooter>
       </ThreadPrimitive.Viewport>
     </ThreadPrimitive.Root>
diff --git a/frontend/components/assistant-ui/welcome.tsx b/frontend/components/assistant-ui/welcome.tsx
index d87f589..c3ce66f 100644
--- a/frontend/components/assistant-ui/welcome.tsx
+++ b/frontend/components/assistant-ui/welcome.tsx
@@ -2,30 +2,22 @@
 
 import { type FC } from "react";
 
-const SUGGESTIONS = [
+const STARTERS = [
   {
-    icon: "🔍",
-    title: "Aké právne dáta",
-    description: "môže agent nájsť?",
-    prompt: "Aké právne dáta môžeš vyhľadať?",
+    icon: "/icons/magnifying-glass.svg",
+    description: "What legal data can the agent find?",
   },
   {
-    icon: "🚫",
-    title: "Čo agent nesmie",
-    description: "robiť alebo používať?",
-    prompt: "Čo nie si oprávnený robiť alebo použiť?",
+    icon: "/icons/ban.svg",
+  description: "What are the agent's limitations?",
   },
   {
-    icon: "🤖",
-    title: "Detaily AI modelu",
-    description: "aký model používaš?",
-    prompt: "Aké sú detaily tvojho AI modelu?",
+    icon: "/icons/hexagon.svg",
+    description: "What are the details of your AI model?",
   },
   {
-    icon: "📊",
-    title: "Zdroje dát",
-    description: "odkiaľ čerpáš informácie?",
-    prompt: "Aké dátové zdroje agent využíva?",
+    icon: "/icons/database.svg",
+    description: "What are your data sources?",
   },
 ];
 
@@ -47,16 +39,22 @@ const ThreadSuggestions: FC = () => {
 
   return (
     <div className="grid w-full grid-cols-2 @lg:grid-cols-4 gap-2 pb-4 px-2">
-      {SUGGESTIONS.map((s) => (
+      {STARTERS.map((s) => (
         <button
-          key={s.title}
-          onClick={() => handleClick(s.prompt)}
-          className="flex flex-col items-start gap-1 rounded-xl border border-blue-500/20 bg-card px-4 py-3 text-left text-sm transition-colors hover:bg-blue-600/10 hover:border-blue-500/40 cursor-pointer"
+          key={s.description}
+          onClick={() => handleClick(s.description)}
+          className="flex flex-row items-center gap-1 rounded-xl border border-blue-500/20 bg-card px-4 py-3 text-left text-sm transition-colors hover:bg-blue-600/10 hover:border-blue-500/40 cursor-pointer"
         >
-          <span className="font-medium text-foreground">
-            {s.icon} {s.title}
+          <div className="size-5 flex-shrink-0">
+            <img
+              src={s.icon}
+              alt=""
+              className="size-full object-contain group-hover:scale-110 transition-transform"
+            />   
+          </div>
+          <span className="text-muted-foreground group-hover:text-foreground transition-colors leading-tight">
+            {s.description}
           </span>
-          <span className="text-muted-foreground text-xs">{s.description}</span>
         </button>
       ))}
     </div>
@@ -65,17 +63,13 @@ const ThreadSuggestions: FC = () => {
 
 export const ThreadWelcome: FC = () => {
   return (
-    <div className="mx-auto my-auto flex w-full max-w-(--thread-max-width) grow flex-col items-center justify-center gap-6">
+    <div className="mx-auto my-auto flex w-full max-w-(--thread-max-width) grow flex-col items-center justify-end gap-6">
       <div className="flex flex-col items-center gap-4">
-        <div className="size-24 rounded-full overflow-hidden ring-2 ring-blue-500/30 shadow-lg shadow-blue-500/20">
+        <div className=" rounded-full overflow-hidden ring-2 ring-blue-500/30 shadow-lg shadow-blue-500/20">
           <img
             src="/logo.png"
             alt="Legal AI"
-            className="size-full object-cover"
-            onError={(e) => {
-              e.currentTarget.style.display = "none";
-              e.currentTarget.parentElement!.innerHTML = `<div class="size-full bg-blue-600/20 flex items-center justify-center text-4xl">⚖️</div>`;
-            }}
+            className="w-50 h-50 object-cover"
           />
         </div>
         <div className="flex flex-col items-center gap-1 text-center">
diff --git a/frontend/package-lock.json b/frontend/package-lock.json
index 09f6508..8684999 100644
--- a/frontend/package-lock.json
+++ b/frontend/package-lock.json
@@ -9,8 +9,9 @@
       "version": "0.1.0",
       "dependencies": {
         "@ai-sdk/openai": "^3.0.52",
-        "@assistant-ui/react": "latest",
+        "@assistant-ui/react": "^0.12.25",
         "@assistant-ui/react-ai-sdk": "latest",
+        "@assistant-ui/react-data-stream": "^0.12.11",
         "@assistant-ui/react-markdown": "latest",
         "@radix-ui/react-avatar": "^1.1.11",
         "@radix-ui/react-collapsible": "^1.1.12",
@@ -227,6 +228,27 @@
         }
       }
     },
+    "node_modules/@assistant-ui/react-data-stream": {
+      "version": "0.12.11",
+      "resolved": "https://registry.npmjs.org/@assistant-ui/react-data-stream/-/react-data-stream-0.12.11.tgz",
+      "integrity": "sha512-DuCV1ILVJsJjl746Ly50jsyt1gUVp43E02Fbq3uau4LU9p3MhUP5CeCV0LLkdS80Hlu9X6xXc34tsL5BCyVzow==",
+      "license": "MIT",
+      "dependencies": {
+        "@ai-sdk/provider": "^3.0.8",
+        "@assistant-ui/core": "^0.1.14",
+        "assistant-cloud": "*",
+        "assistant-stream": "^0.3.11"
+      },
+      "peerDependencies": {
+        "@types/react": "*",
+        "react": "^18 || ^19"
+      },
+      "peerDependenciesMeta": {
+        "@types/react": {
+          "optional": true
+        }
+      }
+    },
     "node_modules/@assistant-ui/react-markdown": {
       "version": "0.12.9",
       "resolved": "https://registry.npmjs.org/@assistant-ui/react-markdown/-/react-markdown-0.12.9.tgz",
@@ -7716,7 +7738,6 @@
       "resolved": "https://registry.npmjs.org/zustand/-/zustand-5.0.12.tgz",
       "integrity": "sha512-i77ae3aZq4dhMlRhJVCYgMLKuSiZAaUPAct2AksxQ+gOtimhGMdXljRT21P5BNpeT4kXlLIckvkPM029OljD7g==",
       "license": "MIT",
-      "peer": true,
       "engines": {
         "node": ">=12.20.0"
       },
diff --git a/frontend/package.json b/frontend/package.json
index f46b612..a67f9e7 100644
--- a/frontend/package.json
+++ b/frontend/package.json
@@ -13,8 +13,9 @@
   },
   "dependencies": {
     "@ai-sdk/openai": "^3.0.52",
-    "@assistant-ui/react": "latest",
+    "@assistant-ui/react": "^0.12.25",
     "@assistant-ui/react-ai-sdk": "latest",
+    "@assistant-ui/react-data-stream": "^0.12.11",
     "@assistant-ui/react-markdown": "latest",
     "@radix-ui/react-avatar": "^1.1.11",
     "@radix-ui/react-collapsible": "^1.1.12",
diff --git a/frontend/public/icons/ban.svg b/frontend/public/icons/ban.svg
new file mode 100644
index 0000000..952620d
--- /dev/null
+++ b/frontend/public/icons/ban.svg
@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 640 640"><!--!Font Awesome Free 7.2.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2026 Fonticons, Inc.--><path fill="rgb(255, 5, 30)" d="M431.2 476.5L163.5 208.8C141.1 240.2 128 278.6 128 320C128 426 214 512 320 512C361.5 512 399.9 498.9 431.2 476.5zM476.5 431.2C498.9 399.8 512 361.4 512 320C512 214 426 128 320 128C278.5 128 240.1 141.1 208.8 163.5L476.5 431.2zM64 320C64 178.6 178.6 64 320 64C461.4 64 576 178.6 576 320C576 461.4 461.4 576 320 576C178.6 576 64 461.4 64 320z"/></svg>
\ No newline at end of file
diff --git a/frontend/public/icons/database.svg b/frontend/public/icons/database.svg
new file mode 100644
index 0000000..eb8a528
--- /dev/null
+++ b/frontend/public/icons/database.svg
@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 640 640"><!--!Font Awesome Free 7.2.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2026 Fonticons, Inc.--><path fill="rgb(45, 53, 196)" d="M544 269.8C529.2 279.6 512.2 287.5 494.5 293.8C447.5 310.6 385.8 320 320 320C254.2 320 192.4 310.5 145.5 293.8C127.9 287.5 110.8 279.6 96 269.8L96 352C96 396.2 196.3 432 320 432C443.7 432 544 396.2 544 352L544 269.8zM544 192L544 144C544 99.8 443.7 64 320 64C196.3 64 96 99.8 96 144L96 192C96 236.2 196.3 272 320 272C443.7 272 544 236.2 544 192zM494.5 453.8C447.6 470.5 385.9 480 320 480C254.1 480 192.4 470.5 145.5 453.8C127.9 447.5 110.8 439.6 96 429.8L96 496C96 540.2 196.3 576 320 576C443.7 576 544 540.2 544 496L544 429.8C529.2 439.6 512.2 447.5 494.5 453.8z"/></svg>
\ No newline at end of file
diff --git a/frontend/public/icons/hexagon.svg b/frontend/public/icons/hexagon.svg
new file mode 100644
index 0000000..83ca571
--- /dev/null
+++ b/frontend/public/icons/hexagon.svg
@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 640 640"><!--!Font Awesome Free 7.2.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2026 Fonticons, Inc.--><path fill="rgb(177, 151, 252)" d="M344 170.6C362.9 161.6 376 142.3 376 120C376 89.1 350.9 64 320 64C289.1 64 264 89.1 264 120C264 142.3 277.1 161.6 296 170.6L296 269.4C293.2 270.7 290.5 272.3 288 274.1L207.9 228.3C209.5 207.5 199.3 186.7 180 175.5C153.2 160 119 169.2 103.5 196C88 222.8 97.2 257 124 272.5C125.3 273.3 126.6 274 128 274.6L128 365.4C126.7 366 125.3 366.7 124 367.5C97.2 383 88 417.2 103.5 444C119 470.8 153.2 480 180 464.5C199.3 453.4 209.4 432.5 207.8 411.7L258.3 382.8C246.8 371.6 238.4 357.2 234.5 341.1L184 370.1C181.4 368.3 178.8 366.8 176 365.4L176 274.6C178.8 273.3 181.5 271.7 184 269.9L264.1 315.7C264 317.1 263.9 318.5 263.9 320C263.9 342.3 277 361.6 295.9 370.6L295.9 469.4C277 478.4 263.9 497.7 263.9 520C263.9 550.9 289 576 319.9 576C350.8 576 375.9 550.9 375.9 520C375.9 497.7 362.8 478.4 343.9 469.4L343.9 370.6C346.7 369.3 349.4 367.7 351.9 365.9L432 411.7C430.4 432.5 440.6 453.3 459.8 464.5C486.6 480 520.8 470.8 536.3 444C551.8 417.2 542.6 383 515.8 367.5C514.5 366.7 513.1 366 511.8 365.4L511.8 274.6C513.2 274 514.5 273.3 515.8 272.5C542.6 257 551.8 222.8 536.3 196C520.8 169.2 486.8 160 460 175.5C440.7 186.6 430.6 207.5 432.2 228.3L381.6 257.2C393.1 268.4 401.5 282.8 405.4 298.9L456 269.9C458.6 271.7 461.2 273.2 464 274.6L464 365.4C461.2 366.7 458.5 368.3 456 370L375.9 324.2C376 322.8 376.1 321.4 376.1 319.9C376.1 297.6 363 278.3 344.1 269.3L344.1 170.5z"/></svg>
\ No newline at end of file
diff --git a/frontend/public/icons/magnifying-glass.svg b/frontend/public/icons/magnifying-glass.svg
new file mode 100644
index 0000000..b18ce93
--- /dev/null
+++ b/frontend/public/icons/magnifying-glass.svg
@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 640 640"><!--!Font Awesome Free 7.2.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2026 Fonticons, Inc.--><path fill="rgb(116, 192, 252)" d="M480 272C480 317.9 465.1 360.3 440 394.7L566.6 521.4C579.1 533.9 579.1 554.2 566.6 566.7C554.1 579.2 533.8 579.2 521.3 566.7L394.7 440C360.3 465.1 317.9 480 272 480C157.1 480 64 386.9 64 272C64 157.1 157.1 64 272 64C386.9 64 480 157.1 480 272zM272 416C351.5 416 416 351.5 416 272C416 192.5 351.5 128 272 128C192.5 128 128 192.5 128 272C128 351.5 192.5 416 272 416z"/></svg>
\ No newline at end of file