Update app/api/ask-ai/route.ts

app/api/ask-ai/route.ts

@@ -1,14 +1,12 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 import type { NextRequest } from "next/server";
 import { NextResponse } from "next/server";
-import { headers } from "next/headers";
 import { GoogleGenerativeAI } from "@google/generative-ai";
 
 import {
   DIVIDER,
   FOLLOW_UP_SYSTEM_PROMPT,
   INITIAL_SYSTEM_PROMPT,
-  MAX_REQUESTS_PER_IP,
   NEW_PAGE_END,
   NEW_PAGE_START,
   REPLACE_END,
@@ -16,11 +14,8 @@ import {
   UPDATE_PAGE_START,
   UPDATE_PAGE_END,
 } from "@/lib/prompts";
-import MY_TOKEN_KEY from "@/lib/get-cookie-name";
 import { Page } from "@/types";
 
-const ipAddresses = new Map();
-
 // Gemini Models Configuration
 const GEMINI_MODELS = {
   "gemini-2.5-flash": "gemini-2.5-flash",
@@ -30,9 +25,6 @@ const GEMINI_MODELS = {
 };
 
 export async function POST(request: NextRequest) {
-  const authHeaders = await headers();
-  const userToken = request.cookies.get(MY_TOKEN_KEY())?.value;
-
   const body = await request.json();
   const { prompt, model, redesignMarkdown, previousPrompts, pages } = body;
 
@@ -55,24 +47,7 @@ export async function POST(request: NextRequest) {
   // Validate model
   const selectedModel = GEMINI_MODELS[model as keyof typeof GEMINI_MODELS] || "gemini-2.5-flash";
 
-  // Rate limiting for non-authenticated users
-  const ip = authHeaders.get("x-forwarded-for")?.includes(",")
-    ? authHeaders.get("x-forwarded-for")?.split(",")[1].trim()
-    : authHeaders.get("x-forwarded-for");
-
-  if (!userToken) {
-    ipAddresses.set(ip, (ipAddresses.get(ip) || 0) + 1);
-    if (ipAddresses.get(ip) > MAX_REQUESTS_PER_IP) {
-      return NextResponse.json(
-        {
-          ok: false,
-          openLogin: true,
-          message: "Log In to continue using the service",
-        },
-        { status: 429 }
-      );
-    }
-  }
+  // Rate limiting and login check has been removed.
 
   try {
     const encoder = new TextEncoder();
@@ -158,9 +133,6 @@ export async function POST(request: NextRequest) {
 }
 
 export async function PUT(request: NextRequest) {
-  const authHeaders = await headers();
-  const userToken = request.cookies.get(MY_TOKEN_KEY())?.value;
-
   const body = await request.json();
   const { prompt, previousPrompts, selectedElementHtml, model, pages, files } = body;
 
@@ -183,24 +155,7 @@ export async function PUT(request: NextRequest) {
   // Validate model
   const selectedModel = GEMINI_MODELS[model as keyof typeof GEMINI_MODELS] || "gemini-2.5-flash";
 
-  // Rate limiting for non-authenticated users
-  const ip = authHeaders.get("x-forwarded-for")?.includes(",")
-    ? authHeaders.get("x-forwarded-for")?.split(",")[1].trim()
-    : authHeaders.get("x-forwarded-for");
-
-  if (!userToken) {
-    ipAddresses.set(ip, (ipAddresses.get(ip) || 0) + 1);
-    if (ipAddresses.get(ip) > MAX_REQUESTS_PER_IP) {
-      return NextResponse.json(
-        {
-          ok: false,
-          openLogin: true,
-          message: "Log In to continue using the service",
-        },
-        { status: 429 }
-      );
-    }
-  }
+  // Rate limiting and login check has been removed.
 
   try {
     const genAI = new GoogleGenerativeAI(geminiApiKey);
@@ -393,7 +348,6 @@ export async function PUT(request: NextRequest) {
           position = replaceEndIndex + REPLACE_END.length;
         }
 
-        // Update the main HTML if it's the index page
         const mainPageIndex = updatedPages.findIndex(p => p.path === '/' || p.path === '/index' || p.path === 'index');
         if (mainPageIndex !== -1) {
           updatedPages[mainPageIndex].html = newHtml;