Create app.py

app.py

@@ -0,0 +1,68 @@
+import requests
+
+def generate_kontext_image(input_image_url, prompt="transform this image", width=1024, height=1024, seed=-1, model="kontext", nologo=True, enhance=False):
+    """
+    Generate a transformed image using the Kontext model from Pollinations API.
+    
+    Args:
+        input_image_url (str): URL of the input image to transform.
+        prompt (str): Prompt for the transformation (default: 'transform this image').
+        width (int): Width of the output image (default: 1024).
+        height (int): Height of the output image (default: 1024).
+        seed (int): Random seed for generation (default: -1 for random).
+        model (str): Model to use (default: 'kontext').
+        nologo (bool): Whether to exclude logo (default: True).
+        enhance (bool): Whether to enhance the image (default: False).
+    
+    Returns:
+        bytes: Raw image data if successful, None otherwise.
+    """
+    # Construct the API URL with query parameters
+    base_url = "https://image.pollinations.ai/prompt"
+    query_params = {
+        "model": model,
+        "image": input_image_url,
+        "width": width,
+        "height": height,
+        "seed": seed,
+        "nologo": str(nologo).lower(),
+        "enhance": str(enhance).lower()
+    }
+    api_url = f"{base_url}/{prompt.replace(' ', '%20')}"
+
+    try:
+        # Make the API request
+        response = requests.get(api_url, params=query_params, stream=True)
+        
+        # Check if the request was successful
+        if response.status_code == 200:
+            return response.content  # Return raw image data
+        else:
+            print(f"Error: API request failed with status code {response.status_code}")
+            return None
+    except requests.RequestException as e:
+        print(f"Error: Failed to connect to the API - {e}")
+        return None
+
+# Example usage
+if __name__ == "__main__":
+    # Replace with a valid input image URL
+    input_image = "https://example.com/input-image.jpg"
+    output_image_data = generate_kontext_image(
+        input_image_url=input_image,
+        prompt="transform this image into a surreal painting",
+        width=800,
+        height=600,
+        seed=42,
+        model="kontext",
+        nologo=True,
+        enhance=False
+    )
+
+    # Save the output image if successful
+    if output_image_data:
+        with open("output_image.jpg", "wb") as f:
+            f.write(output_image_data)
+        print("Image saved as output_image.jpg")
+    else:
+        print("Failed to generate image")