import requests
from g4f.client import Client

# Запрашиваем у пользователя текстовое описание (промпт) для генерации изображения
prompt = input("Введите описание изображения: ")

# Создаём экземпляр клиента
client = Client()

# Отправляем запрос на генерацию изображения по заданному промпту
response = client.images.generate(
    model="flux",
    prompt=prompt,
    response_format="url"
)

# Получаем URL сгенерированного изображения
image_url = response.data[0].url
print(f"URL сгенерированного изображения: {image_url}")

# Загружаем изображение по полученному URL
image_data = requests.get(image_url).content

# Сохраняем изображение
with open("generated_image.jpg", "wb") as file:
    file.write(image_data)

print("Изображение сохранено как generated_image.jpg")

import cv2

# Инициализируем объект VideoCapture для захвата видео с камеры (0 - обычно встроенная камера)
cap = cv2.VideoCapture(0)
# Считываем один кадр с камеры. 'ret' - булево значение (True, если кадр успешно считан),
# 'frame' - сам кадр (изображение)
ret, frame = cap.read()
if ret:  # Проверяем, был ли успешно считан кадр
    # Сохраняем полученный кадр "snapshot.png" в текущей директории
    cv2.imwrite("snapshot.png", frame)
    print("📷 Снимок сохранён как snapshot.png")
    # Освобождаем ресурсы, связанные с объектом VideoCapture (закрываем камеру)
cap.release()