From 415b7a02d8c2fd0655a37802ecbaed41f1b30718 Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Wed, 8 Nov 2023 20:23:28 +0800
Subject: [PATCH] show img button where model is vision

---
 src/chatbox.tsx | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/src/chatbox.tsx b/src/chatbox.tsx
index c0f95a8..bf33469 100644
--- a/src/chatbox.tsx
+++ b/src/chatbox.tsx
@@ -617,15 +617,17 @@ export default function ChatBOX(props: {
       )}
 
       <div className="flex justify-between">
-        <button
-          className="disabled:line-through disabled:bg-slate-500 rounded m-1 p-1 border-2 bg-cyan-400 hover:bg-cyan-600"
-          disabled={showGenerating || !chatStore.apiKey}
-          onClick={() => {
-            setShowAddImage(!showAddImage);
-          }}
-        >
-          Img
-        </button>
+        {chatStore.model.match("vision") && (
+          <button
+            className="disabled:line-through disabled:bg-slate-500 rounded m-1 p-1 border-2 bg-cyan-400 hover:bg-cyan-600"
+            disabled={showGenerating || !chatStore.apiKey}
+            onClick={() => {
+              setShowAddImage(!showAddImage);
+            }}
+          >
+            Img
+          </button>
+        )}
         {showAddImage && (
           <div
             className="absolute z-10 bg-black bg-opacity-50 w-full h-full flex justify-center items-center left-0 top-0 overflow-scroll"