Upload 4 files

Dockerfile

@@ -0,0 +1,20 @@
+# Use an official Python runtime as a parent image
+FROM python:3.9-slim
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Switch to root user to ensure installation permissions
+USER root
+
+# Copy the current directory contents into the container at /app
+COPY . /app
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Make port 7860 available to the world outside this container
+EXPOSE 7860
+
+# Run app.py when the container launches
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,146 @@
+from flask import Flask, request, jsonify, send_file
+import websocket
+import uuid
+import json
+import urllib.request
+import urllib.parse
+import random
+from PIL import Image
+import io
+import base64
+
+# Initialize Flask app
+app = Flask(__name__)
+
+# Set server and websocket addresses
+server_address = "127.0.0.1:8188"
+ws_address = "ws://127.0.0.1:8188/ws"
+client_id = str(uuid.uuid4())
+
+def make_request(url, data=None, headers=None):
+    req = urllib.request.Request(url, data=data, headers=headers)
+    with urllib.request.urlopen(req) as response:
+        return json.loads(response.read())
+
+def queue_prompt(prompt, token):
+    payload = {"prompt": prompt, "client_id": client_id}
+    data = json.dumps(payload).encode('utf-8')
+    headers = {
+        'Authorization': f'Bearer {token}',  # Bearer token
+        'Content-Type': 'application/json'
+    }
+    return make_request(f"http://{server_address}/prompt", data=data, headers=headers)
+
+def get_image(filename, subfolder, image_type, token):
+    url_values = {'filename': filename, 'subfolder': subfolder, 'type': image_type}
+    url = f"http://{server_address}/view?{urllib.parse.urlencode(url_values)}"
+
+    req = urllib.request.Request(url)
+    req.add_header("Authorization", f"Bearer {token}")
+
+    # Debugging output
+    print(f"Request URL: {url}")
+    print(f"Request Headers: {req.headers}")
+
+    try:
+        return urllib.request.urlopen(req).read()
+    except urllib.error.HTTPError as e:
+        print(f"HTTP Error: {e.code} - {e.reason}")
+        print(e.read())  # This will give you more context about the error
+        raise
+
+def get_history(prompt_id, token):
+    headers = {
+        'Authorization': f'Bearer {token}',
+        'Content-Type': 'application/json'
+    }
+    return make_request(f"http://{server_address}/history/{prompt_id}", headers=headers)
+
+def get_images(ws, prompt, token):
+    prompt_id = queue_prompt(prompt, token)['prompt_id']
+    output_images = {}
+
+    while True:
+        out = ws.recv()
+        if isinstance(out, str):
+            message = json.loads(out)
+            if message['type'] == 'executing':
+                data = message['data']
+                if data['node'] is None and data['prompt_id'] == prompt_id:
+                    break  # Execution is done
+
+    history = get_history(prompt_id, token)[prompt_id]
+    for node_id in history['outputs']:
+        node_output = history['outputs'][node_id]
+        images_output = []
+        if 'images' in node_output:
+            for image in node_output['images']:
+                image_data = get_image(image['filename'], image['subfolder'], image['type'], token)
+                images_output.append(image_data)
+        output_images[node_id] = images_output
+
+    return output_images
+# Default route for home welcome
+@app.route('/')
+def home():
+    return "Welcome to ComfyUI API Interface!"
+
+# Generate image route
+@app.route('/generate_image', methods=['POST'])
+def generate_image():
+    data = request.json
+
+    # Extract the token from the request headers
+    token = request.headers.get('Authorization')
+
+    # Ensure the token is present and in the correct format
+    if token is None:
+        return jsonify({'error': 'No token provided'}), 400
+    # If the token does not have a 'Bearer' prefix, use the entire token
+    if token.startswith("Bearer "):
+        token = token.split(" ")[1]  # Extract the actual token if 'Bearer ' exists
+    else:
+        token = token  # Use the entire token if there's no 'Bearer ' prefix
+
+    # Base64 decode the encoded token
+    token = base64.b64decode(token).decode("utf-8")
+
+    if 'text_prompt' not in data:
+        return jsonify({'error': 'No text prompt provided'}), 400
+
+    text_prompt = data['text_prompt']
+    file_path = '/Users/muhammadwaqas/gosign/comfyui-api/workflow_api.json'
+
+    with open(file_path, 'r', encoding='utf-8') as file:
+        workflow_jsondata = file.read()
+
+    prompt = json.loads(workflow_jsondata)
+    prompt["6"]["inputs"]["text"] = text_prompt
+    prompt["7"]["inputs"]["text"] = "text, watermark, low quality, extra hands, extra legs."
+
+    seednum = random.randint(1, 9999999999999)
+    prompt["3"]["inputs"]["seed"] = seednum
+
+    ws = websocket.WebSocket()
+    ws.connect(f"{ws_address}?clientId={client_id}&token={token}")
+    images = get_images(ws, prompt, token)
+    ws.close()
+
+    output_image_paths = []
+
+    for node_id in images:
+        for image_data in images[node_id]:
+            image = Image.open(io.BytesIO(image_data))
+            image_path = f'image-output-{seednum}-{node_id}.png'
+            image.show()
+            image.save(image_path)
+            output_image_paths.append(image_path)
+
+    return jsonify({'images': output_image_paths})
+
+@app.route('/get_image/<filename>', methods=['GET'])
+def get_image_file(filename):
+    return send_file(filename, mimetype='image/png')
+
+if __name__ == '__main__':
+    app.run(debug=True, port=7860)

requirements.txt

@@ -0,0 +1,5 @@
+Flask==2.0.2
+websocket-client
+Pillow
+uuid
+gunicorn

workflow_api.json

@@ -0,0 +1,107 @@
+{
+  "3": {
+    "inputs": {
+      "seed": 1099951066657758,
+      "steps": 20,
+      "cfg": 8,
+      "sampler_name": "euler",
+      "scheduler": "normal",
+      "denoise": 1,
+      "model": [
+        "4",
+        0
+      ],
+      "positive": [
+        "6",
+        0
+      ],
+      "negative": [
+        "7",
+        0
+      ],
+      "latent_image": [
+        "5",
+        0
+      ]
+    },
+    "class_type": "KSampler",
+    "_meta": {
+      "title": "KSampler"
+    }
+  },
+  "4": {
+    "inputs": {
+      "ckpt_name": "Deliberate_v2.safetensors"
+    },
+    "class_type": "CheckpointLoaderSimple",
+    "_meta": {
+      "title": "Load Checkpoint"
+    }
+  },
+  "5": {
+    "inputs": {
+      "width": 512,
+      "height": 512,
+      "batch_size": 1
+    },
+    "class_type": "EmptyLatentImage",
+    "_meta": {
+      "title": "Empty Latent Image"
+    }
+  },
+  "6": {
+    "inputs": {
+      "text": "beautiful scenery nature glass bottle landscape, purple galaxy bottle,",
+      "clip": [
+        "4",
+        1
+      ]
+    },
+    "class_type": "CLIPTextEncode",
+    "_meta": {
+      "title": "CLIP Text Encode (Prompt)"
+    }
+  },
+  "7": {
+    "inputs": {
+      "text": "text, watermark",
+      "clip": [
+        "4",
+        1
+      ]
+    },
+    "class_type": "CLIPTextEncode",
+    "_meta": {
+      "title": "CLIP Text Encode (Prompt)"
+    }
+  },
+  "8": {
+    "inputs": {
+      "samples": [
+        "3",
+        0
+      ],
+      "vae": [
+        "4",
+        2
+      ]
+    },
+    "class_type": "VAEDecode",
+    "_meta": {
+      "title": "VAE Decode"
+    }
+  },
+  "9": {
+    "inputs": {
+      "filename_prefix": "ComfyUI",
+      "images": [
+        "8",
+        0
+      ]
+    },
+    "class_type": "SaveImage",
+    "_meta": {
+      "title": "Save Image"
+    }
+  }
+}