Rooni commited on
Commit
d57b22c
·
1 Parent(s): 802d5b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -12
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
- if auth_key != 'Your-Secret-Key':
16
- return jsonify({"error": "Unauthorized"}), 401
17
-
18
  # Получение описания изображения из тела запроса
19
  description = request.json.get('description')
20
  if not description:
21
  return jsonify({"error": "No description provided"}), 400
22
 
23
- # Вызов API DALL-E 2 для генерации изображения
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__':