Update app.py

app.py

@@ -20,9 +20,12 @@ app.add_middleware(
 
 OLLAMA_LIBRARY_URL = "https://ollama.com/library"
 
-RATE_LIMIT = 25
+RATE_LIMIT = 50
 WINDOW_SECONDS = 60 * 60 * 24
 ip_store = {}  # { ip: { "count": int, "reset": timestamp } }
+AUDIO_RATE_LIMIT = 4
+AUDIO_WINDOW_SECONDS = 60 * 60 * 24
+audio_ip_store = {}
 
 REASONING_KEYWORDS = [
     # explicit reasoning requests
@@ -195,6 +198,29 @@ def extract_user_text(messages: list) -> str:
         if m.get("role") == "user"
     ).lower()
 
+def check_audio_rate_limit(ip: str):
+    now = time.time()
+
+    if ip not in audio_ip_store:
+        audio_ip_store[ip] = {
+            "count": 0,
+            "reset": now + AUDIO_WINDOW_SECONDS
+        }
+
+    entry = audio_ip_store[ip]
+
+    if now > entry["reset"]:
+        entry["count"] = 0
+        entry["reset"] = now + AUDIO_WINDOW_SECONDS
+
+    if entry["count"] >= AUDIO_RATE_LIMIT:
+        raise HTTPException(
+            status_code=429,
+            detail="Daily audio limit reached: 4 per IP"
+        )
+
+    entry["count"] += 1
+
 
 def is_complex_reasoning(prompt: str) -> bool:
     if len(prompt) > 800:
@@ -299,11 +325,14 @@ def check_chat_rate_limit(ip: str):
     entry["count"] += 1
     return entry["count"]
 
+@app.post("/gen/image")
 @app.get("/genimg/{prompt}")
-async def generate_image(prompt: str, request: Request):
+async def generate_image(prompt: str = None, request: Request):
     client_ip = request.client.host
     check_rate_limit(client_ip)
-
+    timeout = httpx.Timeout(300.0, read=300.0)
+    if prompt is None:
+        prompt = (await request.json()).prompt
     if is_cinematic_image_prompt(prompt):
         chosen_model = "flux"
     else:
@@ -313,7 +342,7 @@ async def generate_image(prompt: str, request: Request):
 
     url = f"https://gen.pollinations.ai/image/{prompt}?model={chosen_model}&key={PKEY}"
 
-    async with httpx.AsyncClient() as client:
+    async with httpx.AsyncClient(timeout = timeout) as client:
         response = await client.get(url)
 
     if response.status_code != 200:
@@ -494,3 +523,16 @@ async def generate_text(request: Request):
         )
 
     raise HTTPException(500, "Unknown provider routing error")
+
+@app.post("/gen/sfx")
+async def gensfx(request: Request):
+    client_ip = request.client.host
+    check_audio_rate_limit(client_ip)
+    prompt = (await request.json()).prompt
+    url = f"https://gen.pollinations.ai/audio/{prompt}?model=elevenaudio&key={PKEY}"
+    async with httpx.AsyncClient(timeout=None) as client:
+        response = await client.get(url)
+    return Response(
+        response.content
+        media_type="audio/mpeg"
+    )