fix: format output with markdown

.gitignore

@@ -18,6 +18,7 @@
 # misc
 .DS_Store
 *.pem
+.idea
 
 # debug
 npm-debug.log*
@@ -27,6 +28,7 @@ yarn-error.log*
 
 # local env files
 .env*.local
+.env
 
 # vercel
 .vercel

app/components/Message.js

@@ -1,3 +1,7 @@
+import ReactMarkdown from 'react-markdown';
+import { dark } from "react-syntax-highlighter/src/styles/hljs";
+import { Prism as SyntaxHighlighter } from "react-syntax-highlighter";
+
 const Message = ({ message, isUser }) => {
   let containerClass = "bg-gray-50";
   if (isUser) {
@@ -25,14 +29,30 @@ const Message = ({ message, isUser }) => {
       )}
 
       <div className="flex flex-col text-sm sm:text-base flex-1 gap-y-4 mt-1">
-        {message.split("\n").map(
-          (text, index) =>
-            text.length > 0 && (
-              <span key={index} className="min-w-0">
-                {text}
-              </span>
-            )
-        )}
+        <ReactMarkdown
+          components={{
+            code(props) {
+              const { children, className, node, ...rest } = props
+              const match = /language-(\w+)/.exec(className || '')
+              return match ? (
+                <SyntaxHighlighter
+                  {...rest}
+                  PreTag="div"
+                  language={match[1]}
+                  style={dark}
+                >
+                  {String(children).replace(/\n$/, '')}
+                </SyntaxHighlighter>
+              ) : (
+                <code {...rest} className={className}>
+                  {children}
+                </code>
+              )
+            }
+          }}
+        >
+          {message}
+        </ReactMarkdown>
       </div>
     </div>
   );