diff --git a/src/components/chat/chat-message.tsx b/src/components/chat/chat-message.tsx
index 9f0b543e..20841635 100644
--- a/src/components/chat/chat-message.tsx
+++ b/src/components/chat/chat-message.tsx
@@ -91,10 +91,38 @@ function ChatMessage({ message, isLast, isLoading, reload }: ChatMessageProps) {
)
);
+ const markdownComponents = {
+ p: ({ children }: any) => (
+ {children}
+ ),
+ table: ({ children }: any) => (
+
+ ),
+ th: ({ children }: any) => (
+ {children} |
+ ),
+ td: ({ children }: any) => (
+ {children} |
+ ),
+ tr: ({ children }: any) => (
+ {children}
+ ),
+ };
+
const renderContent = () => (
contentParts.map((part, index) => (
index % 2 === 0 ? (
- {part}
+
+ {part}
+
) : (
diff --git a/src/components/chat/chat-topbar.tsx b/src/components/chat/chat-topbar.tsx
index 48d2c245..ce7fdbec 100644
--- a/src/components/chat/chat-topbar.tsx
+++ b/src/components/chat/chat-topbar.tsx
@@ -99,18 +99,30 @@ export default function ChatTopbar({
{models.length > 0 ? (
- models.map((model) => (
-
- ))
+ models.map((model) => {
+ const visionModels = ['llava', 'bakllava', 'llava-phi3', 'moondream'];
+ const isVisionModel = visionModels.some(visionModel =>
+ model.toLowerCase().includes(visionModel.toLowerCase())
+ );
+
+ return (
+
+ );
+ })
) : (