Skip to content

Commit

Permalink
Added an option to show/hide all tags. Fixing AI prompt bug.
Browse files Browse the repository at this point in the history
  • Loading branch information
frianasoa committed Dec 19, 2023
1 parent 3122ef9 commit 046e5cd
Show file tree
Hide file tree
Showing 4 changed files with 32 additions and 19 deletions.
19 changes: 16 additions & 3 deletions content/notes/menus.js
Original file line number Diff line number Diff line change
Expand Up @@ -351,6 +351,13 @@ Menus = {
{
currentcomment = "";
}

if(!annotation)
{
alert("Annotation text not found!"+annotationkey);
return;
}

var tl = Zotero.ZeNotes.Prefs.get("target-language");
Zotero.ZeNotes.Ai.Google.translate(annotation["annotationText"], tl, mode).then(r=>{
var table = AiUi.createdialog(annotation, currentcomment, r, "g-translate");
Expand Down Expand Up @@ -435,12 +442,18 @@ Menus = {
Zotero.ZeNotes.Ai.Bard.customprompt(JSON.stringify(data), target).then(r=>{
var table = AiUi.createdialog(annotation, currentcomment, r, "bard");
var model = Zotero.ZeNotes.Prefs.get("bard-model");

var div = document.createElement("div");
div.innerHTML = "<h2>Custom prompt</h2> "+customprompt+"<hr/>"+this.displayjson(data);

TabbedDialog.open(table, div, function(){}, "Edit and choose a candidate [Bard: "+model+"]", "close");
try {
TabbedDialog.open(table, div, function(){}, "Edit and choose a candidate [Bard: "+model+"]", "close");
}
catch(e)
{
alert(e);
}
}).catch(r=>{
var div = document.createElement("div");
div.innerHTML = "<h2>Custom prompt</h2> "+customprompt+"<hr/>"+this.displayjson(data);
var html = "";
if(Array.isArray(r))
{
Expand Down
12 changes: 6 additions & 6 deletions content/notes/notes.js
Original file line number Diff line number Diff line change
Expand Up @@ -185,12 +185,12 @@ Notes = {

td.querySelectorAll(".annotation").forEach(a=>{
a.addEventListener("mouseover", function(e){
e.target.closest("td").dataset.annotationid = e.target.dataset.annotationid;
e.target.closest("td").dataset.attachmentid = e.target.dataset.attachmentid;
e.target.closest("td").dataset.attachmentkey = e.target.dataset.attachmentkey;
e.target.closest("td").dataset.annotationpage = e.target.dataset.annotationpage;
e.target.closest("td").dataset.annotationkey = e.target.dataset.annotationkey;
e.target.closest("td").dataset.annotationdomid = e.target.id;
e.target.closest("td").dataset.annotationid = e.currentTarget.dataset.annotationid;
e.target.closest("td").dataset.attachmentid = e.currentTarget.dataset.attachmentid;
e.target.closest("td").dataset.attachmentkey = e.currentTarget.dataset.attachmentkey;
e.target.closest("td").dataset.annotationpage = e.currentTarget.dataset.annotationpage;
e.target.closest("td").dataset.annotationkey = e.currentTarget.dataset.annotationkey;
e.target.closest("td").dataset.annotationdomid = e.currentTarget.id;
});
});

Expand Down
18 changes: 9 additions & 9 deletions core/ai.js
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ Ai={
return Promise.resolve(data.candidates.map(function(v){return v.output}));
}
catch(e) {
return Promise.resolve([data.error.message]);
return Promise.resolve([data.error.message, JSON.stringify(data)]);
}
}
else if(mode=="gemini-pro")
Expand All @@ -24,7 +24,7 @@ Ai={
try {
if(data.candidates[0].finishReason=="OTHER")
{
return Promise.resolve(["Error: No results found!"]);
return Promise.resolve(["Error: No results found!", JSON.stringify(data)]);
}
else
{
Expand All @@ -45,7 +45,7 @@ Ai={
}
catch(e)
{
return Promise.resolve(["Error: "+e]);
return Promise.resolve(["Error: "+e, JSON.stringify(data)]);
}
}
else if(mode=="g-translate-free-0")
Expand All @@ -56,7 +56,7 @@ Ai={
}
catch(e)
{
return Promise.resolve(["Error: "+e]);
return Promise.resolve(["Error: "+e, JSON.stringify(data)]);
}
}
else if(mode=="g-translate-free-1")
Expand All @@ -67,7 +67,7 @@ Ai={
}
catch(e)
{
return Promise.resolve(["Error: "+e]);
return Promise.resolve(["Error: "+e, JSON.stringify(data)]);
}
}
else if(mode=="bing")
Expand All @@ -83,13 +83,13 @@ Ai={
}

}).catch(e=>{
return Promise.reject(["Error: "+e]);
return Promise.reject(["Error: "+e, JSON.stringify(data)]);
});
},
prompts: {
cell: "Paraphrase and summarize 'Direct quotes'",
row: "Paraphrase and summarize 'Direct quotes'",
table: "Summarize all the data",
row: "Summarize the data below into a coherent literature review. Add source for each claim in the form (Author date). ",
table: "Summarize the data below into a coherent literature review. Add source for each claim in the form (Author date). ",
}
}

Expand Down Expand Up @@ -136,7 +136,7 @@ Ai.Bard = {
async customprompt(sentence, target)
{
var model = Zotero.ZeNotes.Prefs.get("bard-model");
var prompts = Zotero.ZeNotes.Prefs.get("cell-custom-prompt");
var prompts = Zotero.ZeNotes.Prefs.get(target+"-custom-prompt");

if(prompts=="")
{
Expand Down
2 changes: 1 addition & 1 deletion zenote-update.json
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
{
"version": "0.6.6",
"update_link": "https://github.com/frianasoa/Ze-Notes/releases/download/v0.6.6/zenotes-v0.6.6.xpi",
"update_hash": "sha256:c9bc7bb01d8b6daf534635de158392d1b6ca39a80f95e6ccc617aed3dae30787",
"update_hash": "sha256:095b1f38f3f1462acbc3db68508ad73ff9c24b321603531083d8ad04dd21cebe",
"applications": {
"gecko": {
"strict_min_version": "60.0"
Expand Down

0 comments on commit 046e5cd

Please sign in to comment.