Генератор персонализированных изображений

Генератор персонализированных изображений


pip install openai flask pillow

import openai

from flask import Flask, request, jsonify

from PIL import Image

import requests

from io import BytesIO


# 📌 Указываем API-ключ OpenAI

openai.api_key = "YOUR_OPENAI_API_KEY"


app = Flask(__name__)


# 📌 Функция для генерации изображения

def generate_image(prompt):

  response = openai.Image.create(

    prompt=prompt,

    n=1,

    size="1024x1024"

  )

  image_url = response["data"][0]["url"]

  return image_url


@app.route("/generate", methods=["POST"])

def generate():

  data = request.get_json()

  prompt = data.get("prompt", "A futuristic cityscape at sunset")

   

  # 📌 Генерация изображения

  image_url = generate_image(prompt)

   

  # 📌 Скачивание изображения

  response = requests.get(image_url)

  img = Image.open(BytesIO(response.content))

  img.save("generated_image.png")


  return jsonify({"image_url": image_url})


if __name__ == "__main__":

  app.run(debug=True)

Report Page