new flux prompt

inference.py

@@ -53,7 +53,7 @@ def generate_prompt_for_flux(user_prompt):
             {
             "role": "user",
             "content": [
-                {"type": "text", "text": f"Generate a prompt for a diffusion model that is a more detailed version of the following prompt: {user_prompt}. Keep it succinct but more descriptive than the original. Just return 1 sentence, do not include specific elements, since you do not know the available space in the room. "},
+                {"type": "text", "text": f"Generate a prompt for a diffusion model that is a more detailed version of the following prompt: {user_prompt}. Keep it succinct but more descriptive than the original. Just return a few words, listing possible relevant furniture elements and objects that will be present in the image. "},
             ],
         }
     ],
@@ -314,6 +314,9 @@ def main(prompt, control_image, index, metadata_df):
     logging.info("\nStep 0: Generating a detailed prompt for the diffusion model...")
     # Step 0: Generate a detailed prompt for the diffusion model
     prompt = generate_prompt_for_flux(prompt)
+
+    prompt += ", realistic, high quality, 8K"
+
     logging.info(f"\nGenerated prompt: {prompt}")
     
     logging.info("\nStep 1: Generating image...")