Diff狩作成

gemini.js

@@ -4,26 +4,44 @@ let lastTokenUpdateTimestamp = 0;
 let summeries = {};
 let lastIndexUpdateTimestamp = 0;
 
+let replaceProofReadHistory = [];
+
 function replaceProofRead(textarea, proofReadText) {
-    let novelContent1TextLines   = document.getElementById("novelContent1").value.split("\n");
+    let novelContent1TextLines = document.getElementById("novelContent1").value.split("\n");
     let proofReadTextLines = proofReadText.split("\n");
     let textareaTextLines = textarea.value.split("\n");
     let start = novelContent1TextLines.indexOf(textareaTextLines[0]);
     let end = novelContent1TextLines.indexOf(textareaTextLines[textareaTextLines.length - 1]);
     console.log(start, end);
+
+    // 差し替え前のテキストを保存
+    let originalText = novelContent1TextLines.slice(start, end + 1);
+    replaceProofReadHistory.push([originalText, proofReadTextLines]);
+
     // novelContent1TextLinesから該当部分を削除し、proofReadTextLinesを挿入
     novelContent1TextLines.splice(start, end - start + 1, ...proofReadTextLines);
-    
+
     // 更新された内容をnovelContent1に反映
     document.getElementById("novelContent1").value = novelContent1TextLines.join("\n");
-    
+
     // textareaの内容も更新
     textarea.value = proofReadTextLines.join("\n");
-    
+
     console.log("校正が完了しました。");
 }
 
 
+async function getModelList() {
+    const url = 'https://generativelanguage.googleapis.com/v1beta/models?key=' + document.getElementById('geminiApiKey').value;
+    const response = await fetch(url);
+    const data = await response.json();
+    return data.models
+        .filter(x => x.supportedGenerationMethods.includes("generateContent"))
+        .filter(x => {
+            return x.name.match(/^models\/gemini/);
+        });
+}
+
 async function proofRead(textarea) {
     let content = textarea.value;
     const ENDPOINT = `https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro-002:generateContent?key=${document.getElementById('geminiApiKey').value}`;
@@ -33,11 +51,37 @@ async function proofRead(textarea) {
         headers: {},
         body: JSON.stringify({
             contents: [{ parts: [{ text: prompt }] }],
+            generationConfig: {
+                "temperature": 1.0,
+                "max_output_tokens": 4096
+            },
+            safetySettings: [
+                {
+                    "category": "HARM_CATEGORY_HATE_SPEECH",
+                    "threshold": "BLOCK_NONE"
+                },
+                {
+                    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                    "threshold": "BLOCK_NONE"
+                },
+                {
+                    "category": "HARM_CATEGORY_HARASSMENT",
+                    "threshold": "BLOCK_NONE"
+                },
+                {
+                    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                    "threshold": "BLOCK_NONE"
+                },
+                {
+                    "category": "HARM_CATEGORY_CIVIC_INTEGRITY",
+                    "threshold": "BLOCK_NONE"
+                }
+            ]
         })
     };
     let proofReadText;
     const response = await fetch(ENDPOINT, payload);
-    try{
+    try {
         const data = await response.json();
         proofReadText = data.candidates[0].content.parts[0].text;
     } catch (error) {
@@ -91,7 +135,25 @@ async function summerize(text) {
         headers: {},
         body: JSON.stringify({
             contents: [{ parts: [{ text: prompt }] }],
-            generationConfig: { temperature: 0.7, max_output_tokens: 256 }
+            generationConfig: { temperature: 0.7, max_output_tokens: 256 },
+            safetySettings: [
+                {
+                    "category": "HARM_CATEGORY_HATE_SPEECH",
+                    "threshold": "BLOCK_NONE"
+                },
+                {
+                    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                    "threshold": "BLOCK_NONE"
+                },
+                {
+                    "category": "HARM_CATEGORY_HARASSMENT",
+                    "threshold": "BLOCK_NONE"
+                },
+                {
+                    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                    "threshold": "BLOCK_NONE"
+                }
+            ]
         })
     };
     try {
@@ -222,7 +284,7 @@ function loadFromUserStorage() {
     const savedData = localStorage.getItem('geminiClient');
     if (savedData) {
         const geminiClientData = JSON.parse(savedData);
-        Object.keys(geminiClientData).forEach(key => {
+        Object.keys(geminiClientData).filter(key => key != "endpointSelect").forEach(key => {
             const elem = document.getElementById(key);
             if (elem) {
                 if (elem.type === 'checkbox') {
@@ -242,6 +304,17 @@ function loadFromUserStorage() {
                 console.debug(`要素が見つかりません: ${key}`);
             }
         });
+        getModelList().then(models => {
+            const endpointSelect = document.getElementById('endpointSelect');
+            endpointSelect.innerHTML = '';
+            models.forEach(model => {
+                const option = document.createElement('option');
+                option.value = model.name;
+                option.textContent = model.name;
+                endpointSelect.appendChild(option);
+            });
+            endpointSelect.value = geminiClientData.endpointSelect;
+        });
     }
 }
 
@@ -316,7 +389,7 @@ function createPayload() {
         headers: {},
         body: JSON.stringify({
             contents: messages,
-            "generationConfig": {
+            generationConfig: {
                 "temperature": 1.0,
                 "max_output_tokens": 4096
             },
@@ -1162,6 +1235,34 @@ function updateAllAccordionHeaderCounts() {
     accordionIds.forEach(updateAccordionHeaderCount);
 }
 
+function showDiffModal() {
+    const diffContainer = document.getElementById('diffContainer');
+    diffContainer.innerHTML = '';
+
+    if (replaceProofReadHistory.length === 0) {
+        diffContainer.innerHTML = '<p>校正履歴がありません。</p>';
+        return;
+    }
+
+    replaceProofReadHistory.forEach((entry, index) => {
+        const [original, corrected] = entry;
+        const diff = Diff.diffLines(original.join('\n'), corrected.join('\n'));
+
+        const diffHtml = diff.map(part => {
+            const color = part.added ? 'green' : part.removed ? 'red' : 'grey';
+            const prefix = part.added ? '+' : part.removed ? '-' : ' ';
+            return `<span style="color: ${color}">${prefix} ${part.value}</span>`;
+        }).join('');
+
+        const entryDiv = document.createElement('div');
+        entryDiv.innerHTML = `<h6>校正 ${index + 1}</h6><pre>${diffHtml}</pre><hr>`;
+        diffContainer.appendChild(entryDiv);
+    });
+
+    const modal = new bootstrap.Modal(document.getElementById('diffModal'));
+    modal.show();
+}
+
 document.addEventListener('DOMContentLoaded', function () {
     // ページ読み込み時にデータを復元
     loadFromUserStorage();
@@ -1227,4 +1328,9 @@ document.addEventListener('DOMContentLoaded', function () {
     updateNavbarBrand();
     generateIndexMenu(true);
     updateAllAccordionHeaderCounts();
+
+    // diff2htmlライブラリの読み込み
+    const script = document.createElement('script');
+    script.src = 'https://cdnjs.cloudflare.com/ajax/libs/jsdiff/5.1.0/diff.min.js';
+    document.head.appendChild(script);
 });