AI FEATURE FLAG

Author: panaC

src/main/redux/store/memory.ts

@@ -15,7 +15,7 @@ import { reduxSyncMiddleware } from "readium-desktop/main/redux/middleware/sync"
 import { rootReducer } from "readium-desktop/main/redux/reducers";
 import { rootSaga } from "readium-desktop/main/redux/sagas";
 import { PersistRootState, RootState } from "readium-desktop/main/redux/states";
-import { IS_DEV } from "readium-desktop/preprocessor-directives";
+import { IS_AI_FEATURE, IS_DEV } from "readium-desktop/preprocessor-directives";
 import { tryCatch, tryCatchSync } from "readium-desktop/utils/tryCatch";
 import { applyMiddleware, legacy_createStore as createStore, type Store } from "redux";
 import createSagaMiddleware, { SagaMiddleware } from "redux-saga";
@@ -30,8 +30,11 @@ import { clone } from "ramda";
 import { TBookmarkState } from "readium-desktop/common/redux/states/bookmark";
 import { TAnnotationState } from "readium-desktop/common/redux/states/renderer/annotation";
 
-// TODO: DEBUG ONLY AISDK
-import "dotenv/config";
+
+if (IS_AI_FEATURE) {
+    // .env AI API KEY":
+    require("dotenv/config");
+}
 
 // import { composeWithDevTools } from "remote-redux-devtools";
 const REDUX_REMOTE_DEVTOOLS_PORT = 7770;
@@ -438,29 +441,33 @@ export async function initStore()
         preloadedState.reader.defaultConfig = { ...readerConfigInitialState, ...preloadedState.reader.defaultConfig };
     }
 
-    if (preloadedState?.aiApiKeys) {
-        for (const aiKey of preloadedState.aiApiKeys) {
-            if (aiKey.provider === "openAI" && aiKey.aiKey) {
-                process.env["OPENAI_API_KEY"] = aiKey.aiKey;
-            } else if (aiKey.provider === "mistralAI" && aiKey.aiKey) {
-                process.env["MISTRAL_API_KEY"] = aiKey.aiKey;
-            } else if (aiKey.provider === "geminiAI" && aiKey.aiKey) {
-                process.env["GOOGLE_GENERATIVE_AI_API_KEY"] = aiKey.aiKey;
+    if (IS_AI_FEATURE) {
+
+        if (preloadedState?.aiApiKeys) {
+            for (const aiKey of preloadedState.aiApiKeys) {
+                if (aiKey.provider === "openAI" && aiKey.aiKey) {
+                    process.env["OPENAI_API_KEY"] = aiKey.aiKey;
+                } else if (aiKey.provider === "mistralAI" && aiKey.aiKey) {
+                    process.env["MISTRAL_API_KEY"] = aiKey.aiKey;
+                } else if (aiKey.provider === "geminiAI" && aiKey.aiKey) {
+                    process.env["GOOGLE_GENERATIVE_AI_API_KEY"] = aiKey.aiKey;
+                }
             }
         }
-    }
-    {
-        preloadedState.aiApiKeys = [];
-        if (process.env["OPENAI_API_KEY"]) {
-            preloadedState.aiApiKeys.push({ provider: "openAI", aiKey: process.env["OPENAI_API_KEY"] });
-        }
-        if (process.env["MISTRAL_API_KEY"]) {
-            preloadedState.aiApiKeys.push({ provider: "mistralAI", aiKey: process.env["MISTRAL_API_KEY"] });
-        }
-        if (process.env["GOOGLE_GENERATIVE_AI_API_KEY"]) {
-            preloadedState.aiApiKeys.push({ provider: "geminiAI", aiKey: process.env["GOOGLE_GENERATIVE_AI_API_KEY"] });
+        {
+            preloadedState.aiApiKeys = [];
+            if (process.env["OPENAI_API_KEY"]) {
+                preloadedState.aiApiKeys.push({ provider: "openAI", aiKey: process.env["OPENAI_API_KEY"] });
+            }
+            if (process.env["MISTRAL_API_KEY"]) {
+                preloadedState.aiApiKeys.push({ provider: "mistralAI", aiKey: process.env["MISTRAL_API_KEY"] });
+            }
+            if (process.env["GOOGLE_GENERATIVE_AI_API_KEY"]) {
+                preloadedState.aiApiKeys.push({ provider: "geminiAI", aiKey: process.env["GOOGLE_GENERATIVE_AI_API_KEY"] });
+            }
         }
     }
+
     if (preloadedState?.creator && !preloadedState.creator.urn) {
         preloadedState.creator.urn = `urn:uuid:${preloadedState.creator.id}`;
     }

src/main/streamer/streamerNoHttp.ts

@@ -11,7 +11,7 @@ import { app, net, protocol, ProtocolRequest, ProtocolResponse, session } from "
 import * as fs from "fs";
 import * as mime from "mime-types";
 import * as path from "path";
-import { _PACKAGING, IS_DEV } from "readium-desktop/preprocessor-directives";
+import { _PACKAGING, IS_AI_FEATURE, IS_DEV } from "readium-desktop/preprocessor-directives";
 
 import { TaJsonSerialize } from "@r2-lcp-js/serializable";
 import { parseDOM, serializeDOM } from "@r2-navigator-js/electron/common/dom";
@@ -311,7 +311,7 @@ const streamProtocolHandler = async (
     headers["Access-Control-Allow-Headers"] = "Content-Type, Content-Length, Accept-Ranges, Content-Range, Range, Link, Transfer-Encoding, X-Requested-With, Authorization, Accept, Origin, User-Agent, DNT, Cache-Control, Keep-Alive, If-Modified-Since";
     headers["Access-Control-Expose-Headers"] = "Content-Type, Content-Length, Accept-Ranges, Content-Range, Range, Link, Transfer-Encoding, X-Requested-With, Authorization, Accept, Origin, User-Agent, DNT, Cache-Control, Keep-Alive, If-Modified-Since";
 
-    if (isAiSdk) {
+    if (IS_AI_FEATURE && isAiSdk) {
 
         // use specific debug instance
         debugAiSdk("AISDK request !!!");

src/renderer/library/components/settings/Settings.tsx

@@ -49,6 +49,7 @@ import { noteExportHtmlMustacheTemplate } from "readium-desktop/common/readium/a
 // import { TagGroup, TagList, Tag, Label } from "react-aria-components";
 import { VisuallyHidden } from "@radix-ui/react-visually-hidden";
 import { ApiKeysList } from "./AiKeyManager";
+import { IS_AI_FEATURE } from "readium-desktop/preprocessor-directives";
 
 interface ISettingsProps {};
 
@@ -447,12 +448,16 @@ export const Settings: React.FC<ISettingsProps> = () => {
                             <SVG ariaHidden svg={KeyReturnIcon} />
                             <h3>{__("settings.tabs.keyboardShortcuts")}</h3>
                         </Tabs.Trigger>
-                        <Tabs.Trigger value="tab5">
-                            <SVG ariaHidden svg={AiIcon} />
-                            <h3>{__("settings.tabs.aiKeyManager")}</h3>
-                        </Tabs.Trigger>
+                        {IS_AI_FEATURE
+                            ?
+                            <Tabs.Trigger value="tab5">
+                                <SVG ariaHidden svg={AiIcon} />
+                                <h3>{__("settings.tabs.aiKeyManager")}</h3>
+                            </Tabs.Trigger>
+                            : <></>
+                        }
                     </Tabs.List>
-                    <div className={stylesSettings.settings_content} style={{marginTop: "70px"}}>
+                    <div className={stylesSettings.settings_content} style={{ marginTop: "70px" }}>
                         <Tabs.Content value="tab1" tabIndex={-1}>
                             <TabHeader title={__("settings.tabs.general")} />
                             <div className={stylesSettings.settings_tab}>
@@ -478,13 +483,16 @@ export const Settings: React.FC<ISettingsProps> = () => {
                                 <KeyboardSettings />
                             </div>
                         </Tabs.Content>
-                        <Tabs.Content value="tab5" tabIndex={-1}>
-                            <TabHeader title={__("settings.tabs.aiKeyManager")}>
-                            </TabHeader>
-                            <div className={stylesSettings.settings_tab}>
-                                <ApiKeysList />
-                            </div>
-                        </Tabs.Content>
+                        {IS_AI_FEATURE ?
+                            <Tabs.Content value="tab5" tabIndex={-1}>
+                                <TabHeader title={__("settings.tabs.aiKeyManager")}>
+                                </TabHeader>
+                                <div className={stylesSettings.settings_tab}>
+                                    <ApiKeysList />
+                                </div>
+                            </Tabs.Content>
+                            : <></>
+                        }
                     </div>
                 </Tabs.Root>
 

src/renderer/reader/components/App.tsx

@@ -18,6 +18,7 @@ import * as React from "react";
 import { Provider } from "react-redux";
 import {
     _NODE_MODULE_RELATIVE_URL, _PACKAGING, _RENDERER_READER_BASE_URL,
+    IS_AI_FEATURE,
 } from "readium-desktop/preprocessor-directives";
 import ToastManager from "readium-desktop/renderer/common/components/toast/ToastManager";
 
@@ -32,6 +33,7 @@ import { getTranslator } from "readium-desktop/common/services/translator";
 import { getStore } from "../createStore";
 import { TranslatorContext } from "readium-desktop/renderer/common/translator.context";
 import { ImageClickManager } from "./ImageClickManagerWithAI";
+import { ImageClickManagerImgViewerOnly } from "./ImageClickManagerViewerOnly";
 // import { ImageClickManagerImgViewerOnly } from "./ImageClickManagerViewerOnly";
 
 export default class App extends React.Component<{}, undefined> {
@@ -239,7 +241,11 @@ url("${rcssPath}/fonts/iAWriterDuospace-Regular.ttf") format("truetype");
                 <TranslatorContext.Provider value={getTranslator()}>
                     <Reader />
                     <ToastManager />
-                    <ImageClickManager />
+                    {
+                        IS_AI_FEATURE
+                            ? <ImageClickManager />
+                            : <ImageClickManagerImgViewerOnly />
+                    }
                 </TranslatorContext.Provider>
             </Provider>
         );

webpack.config-preprocessor-directives.js

@@ -39,7 +39,7 @@ const telemetrySecret = process.env.THORIUM_TELEMETRY_SECRET || "";
 
 // const USE_HTTP_STREAMER = false;
 
-const isAIFeatureEnabled = true; 
+const isAIFeatureEnabled = false; 
 
 const data = {
     __APP_VERSION__: JSON.stringify(version),