Update app.py

app.py

@@ -69,12 +69,27 @@ if uploaded_file is not None:
             hair_color = query_analysis(image_bytes, "What is the hair color and hair style of the person?")
             facial_expression = query_analysis(image_bytes, "What is the facial expression and facial features of the person?")
             age = query_analysis(image_bytes, "What is the estimated age of the person?")
+            background = query_analysis(image_bytes, "What is the background or setting of the image?")
 
-        # Build generation prompt based on VQA responses and user input
+        # Build the generation prompt based on VQA responses and user input
         if gender.lower() == "female":
-            prompt = f"Create a full image from head to foot and use your creativity to generate image of {age}-year-old girl with {hair_color} hair, wearing {clothing}, showing a {facial_expression}. {user_prompt}"
+            prompt = (
+                f"Generate a highly creative and artistic full-body image of a {age}-year-old girl. "
+                f"She has {hair_color} hair, styled uniquely. She is wearing {clothing} that complements her personality. "
+                f"Her facial expression shows {facial_expression}, with distinct facial features. "
+                f"Place her in a visually compelling {background}, enhancing the mood and atmosphere of the image. "
+                f"Use dynamic lighting, textures, and vivid colors to create an image that feels alive. "
+                f"{user_prompt}."
+            )
         else:
-            prompt = f"Create a full image from head to foot and use your creativity to generate image of {age}-year-old person with {hair_color} hair, wearing {clothing}, showing a {facial_expression}. {user_prompt}"
+            prompt = (
+                f"Generate a highly creative and artistic full-body image of a {age}-year-old person. "
+                f"They have {hair_color} hair, styled uniquely. They are wearing {clothing} that complements their personality. "
+                f"Their facial expression shows {facial_expression}, with distinct facial features. "
+                f"Place them in a visually compelling {background}, enhancing the mood and atmosphere of the image. "
+                f"Use dynamic lighting, textures, and vivid colors to create an image that feels alive. "
+                f"{user_prompt}."
+            )
 
         # Call image generation API
         with st.spinner("Generating the image..."):