Update app.py

app.py

@@ -1,3 +1,4 @@
+import gradio as gr
 import PIL
 import torch
 import numpy as np
@@ -102,17 +103,19 @@ def compute_loss(original_image, loss_type):
     return error
 
 
+
 def get_examples():
    examples = [
-      ['A bird sitting on a tree', 'Midjourney', 'edge', 5],
-      ['Cats fighting on the road', 'Marc Allante', 'brightness', 65],
-      ['A mouse with the head of a puppy', 'Hitokomoru Style', 'contrast', 110],
-      ['A woman with a smiling face in front of an Italian Pizza', 'Hanfu Anime', 'brightness', 29],
-      ['A campfire (oil on canvas)', 'Birb Style', 'blue', 47],
+      ['A bird sitting on a tree', 'Midjourney', 'edge'],
+      ['Cats fighting on the road', 'Marc Allante', 'brightness'],
+      ['A mouse with the head of a puppy', 'Hitokomoru Style', 'contrast'],
+      ['A woman with a smiling face in front of an Italian Pizza', 'Hanfu Anime', 'brightness'],
+      ['A campfire (oil on canvas)', 'Birb Style', 'blue'],
    ]
-   return(examples)
-
+   return examples
 
+# Existing functions (latents_to_pil, show_image, generate_image)
+# ... (Copy all the existing functions here)
 def latents_to_pil(latents):
     # bath of latents -> list of images
     latents = (1 / 0.18215) * latents
@@ -205,4 +208,29 @@ def generate_image(seed, prompt, loss_type, loss_flag=False):
 
         latents = scheduler.step(noise_pred,t, latents).prev_sample
 
-    return latents
+    return latents
+
+# Gradio interface function
+def generate_images(prompt, style, guidance_type):
+    images = show_image(prompt, style, guidance_type)
+    return images[0], images[1]
+
+# Create Gradio interface
+iface = gr.Interface(
+    fn=generate_images,
+    inputs=[
+        gr.Textbox(label="Prompt"),
+        gr.Dropdown(list(styles_mapping.keys()), label="Style"),
+        gr.Dropdown(["blue", "edge", "contrast", "brightness", "sharpness", "saturation"], label="Guidance Type"),
+    ],
+    outputs=[
+        gr.Image(label="Image without Loss"),
+        gr.Image(label="Image with Loss"),
+    ],
+    examples=get_examples(),
+    title="Text Inversion Image Generation",
+    description="Generate images using text inversion with different styles and guidance types.",
+)
+
+# Launch the app
+iface.launch()