Fixed message footer

* Add files via upload

* Update ru.po

README.md

@@ -21,24 +21,24 @@ An [Ollama](https://github.com/ollama/ollama) client made with GTK4 and Adwaita.
 - Pull and delete models from the app
 
 ## Future features!
-- Persistent conversations
 - Multiple conversations
 - Image / document recognition
+- Notifications
+- Code highlighting
 
 ## Screenies
 Login to Ollama instance             |  Chatting with models        |  Managing models
 :-------------------------:|:-------------------------:|:-------------------------:
-![Screenshot from 2024-05-12 19-58-28](https://github.com/Jeffser/Alpaca/assets/69224322/e28df5c9-6419-4800-bbbc-38821f096922)  |  ![Screenshot from 2024-05-12 20-01-08](https://github.com/Jeffser/Alpaca/assets/69224322/c4083864-8c39-40e6-83b6-aff9d62183ca)  |  ![Screenshot from 2024-05-12 20-01-31](https://github.com/Jeffser/Alpaca/assets/69224322/76deb8a2-13a5-480a-b99d-4de40159c229)
+![Screenshot from 2024-05-12 19-58-28](https://jeffser.com/images/alpaca/screenie1.png)  |  ![Screenshot from 2024-05-12 20-01-08](https://jeffser.com/images/alpaca/screenie2.png)  |  ![Screenshot from 2024-05-12 20-01-31](https://jeffser.com/images/alpaca/screenie3.png)
 
 ## Preview
 1. Clone repo using Gnome Builder
 2. Press the `run` button
 
 ## Instalation
-1. Clone repo using Gnome Builder
+1. Go to the `releases` page
-2. Build the app using the `build` button
+2. Download the latest flatpak package
-3. Prepare the file using the `install` button (it doesn't actually install it, idk)
+3. Open it
-4. Then press the `export` button, it will export a `com.jeffser.Alpaca.flatpak` file, you can install it just by opening it
 
 ## Usage
 - You'll need an Ollama instance, I recommend using the [Docker image](https://ollama.com/blog/ollama-is-now-available-as-an-official-docker-image)

com.jeffser.Alpaca.json

@@ -57,6 +57,20 @@
 		}
 	    ]
 	},
+	{
+            "name": "python3-pillow",
+            "buildsystem": "simple",
+            "build-commands": [
+                "pip3 install --verbose --exists-action=i --no-index --find-links=\"file://${PWD}\" --prefix=${FLATPAK_DEST} \"pillow\" --no-build-isolation"
+            ],
+            "sources": [
+                {
+                    "type": "file",
+                    "url": "https://files.pythonhosted.org/packages/ef/43/c50c17c5f7d438e836c169e343695534c38c77f60e7c90389bd77981bc21/pillow-10.3.0.tar.gz",
+                    "sha256": "9d2455fbf44c914840c793e89aa82d0e1763a14253a000743719ae5946814b2d"
+                }
+            ]
+        },
         {
             "name" : "alpaca",
             "builddir" : true,
@@ -64,8 +78,7 @@
             "sources" : [
                 {
                     "type" : "git",
-                    "url" : "https://github.com/Jeffser/Alpaca.git",
+                    "url" : "file:///home/tentri/Documents/Alpaca"
-                    "tag": "0.2.2"
                 }
             ]
         }

data/com.jeffser.Alpaca.metainfo.xml.in

@@ -7,7 +7,8 @@
   <name>Alpaca</name>
   <summary>An Ollama client</summary>
 	<description>
-	  <p>Made with GTK4 and Adwaita.</p>
+	  <p>Chat with multiple AI models</p>
+    <p>An Ollama client</p>
     <p>Features</p>
 	  <ul>
 	    <li>Talk to multiple models in the same conversation</li>
@@ -29,8 +30,8 @@
     <category>Chat</category>
   </categories>
   <branding>
-    <color type="primary" scheme_preference="light">#ff00ff</color>
+    <color type="primary" scheme_preference="light">#8cdef5</color>
-    <color type="primary" scheme_preference="dark">#993d3d</color>
+    <color type="primary" scheme_preference="dark">#0f2b78</color>
   </branding>
   <screenshots>
     <screenshot type="default">
@@ -51,8 +52,40 @@
   <url type="homepage">https://github.com/Jeffser/Alpaca</url>
   <url type="donation">https://github.com/sponsors/Jeffser</url>
   <releases>
+    <release version="0.4.0" date="2024-05-17">
+      <url type="details">https://github.com/Jeffser/Alpaca/releases/tag/0.4.0</url>
+      <description>
+        <p>Big Update</p>
+        <ul>
+          <li>Added code highlighting</li>
+          <li>Added image recognition (llava model)</li>
+          <li>Added multiline prompt</li>
+          <li>Fixed some small bugs</li>
+          <li>General optimization</li>
+        </ul>
+        <p>
+          Please report any errors to the issues page, thank you.
+        </p>
+      </description>
+    </release>
+    <release version="0.3.0" date="2024-05-16">
+      <url type="details">https://github.com/Jeffser/Alpaca/releases/tag/0.3.0</url>
+      <description>
+        <p>Fixes and features</p>
+        <ul>
+          <li>Russian translation (thanks github/alexkdeveloper)</li>
+          <li>Fixed: Cannot close app on first setup</li>
+          <li>Fixed: Brand colors for Flathub</li>
+          <li>Fixed: App description</li>
+          <li>Fixed: Only show 'save changes dialog' when you actually change the url</li>
+        </ul>
+        <p>
+          Please report any errors to the issues page, thank you.
+        </p>
+      </description>
+    </release>
     <release version="0.2.2" date="2024-05-14">
-      <url type="details">https://github.com/Jeffser/Alpaca/releases/tag/0.2.1</url>
+      <url type="details">https://github.com/Jeffser/Alpaca/releases/tag/0.2.2</url>
       <description>
         <p>0.2.2 Bug fixes</p>
         <ul>

meson.build

@@ -1,5 +1,5 @@
 project('Alpaca',
-          version: '0.2.2',
+          version: '0.4.0',
     meson_version: '>= 0.62.0',
   default_options: [ 'warning_level=2', 'werror=false', ],
 )

po/LINGUAS

@@ -0,0 +1 @@
+ru

po/ru.po

@@ -0,0 +1,112 @@
+msgid ""
+msgstr ""
+"Project-Id-Version: \n"
+"POT-Creation-Date: 2024-05-16 19:29+0800\n"
+"PO-Revision-Date: 2024-05-16 19:59+0800\n"
+"Last-Translator: \n"
+"Language-Team: \n"
+"Language: ru_RU\n"
+"MIME-Version: 1.0\n"
+"Content-Type: text/plain; charset=UTF-8\n"
+"Content-Transfer-Encoding: 8bit\n"
+"X-Generator: Poedit 3.4.2\n"
+"X-Poedit-Basepath: ../src\n"
+"X-Poedit-SearchPath-0: .\n"
+
+#: gtk/help-overlay.ui:11
+msgctxt "shortcut window"
+msgid "General"
+msgstr "Общие"
+
+#: gtk/help-overlay.ui:14
+msgctxt "shortcut window"
+msgid "Show Shortcuts"
+msgstr "Показывать ярлыки"
+
+#: gtk/help-overlay.ui:20
+msgctxt "shortcut window"
+msgid "Quit"
+msgstr "Выйти"
+
+#: window.ui:30
+msgid "Manage models"
+msgstr "Управление моделями"
+
+#: window.ui:44
+msgid "Menu"
+msgstr "Меню"
+
+#: window.ui:106
+msgid "Send"
+msgstr "Отправить"
+
+#: window.ui:137
+msgid "Pulling Model"
+msgstr "Тянущая модель"
+
+#: window.ui:218
+msgid "Previous"
+msgstr "Предыдущий"
+
+#: window.ui:233
+msgid "Next"
+msgstr "Следующий"
+
+#: window.ui:259
+msgid "Welcome to Alpaca"
+msgstr "Добро пожаловать в Alpaca"
+
+#: window.ui:260
+msgid ""
+"To get started, please ensure you have an Ollama instance set up. You can "
+"either run Ollama locally on your machine or connect to a remote instance."
+msgstr ""
+"Для начала, пожалуйста, убедитесь, что у вас настроен экземпляр Ollama. Вы "
+"можете либо запустить Ollama локально на своем компьютере, либо "
+"подключиться к удаленному экземпляру."
+
+#: window.ui:263
+msgid "Ollama Website"
+msgstr "Веб-сайт Ollama"
+
+#: window.ui:279
+msgid "Disclaimer"
+msgstr "Отказ от ответственности"
+
+#: window.ui:280
+msgid ""
+"Alpaca and its developers are not liable for any damages to devices or "
+"software resulting from the execution of code generated by an AI model. "
+"Please exercise caution and review the code carefully before running it."
+msgstr ""
+"Alpaca и ее разработчики не несут ответственности за любой ущерб, "
+"причиненный устройствам или программному обеспечению в результате "
+"выполнения кода, сгенерированного с помощью модели искусственного "
+"интеллекта. Пожалуйста, будьте осторожны и внимательно ознакомьтесь с кодом "
+"перед его запуском."
+
+#: window.ui:292
+msgid "Setup"
+msgstr "Установка"
+
+#: window.ui:293
+msgid ""
+"If you are running an Ollama instance locally and haven't modified the "
+"default ports, you can use the default URL. Otherwise, please enter the URL "
+"of your Ollama instance."
+msgstr ""
+"Если вы запускаете локальный экземпляр Ollama и не изменили порты по "
+"умолчанию, вы можете использовать URL-адрес по умолчанию. В противном "
+"случае, пожалуйста, введите URL-адрес вашего экземпляра Ollama."
+
+#: window.ui:313
+msgid "_Clear Conversation"
+msgstr "_Очистить разговор"
+
+#: window.ui:317
+msgid "_Change Server"
+msgstr "_Изменить Сервер"
+
+#: window.ui:321
+msgid "_About Alpaca"
+msgstr "_О Программе"

src/available_models.py

@@ -89,3 +89,4 @@ available_models = {
     "llava-phi3":"A new small LLaVA model fine-tuned from Phi 3 Mini."
 }
 
+

src/main.py

@@ -48,9 +48,10 @@ class AlpacaApplication(Adw.Application):
             application_name='Alpaca',
             application_icon='com.jeffser.Alpaca',
             developer_name='Jeffry Samuel Eduarte Rojas',
-            version='0.2.2',
+            version='0.4.0',
             developers=['Jeffser https://jeffser.com'],
             designers=['Jeffser https://jeffser.com'],
+            translator_credits='Alex K (Russian) https://github.com/alexkdeveloper',
             copyright='© 2024 Jeffser',
             issue_url='https://github.com/Jeffser/Alpaca/issues')
         about.present()

src/window.py

@@ -18,9 +18,12 @@
 # SPDX-License-Identifier: GPL-3.0-or-later
 
 import gi
-gi.require_version("Soup", "3.0")
+gi.require_version('GtkSource', '5')
-from gi.repository import Adw, Gtk, GLib
+gi.require_version('GdkPixbuf', '2.0')
-import json, requests, threading, os
+from gi.repository import Adw, Gtk, Gdk, GLib, GtkSource, Gio, GdkPixbuf
+import json, requests, threading, os, re, base64
+from io import BytesIO
+from PIL import Image
 from datetime import datetime
 from .connection_handler import simple_get, simple_delete, stream_post, stream_post_fake
 from .available_models import available_models
@@ -35,9 +38,12 @@ class AlpacaWindow(Adw.ApplicationWindow):
     #In the future I will at multiple chats, for now I'll save it like this so that past chats don't break in the future
     current_chat_id="0"
     chats = {"chats": {"0": {"messages": []}}}
+    attached_image = {"path": None, "base64": None}
 
     #Elements
     bot_message : Gtk.TextBuffer = None
+    bot_message_box : Gtk.Box = None
+    bot_message_view : Gtk.TextView = None
     connection_dialog = Gtk.Template.Child()
     connection_carousel = Gtk.Template.Child()
     connection_previous_button = Gtk.Template.Child()
@@ -49,8 +55,10 @@ class AlpacaWindow(Adw.ApplicationWindow):
     connection_overlay = Gtk.Template.Child()
     chat_container = Gtk.Template.Child()
     chat_window = Gtk.Template.Child()
-    message_entry = Gtk.Template.Child()
+    message_text_view = Gtk.Template.Child()
     send_button = Gtk.Template.Child()
+    image_button = Gtk.Template.Child()
+    file_filter_image = Gtk.Template.Child()
     model_drop_down = Gtk.Template.Child()
     model_string_list = Gtk.Template.Child()
 
@@ -68,7 +76,8 @@ class AlpacaWindow(Adw.ApplicationWindow):
             "Failed to connect to server",
             "Could not list local models",
             "Could not delete model",
-            "Could not pull model"
+            "Could not pull model",
+            "Cannot open image"
         ],
         "info": [
             "Please select a model before chatting",
@@ -90,7 +99,7 @@ class AlpacaWindow(Adw.ApplicationWindow):
         )
         overlay.add_toast(toast)
 
-    def show_message(self, msg:str, bot:bool, footer:str=None):
+    def show_message(self, msg:str, bot:bool, footer:str=None, image_base64:str=None):
         message_text = Gtk.TextView(
             editable=False,
             focusable=False,
@@ -99,19 +108,52 @@ class AlpacaWindow(Adw.ApplicationWindow):
             margin_bottom=12,
             margin_start=12,
             margin_end=12,
+            hexpand=True,
             css_classes=["flat"]
         )
         message_buffer = message_text.get_buffer()
         message_buffer.insert(message_buffer.get_end_iter(), msg)
         if footer is not None: message_buffer.insert_markup(message_buffer.get_end_iter(), footer, len(footer))
 
-        message_box = Adw.Bin(
+        message_box = Gtk.Box(
-            child=message_text,
+            orientation=1,
-            css_classes=["card" if bot else None]
+            css_classes=[None if bot else "card"]
         )
         message_text.set_valign(Gtk.Align.CENTER)
         self.chat_container.append(message_box)
-        if bot: self.bot_message = message_buffer
+
+        if image_base64 is not None:
+            image_data = base64.b64decode(image_base64)
+            loader = GdkPixbuf.PixbufLoader.new()
+            loader.write(image_data)
+            loader.close()
+
+            pixbuf = loader.get_pixbuf()
+            texture = Gdk.Texture.new_for_pixbuf(pixbuf)
+
+            image = Gtk.Image.new_from_paintable(texture)
+            image.set_size_request(360, 360)
+            message_box.append(image)
+
+        message_box.append(message_text)
+
+        if bot:
+            self.bot_message = message_buffer
+            self.bot_message_view = message_text
+            self.bot_message_box = message_box
+
+    def verify_if_image_can_be_used(self, pspec=None, user_data=None):
+        if self.model_drop_down.get_selected_item() == None: return True
+        selected = self.model_drop_down.get_selected_item().get_string().split(":")[0]
+        if selected in ['llava']:
+            self.image_button.set_sensitive(True)
+            return True
+        else:
+            self.image_button.set_sensitive(False)
+            self.image_button.set_css_classes([])
+            self.image_button.get_child().set_icon_name("image-x-generic-symbolic")
+            self.attached_image = {"path": None, "base64": None}
+            return False
 
     def update_list_local_models(self):
         self.local_models = []
@@ -123,6 +165,7 @@ class AlpacaWindow(Adw.ApplicationWindow):
                 self.model_string_list.append(model["name"])
                 self.local_models.append(model["name"])
             self.model_drop_down.set_selected(0)
+            self.verify_if_image_can_be_used()
             return
         else:
             self.show_connection_dialog(True)
@@ -133,64 +176,125 @@ class AlpacaWindow(Adw.ApplicationWindow):
         if response['status'] == 'ok':
             if "Ollama is running" in response['text']:
                 with open(os.path.join(self.config_dir, "server.conf"), "w+") as f: f.write(self.ollama_url)
-                self.message_entry.grab_focus_without_selecting()
+                #self.message_text_view.grab_focus_without_selecting()
                 self.update_list_local_models()
                 return True
         return False
 
+    def add_code_blocks(self):
+        text = self.bot_message.get_text(self.bot_message.get_start_iter(), self.bot_message.get_end_iter(), True)
+        GLib.idle_add(self.bot_message_view.get_parent().remove, self.bot_message_view)
+        # Define a regular expression pattern to match code blocks
+        code_block_pattern = re.compile(r'```(\w+)\n(.*?)\n```', re.DOTALL)
+        parts = []
+        pos = 0
+        for match in code_block_pattern.finditer(text):
+            start, end = match.span()
+            if pos < start:
+                normal_text = text[pos:start]
+                parts.append({"type": "normal", "text": normal_text.strip()})
+            language = match.group(1)
+            code_text = match.group(2)
+            parts.append({"type": "code", "text": code_text, "language": language})
+            pos = end
+        # Extract any remaining normal text after the last code block
+        if pos < len(text):
+            normal_text = text[pos:]
+            if normal_text.strip():
+                parts.append({"type": "normal", "text": normal_text.strip()})
+        for part in parts:
+            if part['type'] == 'normal':
+                message_text = Gtk.TextView(
+                    editable=False,
+                    focusable=False,
+                    wrap_mode= Gtk.WrapMode.WORD,
+                    margin_top=12,
+                    margin_bottom=12,
+                    margin_start=12,
+                    margin_end=12,
+                    hexpand=True,
+                    css_classes=["flat"]
+                )
+                message_buffer = message_text.get_buffer()
+                if part['text'].split("\n")[-1] == parts[-1]['text'].split("\n")[-1]:
+                    footer = "\n<small>" + part['text'].split('\n')[-1] + "</small>"
+                    part['text'] = '\n'.join(part['text'].split("\n")[:-1])
+                    message_buffer.insert(message_buffer.get_end_iter(), part['text'])
+                    message_buffer.insert_markup(message_buffer.get_end_iter(), footer, len(footer))
+                else:
+                    message_buffer.insert(message_buffer.get_end_iter(), part['text'])
+                self.bot_message_box.append(message_text)
+            else:
+                language = GtkSource.LanguageManager.get_default().get_language(part['language'])
+                buffer = GtkSource.Buffer.new_with_language(language)
+                buffer.set_text(part['text'])
+                buffer.set_style_scheme(GtkSource.StyleSchemeManager.get_default().get_scheme('classic-dark'))
+                source_view = GtkSource.View(
+                    auto_indent=True, indent_width=4, buffer=buffer, show_line_numbers=True
+                )
+                source_view.get_style_context().add_class("card")
+                self.bot_message_box.append(source_view)
+        self.bot_message = None
+        self.bot_message_box = None
+
     def update_bot_message(self, data):
         if data['done']:
             formated_datetime = datetime.now().strftime("%Y/%m/%d %H:%M")
-            text = f"\n\n<small>{data['model']}\t|\t{formated_datetime}</small>"
+            text = f"\n<small>{data['model']}\t|\t{formated_datetime}</small>"
             GLib.idle_add(self.bot_message.insert_markup, self.bot_message.get_end_iter(), text, len(text))
             vadjustment = self.chat_window.get_vadjustment()
             GLib.idle_add(vadjustment.set_value, vadjustment.get_upper())
             self.save_history()
-            self.bot_message = None
         else:
-            if self.bot_message is None:
+            if self.chats["chats"][self.current_chat_id]["messages"][-1]['role'] == "user":
-                GLib.idle_add(self.show_message, data['message']['content'], True)
                 self.chats["chats"][self.current_chat_id]["messages"].append({
                     "role": "assistant",
                     "model": data['model'],
                     "date": datetime.now().strftime("%Y/%m/%d %H:%M"),
-                    "content": data['message']['content']
+                    "content": ''
                 })
-            else:
+            GLib.idle_add(self.bot_message.insert, self.bot_message.get_end_iter(), data['message']['content'])
-                GLib.idle_add(self.bot_message.insert, self.bot_message.get_end_iter(), data['message']['content'])
+            self.chats["chats"][self.current_chat_id]["messages"][-1]['content'] += data['message']['content']
-                self.chats["chats"][self.current_chat_id]["messages"][-1]['content'] += data['message']['content']
 
-    def send_message(self):
+    def run_message(self, messages, model):
+        response = stream_post(f"{self.ollama_url}/api/chat", data=json.dumps({"model": model, "messages": messages}), callback=self.update_bot_message)
+        GLib.idle_add(self.add_code_blocks)
+        GLib.idle_add(self.send_button.set_sensitive, True)
+        GLib.idle_add(self.image_button.set_sensitive, True)
+        GLib.idle_add(self.image_button.set_css_classes, [])
+        GLib.idle_add(self.image_button.get_child().set_icon_name, "image-x-generic-symbolic")
+        self.attached_image = {"path": None, "base64": None}
+        GLib.idle_add(self.message_text_view.set_sensitive,  True)
+        if response['status'] == 'error':
+            GLib.idle_add(self.show_toast, 'error', 1, self.connection_overlay)
+            GLib.idle_add(self.show_connection_dialog, True)
+
+    def send_message(self, button):
+        if not self.message_text_view.get_buffer().get_text(self.message_text_view.get_buffer().get_start_iter(), self.message_text_view.get_buffer().get_end_iter(), False): return
         current_model = self.model_drop_down.get_selected_item()
         if current_model is None:
-            GLib.idle_add(self.show_toast, "info", 0, self.main_overlay)
+            self.show_toast("info", 0, self.main_overlay)
             return
         formated_datetime = datetime.now().strftime("%Y/%m/%d %H:%M")
         self.chats["chats"][self.current_chat_id]["messages"].append({
             "role": "user",
             "model": "User",
             "date": formated_datetime,
-            "content": self.message_entry.get_text()
+            "content": self.message_text_view.get_buffer().get_text(self.message_text_view.get_buffer().get_start_iter(), self.message_text_view.get_buffer().get_end_iter(), False)
         })
         data = {
             "model": current_model.get_string(),
             "messages": self.chats["chats"][self.current_chat_id]["messages"]
         }
-        GLib.idle_add(self.message_entry.set_sensitive, False)
+        if self.verify_if_image_can_be_used() and self.attached_image["base64"] is not None:
-        GLib.idle_add(self.send_button.set_sensitive, False)
+            data["messages"][-1]["images"] = [self.attached_image["base64"]]
-        GLib.idle_add(self.show_message, self.message_entry.get_text(), False, f"\n\n<small>{formated_datetime}</small>")
+        self.message_text_view.set_sensitive(False)
-        self.save_history()
+        self.send_button.set_sensitive(False)
-        GLib.idle_add(self.message_entry.get_buffer().set_text, "", 0)
+        self.image_button.set_sensitive(False)
-        response = stream_post(f"{self.ollama_url}/api/chat", data=json.dumps(data), callback=self.update_bot_message)
+        self.show_message(self.message_text_view.get_buffer().get_text(self.message_text_view.get_buffer().get_start_iter(), self.message_text_view.get_buffer().get_end_iter(), False), False, f"\n\n<small>{formated_datetime}</small>", self.attached_image["base64"])
-        GLib.idle_add(self.send_button.set_sensitive, True)
+        self.message_text_view.get_buffer().set_text("", 0)
-        GLib.idle_add(self.message_entry.set_sensitive, True)
+        self.show_message("", True)
-        if response['status'] == 'error':
+        thread = threading.Thread(target=self.run_message, args=(data['messages'], data['model']))
-            GLib.idle_add(self.show_toast, 'error', 1, self.connection_overlay)
-            GLib.idle_add(self.show_connection_dialog, True)
-
-    def send_button_activate(self, button):
-        if not self.message_entry.get_text(): return
-        thread = threading.Thread(target=self.send_message)
         thread.start()
 
     def delete_model(self, dialog, task, model_name, button):
@@ -236,7 +340,6 @@ class AlpacaWindow(Adw.ApplicationWindow):
                 GLib.idle_add(self.show_toast, "error", 4, self.connection_overlay)
                 GLib.idle_add(self.manage_models_dialog.close)
                 GLib.idle_add(self.show_connection_dialog, True)
-                print("pull fail")
 
 
     def pull_model_start(self, dialog, task, model_name, button):
@@ -268,7 +371,7 @@ class AlpacaWindow(Adw.ApplicationWindow):
                 title = model_name,
                 subtitle = model_description,
             )
-            model_name += ":latest"
+            if ":" not in model_name: model_name += ":latest"
             button = Gtk.Button(
                 icon_name = "folder-download-symbolic" if model_name not in self.local_models else "user-trash-symbolic",
                 vexpand = False,
@@ -279,7 +382,6 @@ class AlpacaWindow(Adw.ApplicationWindow):
             self.model_list_box.append(model)
 
     def manage_models_button_activate(self, button):
-
         self.manage_models_dialog.present(self)
         self.update_list_available_models()
 
@@ -351,9 +453,10 @@ class AlpacaWindow(Adw.ApplicationWindow):
                 self.chats = {"chats": {"0": {"messages": []}}}
             for message in self.chats['chats'][self.current_chat_id]['messages']:
                 if message['role'] == 'user':
-                    self.show_message(message['content'], False, f"\n\n<small>{message['date']}</small>")
+                    self.show_message(message['content'], False, f"\n\n<small>{message['date']}</small>", message['images'][0] if 'images' in message and len(message['images']) > 0 else None)
                 else:
                     self.show_message(message['content'], True, f"\n\n<small>{message['model']}\t|\t{message['date']}</small>")
+                    self.add_code_blocks()
                     self.bot_message = None
 
     def closing_connection_dialog_response(self, dialog, task):
@@ -361,6 +464,7 @@ class AlpacaWindow(Adw.ApplicationWindow):
         if result == "cancel": return
         if result == "save":
             self.ollama_url = self.connection_url_entry.get_text()
+        elif result == "discard" and self.ollama_url is None: self.destroy()
         self.connection_dialog.force_close()
         if self.ollama_url is None or self.verify_connection() == False:
             self.show_connection_dialog(True)
@@ -368,31 +472,91 @@ class AlpacaWindow(Adw.ApplicationWindow):
 
 
     def closing_connection_dialog(self, dialog):
-        if self.get_visible() == False:
+        if self.ollama_url is None: self.destroy()
-            self.destroy()
+        if self.ollama_url == self.connection_url_entry.get_text():
-        else:
+            self.connection_dialog.force_close()
+            if self.ollama_url is None or self.verify_connection() == False:
+                self.show_connection_dialog(True)
+                self.show_toast("error", 1, self.connection_overlay)
+            return
+        dialog = Adw.AlertDialog(
+            heading=f"Save Changes?",
+            body=f"Do you want to save the URL change?",
+            close_response="cancel"
+        )
+        dialog.add_response("cancel", "Cancel")
+        dialog.add_response("discard", "Discard")
+        dialog.add_response("save", "Save")
+        dialog.set_response_appearance("discard", Adw.ResponseAppearance.DESTRUCTIVE)
+        dialog.set_response_appearance("save", Adw.ResponseAppearance.SUGGESTED)
+        dialog.choose(
+            parent = self,
+            cancellable = None,
+            callback = self.closing_connection_dialog_response
+        )
+
+    def load_image(self, file_dialog, result):
+        try: file = file_dialog.open_finish(result)
+        except: return
+        try:
+            self.attached_image["path"] = file.get_path()
+            '''with open(self.attached_image["path"], "rb") as image_file:
+                self.attached_image["base64"] = base64.b64encode(image_file.read()).decode("utf-8")'''
+            with Image.open(self.attached_image["path"]) as img:
+                width, height = img.size
+                max_size = 240
+                if width > height:
+                    new_width = max_size
+                    new_height = int((max_size / width) * height)
+                else:
+                    new_height = max_size
+                    new_width = int((max_size / height) * width)
+                resized_img = img.resize((new_width, new_height), Image.LANCZOS)
+                with BytesIO() as output:
+                    resized_img.save(output, format="JPEG")
+                    image_data = output.getvalue()
+                self.attached_image["base64"] = base64.b64encode(image_data).decode("utf-8")
+
+            self.image_button.set_css_classes(["destructive-action"])
+            self.image_button.get_child().set_icon_name("edit-delete-symbolic")
+        except Exception as e:
+            print(e)
+            self.show_toast("error", 5, self.main_overlay)
+
+    def remove_image(self, dialog, task):
+        if dialog.choose_finish(task) == 'remove':
+            self.image_button.set_css_classes([])
+            self.image_button.get_child().set_icon_name("image-x-generic-symbolic")
+            self.attached_image = {"path": None, "base64": None}
+
+    def open_image(self, button):
+        if "destructive-action" in button.get_css_classes():
             dialog = Adw.AlertDialog(
-                heading=f"Save Changes?",
+                heading=f"Remove Image?",
-                body=f"Do you want to save the URL change?",
+                body=f"Are you sure you want to remove image?",
                 close_response="cancel"
             )
             dialog.add_response("cancel", "Cancel")
-            dialog.add_response("discard", "Discard")
+            dialog.add_response("remove", "Remove")
-            dialog.add_response("save", "Save")
+            dialog.set_response_appearance("remove", Adw.ResponseAppearance.DESTRUCTIVE)
-            dialog.set_response_appearance("discard", Adw.ResponseAppearance.DESTRUCTIVE)
-            dialog.set_response_appearance("save", Adw.ResponseAppearance.SUGGESTED)
             dialog.choose(
                 parent = self,
                 cancellable = None,
-                callback = self.closing_connection_dialog_response
+                callback = self.remove_image
-        )
+            )
+        else:
+            file_dialog = Gtk.FileDialog(default_filter=self.file_filter_image)
+            file_dialog.open(self, None, self.load_image)
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
+        GtkSource.init()
         self.manage_models_button.connect("clicked", self.manage_models_button_activate)
-        self.send_button.connect("clicked", self.send_button_activate)
+        self.send_button.connect("clicked", self.send_message)
+        self.image_button.connect("clicked", self.open_image)
         self.set_default_widget(self.send_button)
-        self.message_entry.set_activates_default(self.send_button)
+        self.model_drop_down.connect("notify", self.verify_if_image_can_be_used)
+        #self.message_text_view.set_activates_default(self.send_button)
         self.connection_carousel.connect("page-changed", self.connection_carousel_page_changed)
         self.connection_previous_button.connect("clicked", self.connection_previous_button_activate)
         self.connection_next_button.connect("clicked", self.connection_next_button_activate)
@@ -404,7 +568,6 @@ class AlpacaWindow(Adw.ApplicationWindow):
                 self.ollama_url = f.read()
             if self.verify_connection() is False: self.show_connection_dialog(True)
         else: self.connection_dialog.present(self)
-        self.show_toast("funny", True, self.manage_models_overlay)
 
 
 

src/window.ui

@@ -67,8 +67,6 @@
                     <property name="kinetic-scrolling">1</property>
                     <property name="vexpand">true</property>
                     <style>
-                      <class name="undershoot-top"/>
-                      <class name="undershoot-bottom"/>
                       <class name="card"/>
                     </style>
                     <child>
@@ -92,19 +90,54 @@
                     <property name="orientation">0</property>
                     <property name="spacing">12</property>
                     <child>
-                      <object class="GtkEntry" id="message_entry">
+                      <object class="GtkScrolledWindow">
-                        <property name="hexpand">true</property>
-                      </object>
-                    </child>
-                    <child>
-                      <object class="GtkButton" id="send_button">
                         <style>
-                          <class name="suggested-action"/>
+                          <class name="card"/>
+                          <class name="view"/>
                         </style>
                         <child>
-                          <object class="AdwButtonContent">
+                          <object class="GtkTextView" id="message_text_view">
-                            <property name="label" translatable="true">Send</property>
+                            <property name="wrap-mode">word</property>
-                            <property name="icon-name">send-to-symbolic</property>
+                            <property name="margin-top">6</property>
+                            <property name="margin-bottom">6</property>
+                            <property name="margin-start">6</property>
+                            <property name="margin-end">6</property>
+                            <property name="hexpand">true</property>
+                            <style>
+                              <class name="view"/>
+                            </style>
+                          </object>
+                        </child>
+                      </object>
+
+                    </child>
+                    <child>
+                      <object class="GtkBox">
+                        <property name="orientation">1</property>
+                        <property name="spacing">12</property>
+                        <child>
+                          <object class="GtkButton" id="send_button">
+                            <style>
+                              <class name="suggested-action"/>
+                            </style>
+                            <child>
+                              <object class="AdwButtonContent">
+                                <property name="label" translatable="true">Send</property>
+                                <property name="icon-name">send-to-symbolic</property>
+                              </object>
+                            </child>
+                          </object>
+                        </child>
+                        <child>
+                          <object class="GtkButton" id="image_button">
+                            <property name="sensitive">false</property>
+                            <property name="tooltip-text" translatable="true">Requires model 'llava' to be selected</property>
+                            <child>
+                              <object class="AdwButtonContent">
+                                <property name="label" translatable="true">Image</property>
+                                <property name="icon-name">image-x-generic-symbolic</property>
+                              </object>
+                            </child>
                           </object>
                         </child>
                       </object>
@@ -323,4 +356,14 @@
       </item>
     </section>
   </menu>
+
+  <object class="GtkFileFilter" id="file_filter_image">
+    <mime-types>
+      <mime-type>image/svg+xml</mime-type>
+      <mime-type>image/png</mime-type>
+      <mime-type>image/jpeg</mime-type>
+      <mime-type>image/webp</mime-type>
+      <mime-type>image/gif</mime-type>
+    </mime-types>
+  </object>
 </interface>