rate limit

app.py

@@ -2,6 +2,8 @@ import os
 import logging
 from groq import Groq
 from flask import Flask, render_template_string, request, Response, stream_with_context
+from flask_limiter import Limiter
+from flask_limiter.util import get_remote_address
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -11,6 +13,14 @@ app = Flask(__name__)
 # Set up logging
 logging.basicConfig(level=logging.DEBUG)
 
+# Set up rate limiter
+limiter = Limiter(
+    get_remote_address,
+    app=app,
+    default_limits=["5 per minute"],
+    storage_uri="memory://"
+)
+
 # Set Groq API key
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
@@ -171,6 +181,7 @@ def home():
     return render_template_string(html_template)
 
 @app.route("/generate", methods=["POST"])
+@limiter.limit("5 per minute")
 def generate():
     parent_name = request.form["parent_name"]
     child_name = request.form["child_name"]