diff --git a/components/AIAssist.tsx b/components/AIAssist.tsx index 75f1d9e..02892dd 100644 --- a/components/AIAssist.tsx +++ b/components/AIAssist.tsx @@ -1,11 +1,10 @@ "use client"; -import React, { useState, useEffect, useRef, useCallback, memo } from "react"; +import React, { useState, useEffect, useRef, memo } from "react"; import { - MessageSquare, Send, Sparkles, Brain, Cpu, Code2, Palette, Search, - Terminal, Eye, Trash2, Loader2, Bot, User, X, RotateCcw, - CheckCircle2, Copy, Monitor, StopCircle, Maximize2, Minimize2, - ChevronRight, Layout, Zap, Ghost + MessageSquare, Send, Code2, Palette, Search, + Trash2, Copy, Monitor, StopCircle, X, Zap, Ghost, + Wand2, LayoutPanelLeft, Play, Orbit } from "lucide-react"; import ReactMarkdown from "react-markdown"; import remarkGfm from "remark-gfm"; @@ -41,14 +40,16 @@ const LiveCanvas = memo(({ data, type, isStreaming }: { data: string, type: stri useEffect(() => { if (!iframeRef.current) return; - const isHtml = data.includes(""); + const shouldRender = isHtml || ["web", "app", "design", "html", "ui"].includes(type); + if (shouldRender) { const doc = ` +