anime-kawai-diffusion2

Sleeping

App Files Files Community

soiz commited on 29 days ago

Commit

1eae35f

•

1 Parent(s): e58623d

Update app.py

Browse files

Files changed (1) hide show

app.py +82 -46

app.py CHANGED Viewed

@@ -1,76 +1,112 @@
 from flask import Flask, request, jsonify, send_file, render_template_string
 import io
-from PIL import Image, ImageDraw, ImageFont
 app = Flask(__name__)
-# Function to generate a simple image with text using PIL
-def generate_image_with_text(prompt, width, height):
-    # Create an image with a white background and specified dimensions
-    img = Image.new('RGB', (width, height), color='white')
-    draw = ImageDraw.Draw(img)
-    # Load a font
-    try:
-        font = ImageFont.truetype("arial.ttf", 40)  # Use a truetype font
-    except IOError:
-        font = ImageFont.load_default()  # Use default if arial is not available
-    # Draw the prompt text onto the image
-    draw.text((50, height // 2), prompt, fill="black", font=font)
-    return img
-# HTML template for the index page
-index_html = """
-<!DOCTYPE html>
-<html lang="ja">
-<head>
-    <meta charset="UTF-8">
-    <title>Kawai Diffusion</title>
-</head>
-<body>
-    <h1>Kawai Diffusion</h1>
-    <form action="/generate" method="get">
-        <input type="text" name="prompt" placeholder="Enter your prompt" required>
-        <input type="text" name="negative_prompt" placeholder="Enter negative prompt (optional)">
-        <input type="number" name="steps" placeholder="Steps" value="35">
-        <input type="number" name="cfgs" placeholder="CFG Scale" value="7" step="0.1">
-        <input type="text" name="sampler" placeholder="Sampler" value="DPM++ 2M Karras">
-        <input type="number" name="strength" placeholder="Strength" value="0.7" step="0.1">
-        <input type="number" name="seed" placeholder="Seed" value="-1">
-        <input type="number" name="width" placeholder="Width" value="1024">
-        <input type="number" name="height" placeholder="Height" value="1024">
-        <button type="submit">Generate Image</button>
-    </form>
-</body>
-</html>
-"""
 @app.route('/')
 def index():
-    return render_template_string(index_html)
 @app.route('/generate', methods=['GET'])
 def generate_image():
     prompt = request.args.get("prompt", "")
     negative_prompt = request.args.get("negative_prompt", "bad hands, low quality, low quality, unnatural, dirty eyes, distorted eyes, low quality eyes, distorted arms, distorted legs, distorted face, 3 legs, 3 hands, anatomically Bad, 6 fingers, 7 fingers, less fingers, more fingers,")
     steps = int(request.args.get("steps", 35))
-    cfg_scale = float(request.args.get("cfgs", 7))
-    sampler = request.args.get("sampler", "DPM++ 2M Karras")
     strength = float(request.args.get("strength", 0.7))
     seed = int(request.args.get("seed", -1))
     width = int(request.args.get("width", 1024))
     height = int(request.args.get("height", 1024))
-    # Generate the image with the provided prompt and dimensions
-    image = generate_image_with_text(prompt, width, height)
-    # Save the image to a bytes buffer
     img_bytes = io.BytesIO()
     image.save(img_bytes, format='PNG')
     img_bytes.seek(0)
     return send_file(img_bytes, mimetype='image/png')
 if __name__ == "__main__":

 from flask import Flask, request, jsonify, send_file, render_template_string
+import requests
 import io
+import random
+from PIL import Image
+from deep_translator import GoogleTranslator
 app = Flask(__name__)
+API_URL = "https://api-inference.huggingface.co/models/Ojimi/anime-kawai-diffusion"
+timeout = 3000  # Set timeout to 300 seconds
+def query(prompt, negative_prompt="", steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
+    if not prompt:
+        return None, "Prompt is required"
+    # Translate prompt from Russian to English if needed
+    prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
+    print(f'Translated prompt: {prompt}')
+    # Enhance the prompt
+    prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
+    print(f'Final prompt: {prompt}')
+    payload = {
+        "inputs": prompt,
+        "is_negative": False,
+        "steps": steps,
+        "cfg_scale": cfg_scale,
+        "seed": seed if seed != -1 else random.randint(1, 1000000000),
+        "strength": strength,
+        "parameters": {
+            "width": width,
+            "height": height
+        }
+    }
+    for attempt in range(3):
+        try:
+            response = requests.post(API_URL, json=payload, timeout=timeout)
+            if response.status_code != 200:
+                return None, f"Error: Failed to get image. Status code: {response.status_code}, Details: {response.text}"
+            image_bytes = response.content
+            image = Image.open(io.BytesIO(image_bytes))
+            return image, None
+        except requests.exceptions.Timeout:
+            if attempt < 2:
+                print("Timeout occurred, retrying...")
+                continue
+            return None, "Error: The request timed out. Please try again."
+        except requests.exceptions.RequestException as e:
+            return None, f"Request Exception: {str(e)}"
+        except Exception as e:
+            return None, f"Error when trying to open the image: {e}"
 @app.route('/')
 def index():
+    return render_template_string("""
+    <!DOCTYPE html>
+    <html lang="ja">
+    <head>
+        <meta charset="UTF-8">
+        <title>Image Generator</title>
+    </head>
+    <body>
+        <h1>Kawaii Diffusion Image Generator</h1>
+        <form action="/generate" method="get">
+            <label for="prompt">Prompt:</label><br>
+            <input type="text" id="prompt" name="prompt" required><br>
+            <label for="negative_prompt">Negative Prompt:</label><br>
+            <input type="text" id="negative_prompt" name="negative_prompt"><br>
+            <label for="steps">Steps:</label><br>
+            <input type="number" id="steps" name="steps" value="35"><br>
+            <label for="cfg_scale">CFG Scale:</label><br>
+            <input type="number" step="0.1" id="cfg_scale" name="cfg_scale" value="7"><br>
+            <label for="strength">Strength:</label><br>
+            <input type="number" step="0.1" id="strength" name="strength" value="0.7"><br>
+            <label for="seed">Seed:</label><br>
+            <input type="number" id="seed" name="seed" value="-1"><br>
+            <label for="width">Width:</label><br>
+            <input type="number" id="width" name="width" value="1024"><br>
+            <label for="height">Height:</label><br>
+            <input type="number" id="height" name="height" value="1024"><br><br>
+            <input type="submit" value="Generate">
+        </form>
+    </body>
+    </html>
+    """)
 @app.route('/generate', methods=['GET'])
 def generate_image():
     prompt = request.args.get("prompt", "")
     negative_prompt = request.args.get("negative_prompt", "bad hands, low quality, low quality, unnatural, dirty eyes, distorted eyes, low quality eyes, distorted arms, distorted legs, distorted face, 3 legs, 3 hands, anatomically Bad, 6 fingers, 7 fingers, less fingers, more fingers,")
     steps = int(request.args.get("steps", 35))
+    cfg_scale = float(request.args.get("cfg_scale", 7))
     strength = float(request.args.get("strength", 0.7))
     seed = int(request.args.get("seed", -1))
     width = int(request.args.get("width", 1024))
     height = int(request.args.get("height", 1024))
+    image, error = query(prompt, negative_prompt, steps, cfg_scale, seed=seed, strength=strength, width=width, height=height)
+    if error:
+        return jsonify({"error": error}), 400
     img_bytes = io.BytesIO()
     image.save(img_bytes, format='PNG')
     img_bytes.seek(0)
     return send_file(img_bytes, mimetype='image/png')
 if __name__ == "__main__":