Skip to content

Commit

Permalink
Merge pull request #9 from plateaukao/add_loading_asynchronously
Browse files Browse the repository at this point in the history 
feat: add loading dialog right after clicking Ask button.
  • Loading branch information
@drewbaumann
drewbaumann authored Jul 8, 2024
2 parents 18a13f3 + c118f4a commit a18fe4c
Showing 1 changed file with 98 additions and 96 deletions.
194 changes: 98 additions & 96 deletions dialogs.lua
View file
Original file line number Diff line number Diff line change
@@ -1,117 +1,119 @@
local InputDialog = require("ui/widget/inputdialog")
local ChatGPTViewer = require("chatgptviewer")
local UIManager = require("ui/uimanager")
local InfoMessage = require("ui/widget/infomessage")
local _ = require("gettext")

local queryChatGPT = require("gpt_query")

local function showChatGPTDialog(ui, highlightedText, message_history)
local title, author =
ui.document:getProps().title or _("Unknown Title"),
ui.document:getProps().authors or _("Unknown Author")
local message_history = message_history or {
{
role = "system",
content =
"The following is a conversation with an AI assistant. The assistant is helpful, creative, clever, and very friendly. Answer as concisely as possible.",
},
local function showLoadingDialog()
local loading = InfoMessage:new{
text = _("Loading..."),
timeout = 0.1
}
UIManager:show(loading)
end

local function showChatGPTDialog(ui, highlightedText, message_history)
local title, author =
ui.document:getProps().title or _("Unknown Title"),
ui.document:getProps().authors or _("Unknown Author")
local message_history = message_history or {{
role = "system",
content = "The following is a conversation with an AI assistant. The assistant is helpful, creative, clever, and very friendly. Answer as concisely as possible."
}}
local input_dialog
input_dialog = InputDialog:new {
input_dialog = InputDialog:new{
title = _("Ask a question about the highlighted text"),
input_hint = _("Type your question here..."),
input_type = "text",
buttons = {
{
{
text = _("Cancel"),
callback = function()
UIManager:close(input_dialog)
end,
},
{
text = _("Ask"),
callback = function()
local InfoMessage = require("ui/widget/infomessage")
local loading = InfoMessage:new {
text = _("Loading..."),
timeout = 1,
}
UIManager:show(loading)

-- Give context to the question
local context_message = {
role = "user",
content = "I'm reading something titled '" ..
title ..
"' by " .. author .. ". I have a question about the following highlighted text: " .. highlightedText,
}
table.insert(message_history, context_message)

-- Ask the question
local question = input_dialog:getInputText()
local question_message = {
role = "user",
content = question,
}
table.insert(message_history, question_message)

local answer = queryChatGPT(message_history)
-- Save the answer to the message history
local answer_message = {
role = "assistant",
content = answer,
}

table.insert(message_history, answer_message)
UIManager:close(input_dialog)
local result_text = _("Highlighted text: ") .. "\"" .. highlightedText .. "\"" ..
"\n\n" .. _("User: ") .. question ..
"\n\n" .. _("ChatGPT: ") .. answer

local function createResultText(highlightedText, message_history)
local result_text = _("Highlighted text: ") .. "\"" .. highlightedText .. "\"\n\n"

for i = 3, #message_history do
if message_history[i].role == "user" then
result_text = result_text .. _("User: ") .. message_history[i].content .. "\n\n"
else
result_text = result_text .. _("ChatGPT: ") .. message_history[i].content .. "\n\n"
end
buttons = {{{
text = _("Cancel"),
callback = function()
UIManager:close(input_dialog)
end
}, {
text = _("Ask"),
callback = function()
local question = input_dialog:getInputText()
UIManager:close(input_dialog)
showLoadingDialog()

-- do the actual work after a short delay to allow the loading dialog to show
UIManager:scheduleIn(0.1, function()
-- Give context to the question
local context_message = {
role = "user",
content = "I'm reading something titled '" .. title .. "' by " .. author ..
". I have a question about the following highlighted text: " .. highlightedText
}
table.insert(message_history, context_message)

-- Ask the question
local question_message = {
role = "user",
content = question
}
table.insert(message_history, question_message)

local answer = queryChatGPT(message_history)
-- Save the answer to the message history
local answer_message = {
role = "assistant",
content = answer
}

table.insert(message_history, answer_message)
local result_text = _("Highlighted text: ") .. "\"" .. highlightedText .. "\"" .. "\n\n" .. _("User: ") ..
question .. "\n\n" .. _("ChatGPT: ") .. answer

local function createResultText(highlightedText, message_history)
local result_text = _("Highlighted text: ") .. "\"" .. highlightedText .. "\"\n\n"

for i = 3, #message_history do
if message_history[i].role == "user" then
result_text = result_text .. _("User: ") .. message_history[i].content .. "\n\n"
else
result_text = result_text .. _("ChatGPT: ") .. message_history[i].content .. "\n\n"
end

return result_text
end

return result_text
end

local function handleNewQuestion(chatgpt_viewer, question)
-- Add the new question to the message history
table.insert(message_history, { role = "user", content = question })

-- Send the query to ChatGPT with the updated message_history
local answer = queryChatGPT(message_history)

-- Add the answer to the message history
table.insert(message_history, { role = "assistant", content = answer })

-- Update the result text
local result_text = createResultText(highlightedText, message_history)
local function handleNewQuestion(chatgpt_viewer, question)
-- Add the new question to the message history
table.insert(message_history, {
role = "user",
content = question
})

-- Update the text and refresh the viewer
chatgpt_viewer:update(result_text)
end
-- Send the query to ChatGPT with the updated message_history
local answer = queryChatGPT(message_history)

local chatgpt_viewer = ChatGPTViewer:new {
title = _("AskGPT"),
text = result_text,
onAskQuestion = handleNewQuestion, -- Pass the callback function
}

UIManager:show(chatgpt_viewer)
end,
},
},
},
-- Add the answer to the message history
table.insert(message_history, {
role = "assistant",
content = answer
})

-- Update the result text
local result_text = createResultText(highlightedText, message_history)

-- Update the text and refresh the viewer
chatgpt_viewer:update(result_text)
end

local chatgpt_viewer = ChatGPTViewer:new {
title = _("AskGPT"),
text = result_text,
onAskQuestion = handleNewQuestion -- Pass the callback function
}

UIManager:show(chatgpt_viewer)
end)
end
}}}
}
UIManager:show(input_dialog)
input_dialog:onShowKeyboard()
Expand Down

0 comments on commit a18fe4c

Please sign in to comment.