Update app.py

app.py

@@ -4,17 +4,13 @@ import io
 import random
 import os
 from PIL import Image
-from deep_translator import GoogleTranslator
 
 API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
-models_list = ["AbsoluteReality 1.8.1", "DALL-E 3 XL", "Playground 2", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL", "Vector Art XL", "epiCRealism", "PixelArt XL", "NewReality XL", "Anything 5.0", "Disney", "CleanLinearMix", "OrangeMixs"]
-
-# PLEASE ❤ like ❤ this space. Please like me. I am 12 years old, one of my projects is: https://ai-hub.rf.gd . I live in Russia, I don't know English very well. Therefore, I apologize that there is only Russian here, but I think it will not be difficult to translate all this. (For example, using gpt)
-
-def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7):
+models_list = ["Реалистичность", "Аниме", "Disney"]
+def query(prompt, model, pose, vid, is_negative="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry, text, fuzziness", steps=45, cfg_scale=8, sampler="DPM++ 2M Karras", seed=-1, strength=0.7):
     if prompt == "" or prompt == None:
         return None
 
@@ -22,14 +18,12 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
     API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN"), os.getenv("HF_READ_TOKEN_2"), os.getenv("HF_READ_TOKEN_3"), os.getenv("HF_READ_TOKEN_4"), os.getenv("HF_READ_TOKEN_5")]) # it is free
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
     key = random.randint(0, 999)
-    
-    prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
-    print(f'\033[1mГенерация {key} перевод:\033[0m {prompt}')
 
-    prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
+    prompt = f"The {prompt} in a {pose} pose. View from {vid} | porn ultra detail, ultra elaboration, ultra quality, perfect."
     print(f'\033[1mГенерация {key}:\033[0m {prompt}')
-    if model == 'DALL-E 3 XL':
+    if model == 'Реалистичность':
         API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
+        prompt = f"Ultra realistic porn. {prompt}"
     if model == 'Playground 2':
         API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
     if model == 'Openjourney 4':
@@ -40,7 +34,7 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/lyrielv16"
     if model == 'Animagine XL 2.0':
         API_URL = "https://api-inference.huggingface.co/models/Linaqruf/animagine-xl-2.0"
-        prompt = f"Anime. {prompt}"
+        prompt = f"Anime porn. {prompt}"
     if model == 'Counterfeit 2.5':
         API_URL = "https://api-inference.huggingface.co/models/gsdf/Counterfeit-V2.5"
     if model == 'Realistic Vision 5.1':
@@ -49,7 +43,7 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         API_URL = "https://api-inference.huggingface.co/models/digiplay/incursiosMemeDiffusion_v1.6"
     if model == 'Anime Detailer XL':
         API_URL = "https://api-inference.huggingface.co/models/Linaqruf/anime-detailer-xl-lora"
-        prompt = f"Anime. {prompt}"
+        prompt = f"Anime porn. {prompt}"
     if model == 'epiCRealism':
         API_URL = "https://api-inference.huggingface.co/models/emilianJR/epiCRealism"
     if model == 'PixelArt XL':
@@ -62,9 +56,10 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         API_URL = "https://api-inference.huggingface.co/models/DoctorDiffusion/doctor-diffusion-s-controllable-vector-art-xl-lora"
     if model == 'Disney':
         API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
-        prompt = f"Disney style. {prompt}"
-    if model == 'CleanLinearMix':
+        prompt = f"Disney style porn. {prompt}"
+    if model == 'Аниме':
         API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
+        prompt = f"Anime porn. {prompt}"
     if model == 'OrangeMixs':
         API_URL = "https://api-inference.huggingface.co/models/WarriorMama777/OrangeMixs"
 
@@ -101,13 +96,18 @@ footer {visibility: hidden !important;}
 """
 
 with gr.Blocks(css=css) as dalle:
+    gr.Markdown("## 🔞 Strictly 18+ 🔞")
     with gr.Tab("Базовые настройки"):
         with gr.Row():
             with gr.Column(elem_id="prompt-container"):
                 with gr.Row():
-                    text_prompt = gr.Textbox(label="Prompt", placeholder="Описание изображения", lines=3, elem_id="prompt-text-input")
+                    image = gr.Dropdown(label="На фото", value="Naked girl", choices=["Naked girl", "Sex", "Blow job"])
+                with gr.Row():
+                    model = gr.Radio(label="Стиль", value="Реалистичность", choices=models_list)
                 with gr.Row():
-                    model = gr.Radio(label="Модель", value="DALL-E 3 XL", choices=models_list)
+                    pose = gr.Dropdown(label="Поза", value="Cancer", choices=["Cancer", "Lying down", "on my knees", "sitting"])
+                with gr.Row():
+                    vid = gr.Dropdown(label="Вид", value="Random", choices=["Back", "Front", "Top", "Bottom", "Random"])
              
                 
 
@@ -125,15 +125,12 @@ with gr.Blocks(css=css) as dalle:
         with gr.Row():
             seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
 
-    with gr.Tab("Информация"):
-        with gr.Row():
-            gr.Textbox(label="Шаблон prompt", value="{prompt} | ultra detail, ultra elaboration, ultra quality, perfect.")
-
+    
     with gr.Row():
         text_button = gr.Button("Генерация", variant='primary', elem_id="gen-button")
     with gr.Row():
         image_output = gr.Image(type="pil", label="Изображение", elem_id="gallery")
         
-    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed, strength], outputs=image_output)
+    text_button.click(query, inputs=[image, model, pose, vid, negative_prompt, steps, cfg, method, seed, strength], outputs=image_output)
 
 dalle.launch(show_api=False, share=False)