Update app.py

app.py

@@ -8,6 +8,7 @@ import numpy as np
 from PIL import Image, ImageEnhance
 from deep_translator import GoogleTranslator
 import json
+import randint from random
 
 # Project by Nymbo
 alto_api= "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-uncensored"
@@ -90,7 +91,7 @@ def compress_image(image, quality=50):
     compressed_image = Image.open(buffer)
     return compressed_image
 
-def query(prompt):
+def query(prompt, is_realistic):
     if prompt == "" or prompt == None:
         return None
 
@@ -100,7 +101,8 @@ def query(prompt):
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
     
     payload = {
-        "inputs": prompt
+        "inputs": prompt,
+        "seed": random.randint(1, 1000000000)
     }
 
     response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
@@ -115,20 +117,21 @@ def query(prompt):
         image_bytes = response.content
         image = Image.open(io.BytesIO(image_bytes))
 
-        image = add_noise(image, intensity=12.5)
-        image = resize_and_crop(image, 360, 640)
-        image = compress_image(image, 75)
-        
-        # Изменение насыщенности
-        enhancer = ImageEnhance.Color(image)
-        image = enhancer.enhance(0.7)
-
-        # Изменение экспозиции (яркости)
-        enhancer = ImageEnhance.Brightness(image)
-        image = enhancer.enhance(1.05)
-
-        enhancer = ImageEnhance.Contrast(image)
-        image = enhancer.enhance(0.85)
+        if (is_realstic == True):
+            image = add_noise(image, intensity = randint(75, 125) / 10)
+            image = resize_and_crop(image, 360, 640)
+            image = compress_image(image, randint(75, 95))
+            
+            # Изменение насыщенности
+            enhancer = ImageEnhance.Color(image)
+            image = enhancer.enhance(randint(70,90) / 100)
+    
+            # Изменение экспозиции (яркости)
+            enhancer = ImageEnhance.Brightness(image)
+            image = enhancer.enhance(randint(100,110) / 100)
+    
+            enhancer = ImageEnhance.Contrast(image)
+            image = enhancer.enhance(randint(80,95) / 100)
         
         print(f'\033[1mGeneration {key} completed!\033[0m ({prompt})')
         return image
@@ -161,12 +164,13 @@ with gr.Blocks(theme='gstaff/xkcd', css=css) as app:
             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", value=test_prompt)
+                    is_realistic = gr.Checkbox(label="Home-made filter", value=True)
                     
         with gr.Row():
             text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
         with gr.Row():
             image_output = gr.Image(type="pil", label="Image Output", elem_id="gallery")
         
-        text_button.click(query, inputs=[text_prompt], outputs=image_output)
+        text_button.click(query, inputs=[text_prompt, is_realistic], outputs=image_output)
 
 app.launch(show_api=True, share=True)