Update main.py

main.py

@@ -594,8 +594,7 @@ async def get_completion(payload: Payload, request: Request, authenticated: bool
 @lru_cache(maxsize=256)
 def cached_url(url: str):
     return url
-    
-@app.get("/images/{prompt:path}")
+    @app.get("/images/{prompt:path}")
 async def create_image(
     prompt: str,
     request: Request,
@@ -604,13 +603,15 @@ async def create_image(
     if not server_status:
         raise HTTPException(status_code=503, detail="Server is under maintenance.")
 
-    query = request.url.query
-
+    # Build base URL safely
     base = "https://image.pollinations.ai/prompt/"
-    final_url = f"{base}{prompt}"
-    if query:
-        final_url += f"?{query}"
+    final_url = f"{base}{prompt}?nologo=true"
 
+    # If user provided custom query params, append them
+    if request.url.query:
+        final_url += f"&{request.url.query}"
+
+    # Apply caching wrapper
     final_url = cached_url(final_url)
 
     try:
@@ -629,7 +630,8 @@ async def create_image(
         raise HTTPException(status_code=504, detail="Image generation timeout.")
 
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Unexpected error: {e}")
+        raise HTTPException(status_code=500, detail=f"Unexpected error: {str(e)}")
+
 @app.get("/usage")
 async def get_usage_json(days: int = 7):
     return usage_tracker.get_usage_summary(days)