Upload 2 files

gemini.js

@@ -224,8 +224,11 @@ function createSummarizedText() {
     if (rootUl) {
         processUl(rootUl);
     }
-
-    return summarizedText.trim();
+    if (summarizedText) {
+        return summarizedText.trim();
+    } else {
+        return document.getElementById('novelContent1').value;
+    }
 }
 
 function createPayload() {
@@ -292,9 +295,37 @@ function debugPrompt() {
     });
 }
 
+// 新しい関数を追加
+function updateRequestButtonState(state, flashClass = null) {
+    const requestButton = document.getElementById('requestButton');
+    const stopButton = document.getElementById('stopButton');
+
+    switch (state) {
+        case 'generating':
+            requestButton.disabled = true;
+            stopButton.classList.remove('d-none');
+            break;
+        case 'idle':
+            requestButton.disabled = false;
+            stopButton.classList.add('d-none');
+            break;
+        case 'error':
+            requestButton.disabled = false;
+            stopButton.classList.add('d-none');
+            break;
+    }
+
+    if (flashClass) {
+        requestButton.classList.add(flashClass);
+        setTimeout(() => {
+            requestButton.classList.remove(flashClass);
+        }, 2000);
+    }
+}
+
 function fetchStream(ENDPOINT, payload) {
     const novelContent2 = document.getElementById('novelContent2');
-    const requestButton = document.getElementById('requestButton');
+    updateRequestButtonState('generating');
     controller = new AbortController();
     const signal = controller.signal;
 
@@ -382,11 +413,11 @@ function fetchStream(ENDPOINT, payload) {
                 }).catch(error => {
                     if (error.name === 'AbortError') {
                         console.log('フェッチがユーザーによって中止されました');
+                        updateRequestButtonState('idle');
                     } else {
                         console.error('ストリーム読み取りエラー:', error);
+                        updateRequestButtonState('error', 'red-flash-bg');
                     }
-                    document.getElementById('stopButton').classList.add('d-none');
-                    requestButton.disabled = false;
                 });
             }
 
@@ -395,29 +426,31 @@ function fetchStream(ENDPOINT, payload) {
         .catch(error => {
             if (error.name === 'AbortError') {
                 console.log('フェッチがユーザーよって中止されました');
+                updateRequestButtonState('idle');
             } else {
                 console.error('フェッチエラー:', error);
+                updateRequestButtonState('error', 'red-flash-bg');
             }
-            requestButton.disabled = false;
         });
 }
 
-function fetchNonStream(ENDPOINT, payload) {
+async function fetchNonStream(ENDPOINT, payload) {
     const novelContent2 = document.getElementById('novelContent2');
-    fetch(ENDPOINT, payload)
-        .then(response => response.json())
-        .then(data => {
-            if (data && data.candidates && data.candidates[0].content && data.candidates[0].content.parts && data.candidates[0].content.parts[0].text) {
-                novelContent2.value += data.candidates[0].content.parts[0].text;
-                novelContent2.scrollTop = novelContent2.scrollHeight;
-            }
-        })
-        .catch(error => {
-            console.error('エラー:', error);
-        })
-        .finally(() => {
-            document.getElementById('requestButton').disabled = false;
-        });
+    updateRequestButtonState('generating');
+    try {
+        const response = await fetch(ENDPOINT, payload);
+        const data = await response.json();
+        if (data && data.candidates && data.candidates[0].content && data.candidates[0].content.parts && data.candidates[0].content.parts[0].text) {
+            novelContent2.value += data.candidates[0].content.parts[0].text;
+            novelContent2.scrollTop = novelContent2.scrollHeight;
+            updateRequestButtonState('idle', 'green-flash-bg');
+        } else {
+            throw new Error('予期しないレスポンス形式');
+        }
+    } catch (error) {
+        console.error('エラー:', error);
+        updateRequestButtonState('error', 'red-flash-bg');
+    }
 }
 
 function createOpenAIPayload() {
@@ -464,7 +497,7 @@ function createOpenAIPayload() {
 
 function fetchOpenAIStream(ENDPOINT, payload) {
     const novelContent2 = document.getElementById('novelContent2');
-    const requestButton = document.getElementById('requestButton');
+    updateRequestButtonState('generating');
     controller = new AbortController();
     const signal = controller.signal;
 
@@ -520,11 +553,11 @@ function fetchOpenAIStream(ENDPOINT, payload) {
                 }).catch(error => {
                     if (error.name === 'AbortError') {
                         console.log('フェッチがユーザーによって中止されました');
+                        updateRequestButtonState('idle');
                     } else {
                         console.error('ストリーム読み取りエラー:', error);
+                        updateRequestButtonState('error', 'red-flash-bg');
                     }
-                    document.getElementById('stopButton').classList.add('d-none');
-                    requestButton.disabled = false;
                 });
             }
 
@@ -533,33 +566,35 @@ function fetchOpenAIStream(ENDPOINT, payload) {
         .catch(error => {
             if (error.name === 'AbortError') {
                 console.log('フェッチがユーザーよって中止されました');
+                updateRequestButtonState('idle');
             } else {
                 console.error('フェッチエラー:', error);
+                updateRequestButtonState('error', 'red-flash-bg');
             }
-            requestButton.disabled = false;
         });
 }
 
-function fetchOpenAINonStream(ENDPOINT, payload) {
+async function fetchOpenAINonStream(ENDPOINT, payload) {
     const novelContent2 = document.getElementById('novelContent2');
-    fetch(ENDPOINT, {
-        ...payload,
-        mode: 'cors',  // CORSモードを追加
-        credentials: 'same-origin'  // 必要に応じて認証情報を含める
-    })
-        .then(response => response.json())
-        .then(data => {
-            if (data && data.choices && data.choices[0].message && data.choices[0].message.content) {
-                novelContent2.value += data.choices[0].message.content;
-                novelContent2.scrollTop = novelContent2.scrollHeight;
-            }
-        })
-        .catch(error => {
-            console.error('エラー:', error);
-        })
-        .finally(() => {
-            document.getElementById('requestButton').disabled = false;
+    updateRequestButtonState('generating');
+    try {
+        const response = await fetch(ENDPOINT, {
+            ...payload,
+            mode: 'cors',
+            credentials: 'same-origin'
         });
+        const data = await response.json();
+        if (data && data.choices && data.choices[0].message && data.choices[0].message.content) {
+            novelContent2.value += data.choices[0].message.content;
+            novelContent2.scrollTop = novelContent2.scrollHeight;
+            updateRequestButtonState('idle', 'green-flash-bg');
+        } else {
+            throw new Error('予期しないレスポンス形式');
+        }
+    } catch (error) {
+        console.error('エラー:', error);
+        updateRequestButtonState('error', 'red-flash-bg');
+    }
 }
 
 async function tokenCount() {
@@ -693,8 +728,7 @@ function stopGeneration() {
         controller.abort();
         controller = null;
     }
-    document.getElementById('stopButton').classList.add('d-none');
-    document.getElementById('requestButton').disabled = false;
+    updateRequestButtonState('idle');
 }
 
 // 新しい関数を追加
@@ -907,7 +941,7 @@ function generateIndexMenu() {
     }
 
     updateAllAccordionHeaderCounts();
-    updateTokenCount(true); // トークン数を強制更新
+    updateTokenCount(false);
 }
 
 function toggleSubMenu(li) {
@@ -1050,7 +1084,6 @@ document.addEventListener('DOMContentLoaded', function () {
         document.getElementById(id).addEventListener('input', () => {
             saveToUserStorage(false);
             generateIndexMenu();
-            updateTokenCount(); // 60秒に1回の制限が適用される
         });
     });
 
@@ -1059,7 +1092,6 @@ document.addEventListener('DOMContentLoaded', function () {
         document.getElementById(id).addEventListener('input', () => {
             saveToUserStorage(true);
             generateIndexMenu();
-            updateTokenCount(); // 60秒に1回の制限が適用される
         });
     });
 
@@ -1067,7 +1099,6 @@ document.addEventListener('DOMContentLoaded', function () {
         document.getElementById(id).addEventListener('change', () => {
             saveToUserStorage(true);
             generateIndexMenu();
-            updateTokenCount(); // 60秒に1回の制限が適用される
         });
     });