Update app.py

app.py

@@ -9,8 +9,8 @@ import time
 device = "cpu"
 print(f"Using device: {device}")
 
-# Load a smaller, CPU-friendly PUBLIC model
-model_id = "dreamlike-art/dreamlike-diffusion-1.0"
+# Load a HIGHER QUALITY model
+model_id = "stabilityai/stable-diffusion-2-1"
 
 print("Loading pipeline... This may take a few minutes.")
 try:
@@ -32,24 +32,39 @@ try:
 
 except Exception as e:
     print(f"Error loading model: {e}")
-    raise e
+    # Fallback to original model if needed
+    model_id = "dreamlike-art/dreamlike-diffusion-1.0"
+    pipe = StableDiffusionPipeline.from_pretrained(
+        model_id,
+        torch_dtype=torch.float32,
+        use_safetensors=True,
+        safety_checker=None,
+        requires_safety_checker=False
+    )
+    pipe = pipe.to(device)
+    print(f"Fell back to {model_id}")
 
 # Define the image generation function
 def generate_image(prompt):
     """
     This function takes a text prompt and returns a generated image.
     """
-    # Add a consistent style to all prompts
-    enhanced_prompt = f"children's book illustration, watercolor style, cute, whimsical, {prompt}"
+    # PROFESSIONAL-QUALITY PROMPT ENHANCEMENT
+    enhanced_prompt = f"masterpiece, best quality, 4K, ultra detailed, photorealistic, sharp focus, studio lighting, professional photography, {prompt}"
+    
+    # Remove any negative aspects (optional but improves quality)
+    negative_prompt = "blurry, low quality, low resolution, watermark, signature, text, ugly, deformed"
+    
     print(f"Generating image for prompt: {enhanced_prompt}")
     
-    # Generate the image
+    # Generate the image with better settings
     image = pipe(
         prompt=enhanced_prompt,
+        negative_prompt=negative_prompt,  # Added negative prompt
         width=512,
         height=512,
-        guidance_scale=7.5,
-        num_inference_steps=20,
+        guidance_scale=9.0,  # Slightly higher for better prompt adherence
+        num_inference_steps=25,  # More steps for better quality
         generator=torch.Generator(device=device)
     ).images[0]
     
@@ -66,14 +81,14 @@ demo = gr.Interface(
     inputs=gr.Textbox(
         label="Enter your scene description",
         lines=2,
-        placeholder="A brave little mouse exploring a giant forest..."
+        placeholder="A dragon reading a book under a magical tree"
     ),
     outputs=gr.Image(label="Generated Illustration", type="pil"),
-    title="Children's Book Illustrator (CPU Edition) 🤖🎨",
-    description="This free version runs on CPU. It's slower but gets the job done! Enter a scene description."
+    title="Premium Children's Book Illustrator 🤖🎨",
+    description="Generating high-quality, sharp, detailed images for your stories. Enter a scene description."
 )
 
-# Launch the app with simple settings
+# Launch the app
 demo.launch(
     debug=True,
     server_name="0.0.0.0",