Update app.py
Browse files
app.py
CHANGED
|
@@ -4,29 +4,20 @@ import openai
|
|
| 4 |
# Инициализация Flask приложения
|
| 5 |
app = Flask(__name__)
|
| 6 |
|
| 7 |
-
# Загрузите ваш API-ключ OpenAI в безопасное место
|
| 8 |
-
# Например, из переменных окружения или файла конфигурации
|
| 9 |
-
openai.api_key = 'your-openai-api-key'
|
| 10 |
-
|
| 11 |
@app.route('/generate-image', methods=['POST'])
|
| 12 |
def generate_image():
|
| 13 |
# Проверка наличия авторизационного ключа в заголовках запроса
|
| 14 |
auth_key = request.headers.get('Authorization')
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
# Получение описания изображения из тела запроса
|
| 19 |
description = request.json.get('description')
|
| 20 |
if not description:
|
| 21 |
return jsonify({"error": "No description provided"}), 400
|
| 22 |
|
| 23 |
-
|
| 24 |
-
# Здесь может понадобиться дополнительная логика для работы с API DALL-E 2
|
| 25 |
-
# Например, загрузка сгенерированного изображения на хранилище и получение URL
|
| 26 |
-
response = openai.Image.create(prompt=description, n=1)
|
| 27 |
image_url = response['data'][0]['url'] # Пример получения URL изображения
|
| 28 |
|
| 29 |
-
# Возвращение URL сгенерированного изображения
|
| 30 |
return jsonify({"image_url": image_url})
|
| 31 |
|
| 32 |
if __name__ == '__main__':
|
|
|
|
| 4 |
# Инициализация Flask приложения
|
| 5 |
app = Flask(__name__)
|
| 6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
@app.route('/generate-image', methods=['POST'])
|
| 8 |
def generate_image():
|
| 9 |
# Проверка наличия авторизационного ключа в заголовках запроса
|
| 10 |
auth_key = request.headers.get('Authorization')
|
| 11 |
+
openai.api_key = auth_key
|
| 12 |
+
|
|
|
|
| 13 |
# Получение описания изображения из тела запроса
|
| 14 |
description = request.json.get('description')
|
| 15 |
if not description:
|
| 16 |
return jsonify({"error": "No description provided"}), 400
|
| 17 |
|
| 18 |
+
response = openai.Image.create(prompt=description, n=1, model="dall-e-3", size="1024x1024", quality="hd")
|
|
|
|
|
|
|
|
|
|
| 19 |
image_url = response['data'][0]['url'] # Пример получения URL изображения
|
| 20 |
|
|
|
|
| 21 |
return jsonify({"image_url": image_url})
|
| 22 |
|
| 23 |
if __name__ == '__main__':
|