Spaces:

soiz
/

sdAPI

Running

App Files Files Community

soiz commited on Oct 12, 2024

Commit

725dd48

•

1 Parent(s): 2e4ca03

Update app.py

Browse files

Files changed (1) hide show

app.py +6 -104

app.py CHANGED Viewed

@@ -1,67 +1,3 @@
-from flask import Flask, request, jsonify, send_file, render_template_string
-import requests
-import io
-import os
-import random
-from PIL import Image
-from deep_translator import GoogleTranslator
-app = Flask(__name__)
-API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
-API_TOKEN = os.getenv("HF_READ_TOKEN")
-headers = {"Authorization": f"Bearer {API_TOKEN}"}
-timeout = 300  # タイムアウトを300秒に設定
-# Function to query the API and return the generated image
-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"
-    key = random.randint(0, 999)
-    # Translate the prompt from Russian to English if necessary
-    prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
-    print(f'Generation {key} translation: {prompt}')
-    # Add some extra flair to the prompt
-    prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
-    print(f'Generation {key}: {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):  # 最大3回の再試行
-        try:
-            response = requests.post(API_URL, headers=headers, 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}"
-# HTML template for the index page
-index_html = """
 <!DOCTYPE html>
 <html lang="ja">
 <head>
@@ -82,14 +18,14 @@ index_html = """
     </style>
     <script>
         let isDrawing = false;
-        let x = 0;
-        let y = 0;
         let penSize = 5;
         let penColor = '#000000';
         function startDrawing(event) {
             isDrawing = true;
-            [x, y] = getMousePos(event);
         }
         function draw(event) {
@@ -100,14 +36,14 @@ index_html = """
             const [newX, newY] = getMousePos(event);
             ctx.beginPath();
-            ctx.moveTo(x, y);
             ctx.lineTo(newX, newY);
             ctx.strokeStyle = penColor;
             ctx.lineWidth = penSize;
             ctx.stroke();
             ctx.closePath();
-            [x, y] = [newX, newY];
         }
         function stopDrawing() {
@@ -222,7 +158,7 @@ index_html = """
         <textarea id="negative_prompt" name="negative_prompt" rows="4" cols="50" placeholder="Enter negative prompt (optional)">(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos</textarea><br><br>
         <label for="width">Width:</label>
-        <input type="number" id="width" name="width" value="1024" min="64" max="2048" step="8" style="width:250px" oninput="syncWidth(this.value)">
         <input type="range" id="width-slider" name="width-slider" min="64" max="2048" value="1024" step="8" style="width:250px; display: inline-block;" oninput="updateWidthInput()"><br><br>
         <label for="height">Height:</label>
@@ -279,37 +215,3 @@ index_html = """
     </script>
 </body>
 </html>
-"""
-@app.route('/')
-def index():
-    return render_template_string(index_html)
-@app.route('/generate', methods=['GET'])
-def generate_image():
-    # Retrieve query parameters
-    prompt = request.args.get("prompt", "")
-    negative_prompt = request.args.get("negative_prompt", "")
-    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))
-    # Call the query function to generate the image
-    image, error = query(prompt, negative_prompt, steps, cfg_scale, sampler, seed, strength, width, height)
-    if error:
-        return jsonify({"error": error}), 400
-    # Save the image to a BytesIO object and return it
-    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__":
-    app.run(host='0.0.0.0', port=7860)

 <!DOCTYPE html>
 <html lang="ja">
 <head>
     </style>
     <script>
         let isDrawing = false;
+        let lastX = 0;
+        let lastY = 0;
         let penSize = 5;
         let penColor = '#000000';
         function startDrawing(event) {
             isDrawing = true;
+            [lastX, lastY] = getMousePos(event);
         }
         function draw(event) {
             const [newX, newY] = getMousePos(event);
             ctx.beginPath();
+            ctx.moveTo(lastX, lastY);
             ctx.lineTo(newX, newY);
             ctx.strokeStyle = penColor;
             ctx.lineWidth = penSize;
             ctx.stroke();
             ctx.closePath();
+            [lastX, lastY] = [newX, newY]; // 新しい位置を保存
         }
         function stopDrawing() {
         <textarea id="negative_prompt" name="negative_prompt" rows="4" cols="50" placeholder="Enter negative prompt (optional)">(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos</textarea><br><br>
         <label for="width">Width:</label>
+        <input type="number" id="width" name="width" value="1024" min="64" max="2048" step="8">
         <input type="range" id="width-slider" name="width-slider" min="64" max="2048" value="1024" step="8" style="width:250px; display: inline-block;" oninput="updateWidthInput()"><br><br>
         <label for="height">Height:</label>
     </script>
 </body>
 </html>