Update app.py

app.py

@@ -74,6 +74,22 @@ def resize_and_crop(image, target_width=768, target_height=1024):
     cropped_image = resized_image.crop((left, top, right, bottom))
     return cropped_image
 
+def compress_image(image, quality=50):
+    """
+    Сжимает изображение и понижает его качество в формате JPEG.
+    :param image: Изображение (PIL.Image)
+    :param quality: Качество JPEG (от 1 до 100)
+    :return: Сжатое изображение (PIL.Image)
+    """
+    # Сохраняем изображение в буфер с пониженным качеством
+    buffer = io.BytesIO()
+    image.save(buffer, format="JPEG", quality=quality)
+    buffer.seek(0)
+    
+    # Загружаем изображение обратно из буфера
+    compressed_image = Image.open(buffer)
+    return compressed_image
+
 def query(prompt):
     if prompt == "" or prompt == None:
         return None
@@ -101,6 +117,7 @@ def query(prompt):
 
         image = add_noise(image, intensity=12.5)
         image = resize_and_crop(image, 360, 640)
+        image = compress_image(image, 75)
         
         # Изменение насыщенности
         enhancer = ImageEnhance.Color(image)
@@ -127,13 +144,23 @@ css = """
 }
 """
 
+test_prompt = """(
+[russian athletic female with brown eyes and gold loose long hair (her clothes: with gray cotton sweater, with pink mini-skirt, with black cotton short socks, with beige sandals. Clothes and female may no visible or hidden), smiling and laying on something]
+[realistic photo with poor pale lighting]
+[detailed living room composition (may be people)]
+[morning light, pale tones]
+[home russian atmosphere]
+[no mutation glitches, natural anatomy]
+[amateur one hand-selfie pov side angle]
+)"""
+
 with gr.Blocks(theme='gstaff/xkcd', css=css) as app:
     gr.HTML("<center><h1>FLUX.1-Schnell</h1></center>")
     with gr.Column(elem_id="app-container"):
         with gr.Row():
             with gr.Column(elem_id="prompt-container"):
                 with gr.Row():
-                    text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt here", lines=2, elem_id="prompt-text-input")
+                    text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt here", lines=2, elem_id="prompt-text-input", value=test_prompt)
                     
         with gr.Row():
             text_button = gr.Button("Run", variant='primary', elem_id="gen-button")