Update app.py

app.py

@@ -22,11 +22,34 @@ cut off, low contrast, underexposed, overexposed, bad art, beginner, amateur, di
 
 def enhance_prompt(prompt, style):
     model = genai.GenerativeModel("gemini-2.0-flash-lite")
-    enhanced_prompt_request = f"Enhance the following prompt for image generation in the style of {style}. Make it more detailed and vivid, while keeping the original intent: '{prompt}'"
+    enhanced_prompt_request = f"""
+    Task: Enhance the following prompt for image generation.
+    Style: {style}
+    Original prompt: '{prompt}'
+    
+    Instructions:
+    1. Expand the prompt to be more detailed and vivid.
+    2. Incorporate elements of the specified style.
+    3. Maintain the original intent of the prompt.
+    4. Provide ONLY the enhanced prompt, without any explanations or options.
+    5. Keep the enhanced prompt concise, ideally under 100 words.
+
+    Enhanced prompt:
+    """
     
     response = model.generate_content(enhanced_prompt_request)
     
-    return response.text
+    # Extract only the enhanced prompt, removing any potential explanations or extra text
+    enhanced_prompt = response.text.strip()
+    
+    # If the response starts with "Enhanced prompt:" or similar, remove it
+    prefixes_to_remove = ["Enhanced prompt:", "Here's the enhanced prompt:", "The enhanced prompt is:"]
+    for prefix in prefixes_to_remove:
+        if enhanced_prompt.lower().startswith(prefix.lower()):
+            enhanced_prompt = enhanced_prompt[len(prefix):].strip()
+    
+    return enhanced_prompt
+
 
 def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt):
     url = "https://api.stability.ai/v2beta/stable-image/generate/sd3"