Keep which state chat was started in

lua/gemini/init.lua

@@ -6,6 +6,7 @@ local M = {}
 -- Store the buffer number of the chat window
 local chat_bufnr = nil
 local chat_winnr = nil
+local current_context = nil  -- Store the current context
 
 local function get_current_buffer_content()
     local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false)
@@ -109,11 +110,11 @@ local function update_chat_window(new_content)
             vim.cmd('wincmd p')
         end, { buffer = chat_bufnr, nowait = true })
 
-        -- Add input mapping
+        -- Add input mapping with context awareness
         vim.keymap.set('n', 'i', function()
             vim.ui.input({ prompt = "Gemini: " }, function(input)
                 if input then
-                    M.gemini_query(input)
+                    M.gemini_query(input, current_context)
                 end
             end)
         end, { buffer = chat_bufnr, nowait = true })
@@ -157,6 +158,9 @@ local function update_chat_window(new_content)
 end
 
 local function gemini_query(prompt, context)
+    -- Store the context for subsequent queries
+    current_context = context
+    
     local response = api.get_response(prompt, context)
     if response then
         local formatted_content = "\nUser: " .. prompt .. "\n\nAssistant: " .. response