feature: special command for image generation

Author: EliasPereirah

README.md

@@ -69,6 +69,14 @@ Some companies offer free API access. Check their terms and conditions before yo
 # Special Commands
 Use special commands to perform an action quickly and easily.
 
+### Image generation with Google Gemini
+
+- `pic: image of a dog having fun on the beach`
+- `imagine: image of a cat playing with a basket`
+
+Both commands above are equivalent, by using "imagine:" or "pic" you will be sending your prompt to the Google Gemini image generation model. 
+It will only work when the currently active model is any of Google Gemini.
+
 ###  Grounding with Google Search 
 - `g: what's the main news today`
 

index.html

@@ -75,9 +75,9 @@
 <script src="js/md5.js?v=1.0.1"></script>
 <script src="js/google_cse.js?v=1.1.1"></script>
 <script src="js/4devs.js?v=1.0.1"></script>
-<script src="js/prompts.js?v=1.0.8"></script>
+<script src="js/prompts.js?v=1.1.0"></script>
 <script src="js/tools_list.js?v=1.2.0"></script>
-<script src="js/script.js?v=3.2.2"></script>
+<script src="js/script.js?v=3.2.3"></script>
 <script src="js/search_chats.js?v=0.0.3"></script>
 <script src="js/voice_rec.js?v=1.0.6"></script>
 <script src="js/nuggets.js?v=0.0.1"></script>

js/prompts.js

@@ -1086,6 +1086,7 @@ function createDialog(text, duration_seconds = 0, add_class_name = '', can_delet
 }
 
 function geminiChat(fileUri = '', with_stream = true, the_data = '') {
+    let gemini_model = model;
     let all_parts = [];
     let system_prompt = getSystemPrompt();
     addFileToPrompt();
@@ -1179,18 +1180,23 @@ function geminiChat(fileUri = '', with_stream = true, the_data = '') {
         // "maxOutputTokens": 8192,
     };
 
-   if(model.includes("image")){
-       with_stream = false;
-       if(data.systemInstruction){
-           delete data.systemInstruction;
-       }
-       data.generationConfig.responseModalities = ["IMAGE", "TEXT"];
-   }
 
     let pog = whichTool(last_user_input);
     if(pog === 'dt'){
         data.generationConfig.thinkingConfig =  { thinkingBudget: 8000 };
     }
+    if(pog === 'pic' || pog === 'imagine'){
+        console.log('image generation command activated');
+        gemini_model = "gemini-2.0-flash-preview-image-generation";
+    }
+
+    if(gemini_model.includes("image")){
+        with_stream = false;
+        if(data.systemInstruction){
+            delete data.systemInstruction;
+        }
+        data.generationConfig.responseModalities = ["IMAGE", "TEXT"];
+    }
 
     if (the_data) {
         data = the_data;
@@ -1230,7 +1236,7 @@ function geminiChat(fileUri = '', with_stream = true, the_data = '') {
         return geminiStreamChat(fileUri, data, allow_tool_use);
     }
 
-    let gemini_endpoint = endpoint.replaceAll("{{model}}", model);
+    let gemini_endpoint = endpoint.replaceAll("{{model}}", gemini_model);
     gemini_endpoint = gemini_endpoint.replaceAll("{{api_key}}", api_key);
     gemini_endpoint = gemini_endpoint.replaceAll("{{gen_mode}}", "generateContent");
 
@@ -2009,7 +2015,8 @@ function needToolUse(last_user_input) {
         'search:', 's:',
         'javascript:', 'js:',
         'youtube:', 'yt:',
-        'dt:'
+        'dt:',
+        'pic:','imagine:'
     ]
     if (cmd_list.includes(cmd)) {
         return true;
@@ -2038,6 +2045,8 @@ function whichTool(last_user_input) {
     }else if(cmd === 'dt:'){
         // Activate extend thinking for Claude
         return 'dt';
+    }else if(cmd.trim() !== ''){
+        return cmd.replaceAll(":","");
     }
     return '';
 }