Update app.py

app.py

@@ -62,26 +62,46 @@ DEFAULT_IMAGE_STYLE = "Cinematic"
 
 
 # ... (Previous code remains unchanged)
+def generate_txt2img(model_interface, prompt, is_negative=False, image_style="None style", steps=50, cfg_scale=7, seed=None):
+    if image_style == "None style":
+        payload = {
+            "inputs": prompt + ", 8k",
+            "is_negative": is_negative,
+            "steps": steps,
+            "cfg_scale": cfg_scale,
+            "seed": seed if seed is not None else random.randint(-1, 2147483647)
+        }
+    elif image_style == "Cinematic":
+        payload = {
+            "inputs": prompt + ", realistic, detailed, textured, skin, hair, eyes, by Alex Huguet, Mike Hill, Ian Spriggs, JaeCheol Park, Marek Denko",
+            "is_negative": is_negative + ", abstract, cartoon, stylized",
+            "steps": steps,
+            "cfg_scale": cfg_scale,
+            "seed": seed if seed is not None else random.randint(-1, 2147483647)
+        }
+    elif image_style == "Digital Art":
+        payload = {
+            "inputs": prompt + ", faded , vintage , nostalgic , by Jose Villa , Elizabeth Messina , Ryan Brenizer , Jonas Peterson , Jasmine Star",
+            "is_negative": is_negative + ", sharp , modern , bright",
+            "steps": steps,
+            "cfg_scale": cfg_scale,
+            "seed": seed if seed is not None else random.randint(-1, 2147483647)
+        }
+    elif image_style == "Portrait":
+        payload = {
+            "inputs": prompt + ", soft light, sharp, exposure blend, medium shot, bokeh, (hdr:1.4), high contrast, (cinematic, teal and orange:0.85), (muted colors, dim colors, soothing tones:1.3), low saturation, (hyperdetailed:1.2), (noir:0.4), (natural skin texture, hyperrealism, soft light, sharp:1.2)",
+            "is_negative": is_negative,
+            "steps": steps,
+            "cfg_scale": cfg_scale,
+            "seed": seed if seed is not None else random.randint(-1, 2147483647)
+        }
+
+    image_bytes = requests.post(API_URL, headers=headers, json=payload).content
+    image = Image.open(io.BytesIO(image_bytes))
+    return image
 
-def send_it1(inputs, noise_level, style, proc=model_interface):
-    global request_counter
-    request_counter += 1
-    timestamp = f"{time.time()}_{request_counter}"
-    advanced_options = f"\nStyle: {style}\n"
-    negative_prompt = " " 
-    prompt_with_noise = add_random_noise(inputs, noise_level) + f" - {timestamp}" + advanced_options + negative_prompt
-    while queue.qsize() >= queue_threshold:
-        time.sleep(2)
-    queue.put(prompt_with_noise)
-    output = proc(prompt_with_noise)
-    return output
 
-# ... (Remaining code remains unchanged)
-
-
-# ... (existing code)
-
-# ... (existing code)
+# ... (Previous code remains unchanged)
 
 with gr.Blocks(css=".gradio-container {background-color: #fdf7e6;} footer{display:none !important;}",) as demo:
 
@@ -121,8 +141,8 @@ with gr.Blocks(css=".gradio-container {background-color: #fdf7e6;} footer{displa
                 output1 = gr.Image(label="", show_label=False, show_share_button=False)
                 output2 = gr.Image(label="", show_label=False, show_share_button=False)
 
-        # Adjust the click event to include the style_selection Radio component
-        run.click(send_it1, inputs=[prompt, noise_level, style_selection, negative_prompt], outputs=[output1])
-        run.click(send_it1, inputs=[prompt, noise_level, style_selection, negative_prompt], outputs=[output2])
+        # Adjust the click event to use generate_txt2img function
+        run.click(generate_txt2img, inputs=[prompt, negative_prompt, style_selection, noise_level], outputs=[output1])
+        run.click(generate_txt2img, inputs=[prompt, negative_prompt, style_selection, noise_level], outputs=[output2])
         
     demo.launch(enable_queue=True, inline=True)