From f222084f94ee447ca7bc49569972d5dfe8c096d2 Mon Sep 17 00:00:00 2001
From: Dmitri Nasonov <dmitri.nasonov.b@gmail.com>
Date: Wed, 3 Jul 2024 00:24:51 +0100
Subject: [PATCH] Added ability to specify images via URL instead of base64

---
 public/locales/en/main.json                   |   2 +
 .../ChatContent/Message/View/ContentView.tsx  |  23 ++-
 .../ChatContent/Message/View/EditView.tsx     | 131 ++++++++++++------
 3 files changed, 106 insertions(+), 50 deletions(-)

diff --git a/public/locales/en/main.json b/public/locales/en/main.json
index d0b50592a..831f51249 100644
--- a/public/locales/en/main.json
+++ b/public/locales/en/main.json
@@ -1,6 +1,8 @@
 {
   "save": "Save",
   "generate": "Generate",
+  "add_image_url": "Add image URL",
+  "enter_image_url_placeholder": "Enter image URL",
   "cancel": "Cancel",
   "confirm": "Confirm",
   "warning": "Warning",
diff --git a/src/components/Chat/ChatContent/Message/View/ContentView.tsx b/src/components/Chat/ChatContent/Message/View/ContentView.tsx
index 11556aec5..33a12cae1 100644
--- a/src/components/Chat/ChatContent/Message/View/ContentView.tsx
+++ b/src/components/Chat/ChatContent/Message/View/ContentView.tsx
@@ -19,7 +19,12 @@ import CrossIcon from '@icon/CrossIcon';
 
 import useSubmit from '@hooks/useSubmit';
 
-import { ChatInterface, ContentInterface, ImageContentInterface, TextContentInterface } from '@type/chat';
+import {
+  ChatInterface,
+  ContentInterface,
+  ImageContentInterface,
+  TextContentInterface,
+} from '@type/chat';
 
 import { codeLanguageSubset } from '@constants/chat';
 
@@ -41,7 +46,7 @@ const ContentView = memo(
     messageIndex,
   }: {
     role: string;
-    content: ContentInterface[],
+    content: ContentInterface[];
     setIsEdit: React.Dispatch<React.SetStateAction<boolean>>;
     messageIndex: number;
   }) => {
@@ -132,13 +137,19 @@ const ContentView = memo(
               {(content[0] as TextContentInterface).text}
             </ReactMarkdown>
           ) : (
-            <span className='whitespace-pre-wrap'>{(content[0] as TextContentInterface).text}</span>
+            <span className='whitespace-pre-wrap'>
+              {(content[0] as TextContentInterface).text}
+            </span>
           )}
         </div>
-        <div className="flex gap-4">
+        <div className='flex gap-4'>
           {(content.slice(1) as ImageContentInterface[]).map((image, index) => (
-            <div key={index} className="image-container">
-              <img src={image.image_url.url} alt={`uploaded-${index}`} className="h-20" />
+            <div key={index} className='image-container'>
+              <img
+                src={image.image_url.url}
+                alt={`uploaded-${index}`}
+                className='h-20'
+              />
             </div>
           ))}
         </div>
diff --git a/src/components/Chat/ChatContent/Message/View/EditView.tsx b/src/components/Chat/ChatContent/Message/View/EditView.tsx
index 7bfdf2830..ff5f0e85f 100644
--- a/src/components/Chat/ChatContent/Message/View/EditView.tsx
+++ b/src/components/Chat/ChatContent/Message/View/EditView.tsx
@@ -34,6 +34,7 @@ const EditView = ({
 
   const [_content, _setContent] = useState<ContentInterface[]>(content);
   const [isModalOpen, setIsModalOpen] = useState<boolean>(false);
+  const [imageUrl, setImageUrl] = useState<string>('');
   const textareaRef = React.createRef<HTMLTextAreaElement>();
 
   const { t } = useTranslation();
@@ -104,6 +105,22 @@ const EditView = ({
     _setContent(updatedContent);
   };
 
+  const handleImageUrlChange = () => {
+    if (imageUrl.trim() === '') return;
+
+    const newImage: ImageContentInterface = {
+      type: 'image_url',
+      image_url: {
+        detail: 'auto',
+        url: imageUrl,
+      },
+    };
+
+    const updatedContent = [..._content, newImage];
+    _setContent(updatedContent);
+    setImageUrl('');
+  };
+
   const handleImageDetailChange = (index: number, detail: string) => {
     const updatedImages = [..._content];
     updatedImages[index + 1].image_url.detail = detail;
@@ -229,6 +246,9 @@ const EditView = ({
         setIsEdit={setIsEdit}
         _setContent={_setContent}
         _content={_content}
+        imageUrl={imageUrl}
+        setImageUrl={setImageUrl}
+        handleImageUrlChange={handleImageUrlChange}
       />
       {isModalOpen && (
         <PopupModal
@@ -254,6 +274,9 @@ const EditViewButtons = memo(
     setIsEdit,
     _setContent,
     _content,
+    imageUrl,
+    setImageUrl,
+    handleImageUrlChange,
   }: {
     sticky?: boolean;
     handleFileChange: (e: React.ChangeEvent<HTMLInputElement>) => void;
@@ -265,6 +288,9 @@ const EditViewButtons = memo(
     setIsEdit: React.Dispatch<React.SetStateAction<boolean>>;
     _setContent: React.Dispatch<React.SetStateAction<ContentInterface[]>>;
     _content: ContentInterface[];
+    imageUrl: string;
+    setImageUrl: React.Dispatch<React.SetStateAction<string>>;
+    handleImageUrlChange: () => void;
   }) => {
     const { t } = useTranslation();
     const generating = useStore.getState().generating;
@@ -286,65 +312,82 @@ const EditViewButtons = memo(
     return (
       <div>
         {modelTypes[model] == 'image' && (
-          <div className='flex justify-center'>
-            <div className='flex gap-5'>
-              {_content.slice(1).map((image, index) => (
-                <div
-                  key={index}
-                  className='image-container flex flex-col gap-2'
-                >
-                  <img
-                    src={image.image_url.url}
-                    alt={`uploaded-${index}`}
-                    className='h-10'
-                  />
-                  <div className='flex flex-row gap-3'>
-                    <select
-                      onChange={(event) =>
-                        handleImageDetailChange(index, event.target.value)
-                      }
-                      title='Select image resolution'
-                      aria-label='Select image resolution'
-                      defaultValue={image.image_url.detail}
-                      style={{ color: 'black' }}
-                    >
-                      <option value='auto'>Auto</option>
-                      <option value='high'>High</option>
-                      <option value='low'>Low</option>
-                    </select>
-                    <button
-                      className='close-button'
-                      onClick={() => handleRemoveImage(index)}
-                      aria-label='Remove Image'
-                    >
-                      &times;
-                    </button>
+          <>
+            <div className='flex justify-center'>
+              <div className='flex gap-5'>
+                {_content.slice(1).map((image, index) => (
+                  <div
+                    key={index}
+                    className='image-container flex flex-col gap-2'
+                  >
+                    <img
+                      src={image.image_url.url}
+                      alt={`uploaded-${index}`}
+                      className='h-10'
+                    />
+                    <div className='flex flex-row gap-3'>
+                      <select
+                        onChange={(event) =>
+                          handleImageDetailChange(index, event.target.value)
+                        }
+                        title='Select image resolution'
+                        aria-label='Select image resolution'
+                        defaultValue={image.image_url.detail}
+                        style={{ color: 'black' }}
+                      >
+                        <option value='auto'>Auto</option>
+                        <option value='high'>High</option>
+                        <option value='low'>Low</option>
+                      </select>
+                      <button
+                        className='close-button'
+                        onClick={() => handleRemoveImage(index)}
+                        aria-label='Remove Image'
+                      >
+                        &times;
+                      </button>
+                    </div>
                   </div>
-                </div>
-              ))}
+                ))}
 
+                <button
+                  className='btn relative btn-neutral h-10'
+                  onClick={handleUploadButtonClick}
+                  aria-label={'Upload Images'}
+                >
+                  <div className='flex items-center justify-center gap-2'>
+                    <FolderIcon />
+                  </div>
+                </button>
+              </div>
+            </div>
+            <div className='flex justify-center mt-4'>
+              <input
+                type='text'
+                value={imageUrl}
+                onChange={(e) => setImageUrl(e.target.value)}
+                placeholder={t('enter_image_url_placeholder') as string}
+                className='input input-bordered w-full max-w-xs text-gray-800 dark:text-white p-3  border-none bg-gray-200 dark:bg-gray-600 rounded-md m-0 w-full mr-0 h-10 focus:outline-none'
+              />
               <button
-                className='btn relative btn-neutral h-10'
-                onClick={handleUploadButtonClick}
-                aria-label={'Upload Images'}
+                className='btn btn-primary ml-2'
+                onClick={handleImageUrlChange}
+                aria-label={t('add_image_url') as string}
               >
-                <div className='flex items-center justify-center gap-2'>
-                  <FolderIcon />
-                </div>
+                {t('add_image_url')}
               </button>
             </div>
-
             {/* Hidden file input */}
             <input
               type='file'
               ref={fileInputRef}
               style={{ display: 'none' }}
               onChange={handleFileChange}
-              accept='image/*'
               multiple
             />
-          </div>
+          </>
         )}
+
         <div className='flex'>
           <div className='flex-1 text-center mt-2 flex justify-center'>
             {sticky && (