Updated files.

Author: CodeRTX

src/main.ts

@@ -13,6 +13,11 @@ import { SplitLayout } from "@googlemaps/extended-component-library/split_layout
 import Base64 from "base64-js";
 import "./style.css";
 
+// Add a delay function here
+function delay(ms: number) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}
+
 // TODO: Get a Gemini API key:
 /*
  * 1. Open the Project IDX view by pressing Ctrl+Shift+P / Cmd+Shift+P and type "IDX focus", then select "IDX: Focus on Project IDX View"
@@ -95,7 +100,7 @@ form.addEventListener("submit", async (e) => {
     // Call the gemini-pro-vision model, and get a stream of results
     const genAI = new GoogleGenerativeAI(GEMINI_API_KEY);
     const model = genAI.getGenerativeModel({
-      model: "gemini-pro-vision",
+      model: "gemini-2.5-flash",
       safetySettings: [
         {
           category: HarmCategory.HARM_CATEGORY_HARASSMENT,
@@ -104,6 +109,8 @@ form.addEventListener("submit", async (e) => {
       ],
     });
 
+    await delay(60000); // This will wait for 60 seconds
+
     // Read from the stream and render the output
     const result = await model.generateContentStream({ contents });
     for await (const response of result.stream) {

src/travel.js

@@ -84,10 +84,8 @@ async function generateReviewTitle() {
     outputHTML.innerHTML = `<h2>${promptTitle.value}</h2><p>${reviewText}</p>`;
   } else {
     // Fallback to Gemini API
-    const genAI = new GoogleGenerativeAI(
-      "AIzaSyA8bYErDRE8xfRAIvZJkL06l0oMp5wHVx8"
-    );
-    const model = await genAI.getGenerativeModel({ model: "gemini-pro" });
+    const genAI = new GoogleGenerativeAI(GEMINI_API_KEY);
+    const model = await genAI.getGenerativeModel({ model: "gemini-2.5-flash" });
     const result = await model.generateContent([
       `Generate a catchy title for this travel review: ${reviewText} to ${placeName}`,
     ]);
@@ -163,10 +161,8 @@ async function checkToxicity() {
     // outputHTML.innerHTML = (`Tone Score: ${toxicityScore}`); // Or display in a more user-friendly way
 
     // Fallback to Gemini API
-    const genAI = new GoogleGenerativeAI(
-      "AIzaSyA8bYErDRE8xfRAIvZJkL06l0oMp5wHVx8"
-    );
-    const model = await genAI.getGenerativeModel({ model: "gemini-pro" });
+    const genAI = new GoogleGenerativeAI(GEMINI_API_KEY);
+    const model = await genAI.getGenerativeModel({ model: "gemini-2.5-flash" });
     const result = await model.generateContent([
       `Analyze the toxicity level of this text:: ${reviewText}`,
     ]);
@@ -199,10 +195,8 @@ async function runBuiltInAIModel(streaming) {
 
 // Function to run the Gemini API model
 async function runGeminiAPIModel(streaming) {
-  const genAI = new GoogleGenerativeAI(
-    "AIzaSyA8bYErDRE8xfRAIvZJkL06l0oMp5wHVx8"
-  );
-  const model = await genAI.getGenerativeModel({ model: "gemini-pro" });
+  const genAI = new GoogleGenerativeAI(GEMINI_API_KEY);
+  const model = await genAI.getGenerativeModel({ model: "gemini-2.5-flash" });
   outputHTML.innerHTML = "Generating answer...Please wait...";
 
   if (streaming) {