add ollama functionality

home-manager/sajenim/features/nvim/config/lua/plugins/ui.lua

@@ -26,7 +26,25 @@ return {
     dependencies = {
       'nvim-tree/nvim-web-devicons',
     },
+
     config = function()
+      -- Define a function to check that ollama is installed and working
+      local function get_condition()
+          return package.loaded["ollama"] and require("ollama").status ~= nil
+      end
+
+
+      -- Define a function to check the status and return the corresponding icon
+      local function get_status_icon()
+        local status = require("ollama").status()
+
+        if status == "IDLE" then
+          return "OLLAMA IDLE"
+        elseif status == "WORKING" then
+          return "OLLAMA BUSY"
+        end
+      end
+
       require('lualine').setup {
         options = {
           icons_enabled = true,
@@ -41,6 +59,14 @@ return {
         tabline = {
           lualine_a = {'buffers'},
         };
+        sections = {
+          lualine_a = {'mode'},
+          lualine_b = {'branch', 'diff', 'diagnostics'},
+          lualine_c = { { 'filename', path = 1 } },
+          lualine_x = { get_status_icon, get_condition, 'encoding', 'fileformat', 'filetype' },
+          lualine_y = { "progress" },
+          lualine_z = { "location" },
+        },
       }
     end,
   },
@@ -55,6 +81,6 @@ return {
     config = function()
       require('gitsigns').setup()
     end
-  }
+  },
 }