ai: Enable authentication for Pollinations AI.

Required for audio and image generation.

* Goodbye anonymous.

src/tools/audio.py

@@ -40,7 +40,7 @@ class AudioGeneration:
         generate_audio_instruct = quote(generate_audio_instruction)
 
         # Initialize tools and extract the audio generation service endpoint (third element in the returned tuple)
-        _, _, audio_tool = initialize_tools()
+        _, _, audio_tool, poll_token = initialize_tools()
 
         # Construct the full URL by appending the encoded instruction to the base audio tool URL
         url = f"{audio_tool}/{generate_audio_instruct}"
@@ -57,6 +57,7 @@ class AudioGeneration:
             while True:
                 # Generate a random IP address to spoof the client's origin in the request headers
                 headers = {
+                    "Authorization": f"Bearer {poll_token}",  # Pollinations AI API Token
                     "X-Forwarded-For": generate_ip()  # Set the X-Forwarded-For header to a random IP address
                 }
 
@@ -92,7 +93,7 @@ class AudioGeneration:
         generate_audio_instruct = quote(generate_audio_instruction)
 
         # Initialize tools and get audio generation endpoint
-        _, _, audio_tool = initialize_tools()
+        _, _, audio_tool, poll_token = initialize_tools()
 
         # Construct request URL
         url = f"{audio_tool}/{generate_audio_instruct}"
@@ -109,6 +110,7 @@ class AudioGeneration:
             while True:
               # Define HTTP headers for the request, including random IP address to simulate different client origins
                 headers = {
+                    "Authorization": f"Bearer {poll_token}",  # Pollinations AI API Token
                     "X-Forwarded-For": generate_ip()  # Generate and set a random IP address for the request header
                 }
 

src/tools/image.py

@@ -73,7 +73,7 @@ class ImageGeneration:
         width, height = ImageGeneration.FORMATS[image_format]
 
         # Initialize tools and get image generation service endpoint URL
-        _, image_tool, _ = initialize_tools()
+        _, image_tool, poll_token, _ = initialize_tools()
 
         # Encode instruction safely for URL path usage
         generate_image_instruct = quote(generate_image_instruction)
@@ -97,6 +97,7 @@ class ImageGeneration:
 
         # Prepare headers
         headers = {
+            "Authorization": f"Bearer {poll_token}",  # Pollinations AI API Token
             "X-Forwarded-For": generate_ip()  # Random IP address for request header to simulate client origin
         }
 

src/utils/tools.py

@@ -51,13 +51,14 @@ def initialize_tools():
     tool_setup = selected.get("done")  # Main tool setup endpoint or configuration
     image_tool = selected.get("image")  # Image generation service endpoint
     audio_tool = selected.get("audio")  # Audio generation service endpoint
+    poll_token = selected.get("pol")  # Pollinations AI API Token
 
     # Verify that all required tool endpoints are present, raise exception if any is missing
-    if not tool_setup or not image_tool or not audio_tool:
+    if not tool_setup or not image_tool or not audio_tool or not poll_token:
         raise Exception("Selected host is missing required tool endpoints.")
 
     # Return the three tool endpoints as a tuple
-    return tool_setup, image_tool, audio_tool
+    return tool_setup, image_tool, audio_tool, poll_token
 
 # Initialize the tools by selecting an available host and retrieving its endpoints
-tool_setup, image_tool, audio_tool = initialize_tools()
+tool_setup, image_tool, audio_tool, poll_token = initialize_tools()