Update app.py

app.py

@@ -24,8 +24,39 @@ MAX_IMAGE_SIZE = 2048
 pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
 
 @spaces.GPU(duration=75)
-def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
-    if randomize_seed:
+def infer(name, pet, background, style, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+    if pet == "Kaatz":
+        intro = "please generate an image of a cat sitting "
+    elif pet == "Mupp":
+        intro = "please generate an image of a dog sitting "
+    elif pet == "Hues":
+        intro = "please generate an image of a bunny sitting "
+    else:
+        intro = "please generate an image of an hamster sitting "
+
+    if background == "Wunnzëmmer":
+        place = intro + "in a living space "
+    elif background == "Grafitti Mauer":  
+        place = intro + "in front of a wall with graffiti "
+    elif background == "Strooss":   
+        place = intro + "in a street in the city "
+    elif background == "Plage":
+        place = intro + "at the beach "
+    else:
+        place = intro + " in the forest "
+
+    if style == "photo":
+        prompt = place + "holding a signal that says " + name + "in a photorealistic style"
+    elif style == "Cartoon":
+        prompt = place + "holding a signal that says " + name + "in a cartoon style"
+    elif style == "Woll":
+        prompt = place + "holding a signal that says " + name + "in a knitted with wool style"
+    elif style == "Aquarell":
+        prompt = place + "holding a signal that says " + name + "in a watercolorl style"
+    else:
+        prompt = place + "holding a signal that says " + name + "in a 3D style"
+        
+        if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
     
@@ -50,7 +81,7 @@ examples = [
 css="""
 #col-container {
     margin: 0 auto;
-    max-width: 600px;
+    max-width: 640px;
 }
 """
 
@@ -74,23 +105,23 @@ with gr.Blocks(css=css) as demo:
             run_button = gr.Button("Run", scale=0)
 
         with gr.Row():
-            radio_buttons_pet = gr.Radio(
+            pet = gr.Radio(
                 choices=["Kaatz", "Mupp", "Hues", "Hamster"],
                 label="Hausdéier",
                 value="Kaatz"
             ) 
 
         with gr.Row():
-            radio_buttons_background = gr.Radio(
+            background = gr.Radio(
                 choices=["Wunnzëmmer", "Graffiti Mauer", "Strooss", "Plage", "Bësch"],
                 label="Hannergronn",
                 value="Strooss"
             ) 
 
         with gr.Row():
-            radio_buttons_style = gr.Radio(
+            style = gr.Radio(
                 choices=["Photo", "Cartoon", "Woll", "Aquarell", "3D"],
-                label="Stlye",
+                label="Style",
                 value="Photo"
             )