Spaces:
Running
Running
| from flask import Flask, request, jsonify | |
| import requests | |
| import base64 | |
| import io | |
| import json | |
| app = Flask(__name__) | |
| STYLES = { | |
| 'pixel': { | |
| 'prompt': 'Turn this image into the Pixel style.', | |
| 'lora_url': 'https://huggingface.co/Owen777/Kontext-Style-Loras/resolve/main/Pixel_lora_weights.safetensors' | |
| }, | |
| 'snoopy': { | |
| 'prompt': 'Turn this image into the Snoopy style.', | |
| 'lora_url': 'https://huggingface.co/Owen777/Kontext-Style-Loras/resolve/main/Snoopy_lora_weights.safetensors' | |
| }, | |
| 'jojo': { | |
| 'prompt': 'Turn this image into the JoJo style.', | |
| 'lora_url': 'https://huggingface.co/Owen777/Kontext-Style-Loras/resolve/main/Jojo_lora_weights.safetensors' | |
| }, | |
| 'clay': { | |
| 'prompt': 'Turn this image into the Clay style.', | |
| 'lora_url': 'https://huggingface.co/Owen777/Kontext-Style-Loras/resolve/main/Clay_Toy_lora_weights.safetensors' | |
| }, | |
| 'ghibli': { | |
| 'prompt': 'Turn this image into the Ghibli style.', | |
| 'lora_url': 'https://huggingface.co/Owen777/Kontext-Style-Loras/resolve/main/Ghibli_lora_weights.safetensors' | |
| }, | |
| 'american-cartoon': { | |
| 'prompt': 'Turn this image into the American Cartoon style.', | |
| 'lora_url': 'https://huggingface.co/Owen777/Kontext-Style-Loras/resolve/main/American_Cartoon_lora_weights.safetensors' | |
| }, | |
| 'lego': { | |
| 'prompt': 'convert to lego style', | |
| 'lora_url': 'https://huggingface.co/Owen777/Kontext-Style-Loras/resolve/main/LEGO_lora_weights.safetensors' | |
| }, | |
| 'broccoli-hair': { | |
| 'prompt': 'Change hair to a broccoli haircut', | |
| 'lora_url': 'https://huggingface.co/fal/Broccoli-Hair-Kontext-Dev-LoRA/resolve/main/broccoli-hair-kontext-dev-lora.safetensors' | |
| }, | |
| 'plushie': { | |
| 'prompt': 'Convert to plushie style', | |
| 'lora_url': 'https://huggingface.co/fal/Plushie-Kontext-Dev-LoRA/resolve/main/plushie-kontext-dev-lora.safetensors' | |
| }, | |
| 'wojak': { | |
| 'prompt': 'Convert to wojak style drawing', | |
| 'lora_url': 'https://huggingface.co/fal/Wojak-Kontext-Dev-LoRA/resolve/main/wojak-kontext-dev-lora.safetensors' | |
| }, | |
| 'upscalecompression': { | |
| 'prompt': 'fix the jpeg compression', | |
| 'lora_url': 'https://huggingface.co/fofr/flux-kontext-dev-jpeg-compression-fix-lora/resolve/main/flux-kontext-dev-jpeg-compression-fix-lora.safetensors' | |
| }, | |
| 'gfx': { | |
| 'prompt': 'render this image into a gfx image, fill in the background perfect and edit the image to look like awesome graphic gfx image', | |
| 'lora_url': 'https://huggingface.co/jerrrycans/gfx/resolve/main/flux-kontext-gfx-lora.safetensors' | |
| }, | |
| 'watermarkremover': { | |
| 'prompt': 'remove all the watermarks from this image, all watermarks that are over this image', | |
| 'lora_url': 'https://huggingface.co/jerrrycans/watermark20000x2/resolve/main/flux-kontext-watermark20000x2-lora.safetensors' | |
| }, | |
| 'gfxultra': { | |
| 'prompt': 'render this image into a gfx image, fill in the background perfect and edit the image to look like awesome graphic gfx image', | |
| 'lora_url': 'https://huggingface.co/jerrrycans/gfx20000/resolve/main/flux-kontext-gfx20000-lora.safetensors' | |
| }, | |
| 'fluffy': { | |
| 'prompt': 'make this object fluffy', | |
| 'lora_url': None | |
| }, | |
| 'glass': { | |
| 'prompt': 'make the character/object look like it was made out of glass, black background', | |
| 'lora_url': None | |
| }, | |
| 'simpsons': { | |
| 'prompt': 'convert to Simpsons cartoon style', | |
| 'lora_url': None | |
| }, | |
| 'anime': { | |
| 'prompt': 'convert to anime art style with large eyes and stylized features', | |
| 'lora_url': None | |
| } | |
| } | |
| def upload_base64_image(base64_data): | |
| try: | |
| header, data = base64_data.split(',', 1) | |
| image_data = base64.b64decode(data) | |
| files = {'file': ('generated_image.png', io.BytesIO(image_data), 'image/png')} | |
| headers = { | |
| 'Origin': 'https://jerrrycans-file.hf.space', | |
| 'Referer': 'https://jerrrycans-file.hf.space/', | |
| 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36' | |
| } | |
| response = requests.post('https://jerrrycans-file.hf.space/upload', files=files, headers=headers) | |
| if response.status_code == 200: | |
| result = response.json() | |
| return f"https://jerrrycans-file.hf.space{result['url']}" | |
| except Exception as e: | |
| print(f"Upload error: {e}") | |
| return None | |
| def get_style_config(style=None, custom_prompt=None, custom_lora_url=None): | |
| if style and style.lower() in STYLES: | |
| sc = STYLES[style.lower()] | |
| return { | |
| 'prompt': custom_prompt or sc['prompt'], | |
| 'lora_url': custom_lora_url or sc['lora_url'] | |
| } | |
| elif custom_prompt: | |
| return {'prompt': custom_prompt, 'lora_url': custom_lora_url} | |
| else: | |
| return None | |
| def process_stream_response(response): | |
| uploaded_images = [] | |
| buffer = '' | |
| for chunk in response.iter_content(chunk_size=1024, decode_unicode=True): | |
| if chunk: | |
| buffer += chunk | |
| lines = buffer.split('\n') | |
| buffer = lines.pop() | |
| for line in lines: | |
| if line.startswith('data: ') and len(line) > 6: | |
| try: | |
| data_content = line[6:].strip() | |
| if not data_content: | |
| continue | |
| data = json.loads(data_content) | |
| if (data.get('json', {}).get('type') in ['progress', 'completed'] and | |
| data.get('json', {}).get('data', {}).get('images')): | |
| for image in data['json']['data']['images']: | |
| if image.get('url', '').startswith('data:image/'): | |
| uploaded_url = upload_base64_image(image['url']) | |
| if uploaded_url: | |
| uploaded_images.append(uploaded_url) | |
| except: | |
| continue | |
| return uploaded_images | |
| def transform_image(): | |
| try: | |
| data = request.get_json() | |
| if not data: | |
| return jsonify({'success': False, 'error': 'No JSON data provided'}), 400 | |
| image_url = data.get('image_url') | |
| style = data.get('style') | |
| custom_prompt = data.get('custom_prompt') | |
| custom_lora_url = data.get('lora_url') | |
| if not image_url or not image_url.startswith(('http://', 'https://')): | |
| return jsonify({'success': False, 'error': 'Valid image_url is required'}), 400 | |
| if not (style or custom_prompt): | |
| return jsonify({'success': False, 'error': 'Either style or custom_prompt is required'}), 400 | |
| style_config = get_style_config(style, custom_prompt, custom_lora_url) | |
| if not style_config or not style_config.get('prompt'): | |
| return jsonify({'success': False, 'error': 'Valid style or custom_prompt required'}), 400 | |
| generate_params = { | |
| 'json': { | |
| 'imageUrl': image_url, | |
| 'prompt': style_config['prompt'] | |
| } | |
| } | |
| if style_config.get('lora_url'): | |
| generate_params['json']['loraUrl'] = style_config['lora_url'] | |
| generate_url = f"https://fal-kontext-demo.vercel.app/api/trpc/generateImageStream?input={requests.utils.quote(json.dumps(generate_params))}" | |
| response = requests.get(generate_url, stream=True) | |
| if response.status_code != 200: | |
| return jsonify({'success': False, 'error': 'Failed to generate styled image'}), 500 | |
| uploaded_images = process_stream_response(response) | |
| return jsonify({ | |
| 'success': True, | |
| 'originalImage': image_url, | |
| 'style': style, | |
| 'promptUsed': style_config['prompt'], | |
| 'generatedImages': uploaded_images, | |
| 'count': len(uploaded_images) | |
| }) | |
| except Exception as e: | |
| return jsonify({'success': False, 'error': str(e)}), 500 | |
| def index(): | |
| return jsonify({ | |
| 'message': 'Image Style Transfer API', | |
| 'usage': 'POST /api/transform with JSON body: {"image_url": "...", "style": "...", "custom_prompt": "(optional)", "lora_url": "(optional)"}', | |
| 'note': 'Either "style" or "custom_prompt" must be provided. If "custom_prompt" is provided, it overrides the default prompt for a known "style"; if only "custom_prompt" is given, style can be omitted.', | |
| 'styles': list(STYLES.keys()) | |
| }) | |
| if __name__ == '__main__': | |
| app.run(host='0.0.0.0', port=7860, debug=True) |