Merge pull request #12 from Saarthakkj/gemini

Gemini integeration

Author: Saarthakkj

extension/background.js

@@ -1,19 +1,18 @@
-import { GoogleGenerativeAI, system_prompt } from './dist/generative-ai-bundle.js';
+import {GoogleGenerativeAI , system_prompt} from './dist/geneartive-ai-bundle.js';
 
 let genModel = null;
 
 // Initialize the model
 async function initializeModel() {
     try {
-        const GEMINI_API_KEY = "AIzaSyARc1EeyqVha3psd2ZSazZNvUYTCC8RuNA";
-        
+        const result = await chrome.storage.sync.get(['GEMINI_API_KEY']);
+        const GEMINI_API_KEY = result.GEMINI_API_KEY;
         if (!GEMINI_API_KEY) {
             throw new Error('API key not found');
         }
-
         const genai = new GoogleGenerativeAI(GEMINI_API_KEY);
         genModel = genai.getGenerativeModel({
-            model: "gemini-1.5-flash",
+            model: "gemini-2.0-flash-lite-preview-02-05",
             systemInstruction: system_prompt
         });
         console.log("[background.js] Gemini model initialized successfully");
@@ -68,8 +67,14 @@ chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
         return true; // Keep the message channel open for sendResponse
     }
 });
-// ... existing code ...
+
+
 
 // Initialize on install and startup
-chrome.runtime.onInstalled.addListener(initializeModel);
-chrome.runtime.onStartup.addListener(initializeModel);
+chrome.runtime.onStartup.addListener(initializeModel);
+
+//! store your api key running this permanently in your browser:
+
+// chrome.storage.sync.set({ GEMINI_API_KEY: 'your_api_key_here' }, () => {
+//     console.log('API key saved to Chrome storage');
+// });