Update app.py

app.py

@@ -15,6 +15,26 @@ API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
 
+def add_noise(image, intensity=25):
+    """
+    Добавляет цифровой шум к изображению.
+    :param image: Изображение (PIL.Image)
+    :param intensity: Интенсивность шума (от 0 до 255)
+    :return: Изображение с шумом (PIL.Image)
+    """
+    # Преобразуем изображение в массив NumPy
+    img_array = np.array(image)
+    
+    # Генерируем шум
+    noise = np.random.randint(-intensity, intensity, img_array.shape, dtype=np.int32)
+    
+    # Добавляем шум к изображению
+    noisy_array = np.clip(img_array + noise, 0, 255).astype(np.uint8)
+    
+    # Преобразуем массив обратно в изображение
+    noisy_image = Image.fromarray(noisy_array)
+    return noisy_image
+
 def query(prompt):
     if prompt == "" or prompt == None:
         return None
@@ -41,8 +61,8 @@ def query(prompt):
         image = Image.open(io.BytesIO(image_bytes))
 
         width, height = image.size
-        new_width = 640
-        new_height = 1024
+        new_width = 480
+        new_height = 960
         left = (width - new_width) / 2
         top = (height - new_height) / 2
         right = (width + new_width) / 2
@@ -56,6 +76,9 @@ def query(prompt):
         # Изменение насыщенности
         enhancer = ImageEnhance.Color(image)
         image = enhancer.enhance(0.8)  # Увеличиваем насыщенность на 50%
+
+        # Добавление цифрового шума
+        image = add_noise(image, intensity=25)  # Интенсивность шума
         
         print(f'\033[1mGeneration {key} completed!\033[0m ({prompt})')
         return image