Fixes / adding more options

app.py

@@ -59,6 +59,23 @@ def save_image_to_hf_space(image_np, image_name):
     shutil.copy(local_image_path, persistent_storage_path)
 
    
+
+
+def genie(prompt, negative_prompt, prompt_2, negative_prompt_2, scale, guidance_scale, aesthetic_score, negative_aesthetic_score, steps, seed):
+    torch.cuda.empty_cache()
+    generator = torch.Generator(device=device).manual_seed(seed)
+    int_images = pipe1(prompt=prompt, prompt_2=prompt_2, negative_prompt=negative_prompt, negative_prompt_2=negative_prompt_2, num_inference_steps=steps, guidance_scale=scale, num_images_per_prompt=1, generator=generator).images
+    torch.cuda.empty_cache()
+    refined_images = pipe2(prompt=prompt, prompt_2=prompt_2, image=int_images, guidance_scale=guidance_scale, aesthetic_score=aesthetic_score, negative_aesthetic_score=negative_aesthetic_score).images
+    int_image_np = np.array(int_images[0])
+    refined_image_np = np.array(refined_images[0])
+
+    # Save the generated images to Hugging Face Spaces
+    save_image_to_hf_space(int_image_np, "int_image")
+    save_image_to_hf_space(refined_image_np, "refined_image")
+
+    return int_image_np, refined_image_np
+
 article = f"""
 <div style="text-align: center;">
     <img src="https://cdn-uploads.huggingface.co/production/uploads/6380cf05f496d57325c12194/LIONhgnyxUuEynNivGsME.jpeg" alt="ffusionAI-FFusionXL-SDXL-preview" width="100%" height="600">
@@ -91,9 +108,15 @@ article = f"""
 # Create the Gradio interface
 gr.Interface(fn=genie, 
              inputs=[gr.Textbox(label='Describe your FFusion idea. 77 Token Limit.', lines=2), 
-                     gr.Textbox(label='Things the AI should not create (negative prompt)', lines=2), 
-                     gr.Slider(1, 15, 10), gr.Slider(25, maximum=100, value=50, step=1), 
-                     gr.Slider(minimum=1, step=1, maximum=999999999999999999, randomize=True)], 
+                     gr.Textbox(label='Things the AI should not create (negative prompt)', lines=2),
+                     gr.Textbox(label='Additional prompt for text-encoder 2 (optional)', lines=2, optional=True),
+                     gr.Textbox(label='Negative prompt for text-encoder 2 (optional)', lines=2, optional=True),
+                     gr.Slider(1, 15, 10, label='Scale'),
+                     gr.Slider(1, 10, 5, label='Guidance Scale'),
+                     gr.Slider(1, 10, 6, label='Aesthetic Score'),
+                     gr.Slider(1, 10, 2.5, label='Negative Aesthetic Score'),
+                     gr.Slider(25, maximum=5, value=50, step=1, label='Steps'), 
+                     gr.Slider(minimum=1, step=1, maximum=999999999999999999, randomize=True, label='Seed')], 
              outputs=[gc.Image(type='numpy', label="FFusionXL Base Image"), gc.Image(type='numpy', label="Refined Image")],
              title="FFusionXL Base - Generate and Refine", 
              description='<div style="display: flex; flex-wrap: wrap; gap: 2px; justify-content: center;"><a href="https://huggingface.co/FFusion/FFusionXL-BASE" target="_new" rel="ugc"><img src="https://img.shields.io/badge/FFusionXL--BASE--SDXL-Model-pink" alt="FFusionXL-BASE-SDXL"></a>  <a href="https://huggingface.co/FFusion/FFusionXL-09-SDXL/blob/main/LICENSE.md" target="_new" rel="ugc"><img src="https://img.shields.io/badge/License-FFXL%20Research%20License-blue"></a></div>', 
@@ -107,4 +130,4 @@ gr.Interface(fn=genie,
                      display: inline-block;
                  }
              """
-).launch(debug=True, max_threads=10)
+).launch(debug=True, max_threads=10)