Update app.py

app.py

@@ -59,10 +59,8 @@ app.layout = dbc.Container([
                                     ],
                                     value="1:1"
                                 ),
-                                dbc.Label("CFG Scale"),
-                                dcc.Slider(id="cfg-scale", min=1, max=10, step=0.1, value=4, marks={1: '1', 5: '5', 10: '10'}),
                                 dbc.Label("Steps"),
-                                dcc.Slider(id="steps", min=4, max=50, step=1, value=30, marks={4: '4', 25: '25', 50: '50'}),
+                                dcc.Slider(id="steps", min=4, max=50, step=1, value=20, marks={4: '4', 25: '25', 50: '50'}),
                             ],
                             title="Advanced Settings",
                         ),
@@ -95,17 +93,17 @@ def enhance_prompt(google_api_key, prompt, style):
     genai.configure(api_key=google_api_key)
     model = genai.GenerativeModel("gemini-2.0-flash-lite")
     enhanced_prompt_request = f"""
-    Task: Enhance the following prompt for high-quality, realistic image generation.
+    Task: Enhance the following prompt with details to match the specified style
     Style: {style}
     Original prompt: '{prompt}'
     
     Instructions:
     1. Expand the prompt to be more detailed, vivid, and realistic with camera used and the setting for that camera like ISO etc.
-    2. Incorporate elements of the specified style, focusing on realism and natural appearances.
-    3. Add details that enhance the realism of the scene, especially for elements like trees, textures, and lighting.
-    4. Emphasize natural lighting and enhance the realism of textures and colors.
+    2. Incorporate elements of the specified style.
+    3. Add details that enhance the scene to the specified style
+    4. Emphasize natural lighting and enhance the realism of textures and colors based on the specified style.
     5. Avoid terms that might result in artificial or cartoonish appearance unless specified by user.
-    6. Maintain the original intent of the prompt while significantly improving its descriptive quality.
+    6. Maintain the original intent of the prompt while significantly improving its descriptive quality with details.
     7. Provide ONLY the enhanced prompt, without any explanations or options.
     8. Keep the enhanced prompt concise, ideally under 100 words.
 
@@ -128,7 +126,7 @@ def enhance_prompt(google_api_key, prompt, style):
         logging.error(f"Error in enhance_prompt: {str(e)}")
         raise
 
-def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt, cfg_scale=7, steps=30, aspect_ratio="1:1"):
+def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt, steps=30, aspect_ratio="1:1"):
     url = "https://api.stability.ai/v2beta/stable-image/generate/sd3"
     
     headers = {
@@ -137,13 +135,13 @@ def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt, c
     }
     
     data = {
-        "prompt": f"{enhanced_prompt}, Style: {style}, highly detailed, photorealistic, high quality, natural lighting, realistic textures and colors",
+        "prompt": f"{enhanced_prompt}, Style: {style}, highly detailed, high quality, descriptive",
         "negative_prompt": negative_prompt,
         "model": "sd3.5-large-turbo",
         "output_format": "jpeg",
         "num_images": 1,
         "steps": steps,
-        # "cfg_scale": cfg_scale,
+        "cfg_scale": 7,  # Default value
         "style_preset": style,
         "aspect_ratio": aspect_ratio,
     }
@@ -169,7 +167,7 @@ def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt, c
         logging.error(f"Request failed: {str(e)}")
         raise Exception(f"Request failed: {str(e)}")
 
-def process_and_generate(google_api_key, stability_api_key, prompt, style, cfg_scale, steps, aspect_ratio, set_status):
+def process_and_generate(google_api_key, stability_api_key, prompt, style, steps, aspect_ratio, set_status):
     try:
         set_status("Enhancing prompt...")
         enhanced_prompt = enhance_prompt(google_api_key, prompt, style)
@@ -178,7 +176,7 @@ def process_and_generate(google_api_key, stability_api_key, prompt, style, cfg_s
         max_attempts = 3
         for attempt in range(max_attempts):
             try:
-                image_bytes = generate_image(stability_api_key, enhanced_prompt, style, DEFAULT_NEGATIVE_PROMPT, cfg_scale, steps, aspect_ratio)
+                image_bytes = generate_image(stability_api_key, enhanced_prompt, style, DEFAULT_NEGATIVE_PROMPT, steps, aspect_ratio)
                 set_status("Image generated successfully!")
                 return image_bytes, enhanced_prompt
             except Exception as e: