Merge pull request #82 from E-mran/Code-Blocks

code blocks update

Author: jakobhoeg

src/components/chat/chat-list.tsx

@@ -217,7 +217,7 @@ export default function ChatList({
                       } else {
                         return (
                           <pre className="whitespace-pre-wrap" key={index}>
-                            <CodeDisplayBlock code={part} lang="" />
+                            <CodeDisplayBlock code={part} />
                           </pre>
                         );
                       }

src/components/code-display-block.tsx

@@ -8,15 +8,17 @@ import { useTheme } from "next-themes";
 
 interface ButtonCodeblockProps {
   code: string;
-  lang: string;
 }
 
-export default function CodeDisplayBlock({ code, lang }: ButtonCodeblockProps) {
+export default function CodeDisplayBlock({ code }: ButtonCodeblockProps) {
   const [isCopied, setisCopied] = React.useState(false);
   const { theme } = useTheme();
 
+  const filteredCode = code.split("\n").slice(1).join("\n") || code;
+  const language = code.split("\n")[0] || "tsx";
+  
   const copyToClipboard = () => {
-    navigator.clipboard.writeText(code);
+    navigator.clipboard.writeText(filteredCode);
     setisCopied(true);
     toast.success("Code copied to clipboard!");
     setTimeout(() => {
@@ -45,7 +47,7 @@ export default function CodeDisplayBlock({ code, lang }: ButtonCodeblockProps) {
             : { background: "#fcfcfc" }
         }
         text={code}
-        language="tsx"
+        language={language}
         showLineNumbers={false}
         theme={theme === "dark" ? dracula : github}
       />