hiding unnecessary inputs

app.py

@@ -104,28 +104,36 @@ def get_negative_prompt(gender):
     else:  # Random
         return "(deformed iris, deformed pupils, semi-realistic, cgi, 3d, render, sketch, cartoon, drawing, anime, mutated hands and fingers:1.4), (deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, disconnected limbs, mutation, mutated, ugly, disgusting, amputation"
 
-@spaces.GPU(duration=40)
-def generate_image(face_image_1, face_image_2, prompt, gender, width, height, num_inference_steps, seed):
-    if seed is None:
-        seed = int.from_bytes(os.urandom(4), "big")
+def construct_prompt(base_prompt, additional_prompt, gender):
+    full_prompt = base_prompt
+    if gender == "Boy":
+        full_prompt += ", male child, boy"
+    elif gender == "Girl":
+        full_prompt += ", female child, girl"
+    
+    if additional_prompt:
+        full_prompt += ", " + additional_prompt
+    
+    return full_prompt
 
-    if gender == "Random":
-        gender = random.choice(["Boy", "Girl"])
+@spaces.GPU(duration = 40)
+def generate_image(face_image_1, face_image_2, additional_prompt, gender):
+    base_prompt = "portrait of a 6 y.o. child, 8k, HD, happy, perfect eyes, cute"
+    full_prompt = construct_prompt(base_prompt, additional_prompt, gender)
 
     negative_prompt = get_negative_prompt(gender)
-    
     baby_image_path = morph_faces(face_image_1, face_image_2)
 
     def generate_images(faceid_embeds, num_outputs=1):
         images = ip_model.generate(
-            prompt=prompt,
+            prompt=full_prompt,
             negative_prompt=negative_prompt,
             faceid_embeds=faceid_embeds,
             num_samples=num_outputs,
-            width=width,
-            height=height,
-            num_inference_steps=num_inference_steps,
-            seed=seed
+            width=768,
+            height=768,
+            num_inference_steps=40,
+            seed=None
         )
         return images
 
@@ -141,15 +149,11 @@ gr_interface = gr.Interface(
     inputs=[
         gr.Image(type="filepath", label="First Face Image"),
         gr.Image(type="filepath", label="Second Face Image"),
-        gr.Textbox(value="portrait of a 6 y.o. child, 8k, HD, happy, in living room, perfect eyes, cute", label="Prompt"),
-        gr.Dropdown(choices=["Boy", "Girl", "Random"], value="Boy", label="Gender"),
-        gr.Slider(256, 1024, value=768, step=64, label="Width"),
-        gr.Slider(256, 1024, value=768, step=64, label="Height"),
-        gr.Slider(1, 200, value=30, step=1, label="Number of Inference Steps"),
-        gr.Number(value=None, label="Random Seed")
+        gr.Textbox(label="Prompt"),
+        gr.Dropdown(choices=["Boy", "Girl", "Random"], value="Boy", label="Gender")
     ],
     outputs=gr.Image(label="Generated Image"),
-    title="Face Morphing and Image Generation with Stable Diffusion"
+    title="Face Morphing and Image Generation with Stable Diffusion",
 )
 
 gr_interface.launch(share=True)