Update app.py

app.py

@@ -3,28 +3,46 @@ from diffusers import StableDiffusionPipeline
 import imageio
 import os
 
+def get_prompt():
+    """Ensures user provides a valid prompt, avoiding EOFError."""
+    while True:
+        try:
+            prompt = input("Enter your prompt: ").strip()
+            if prompt:
+                return prompt
+            else:
+                print("Prompt cannot be empty. Please enter a valid prompt.")
+        except EOFError:
+            print("\nError: Input required. Restarting prompt entry...")
+            continue
+
 def generate_images(prompt, num_images=5, output_folder="generated_frames"):
+    """Generates images using Stable Diffusion and creates an animated GIF."""
+    
     # Ensure output folder exists
     os.makedirs(output_folder, exist_ok=True)
     
     # Load model (CPU only)
+    print("Loading Stable Diffusion model...")
     pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
     pipe.to("cpu")  # Force to run on CPU
-    
+
     images = []
     for i in range(num_images):
+        print(f"Generating image {i+1}/{num_images}...")
         image = pipe(prompt).images[0]  # Generate image
         image_path = os.path.join(output_folder, f"frame_{i}.png")
         image.save(image_path)
-        images.append(imageio.imread(image_path))
-    
+        images.append(imageio.v3.imread(image_path))  # Use v3 API to avoid deprecation warnings
+
     # Create GIF animation
     gif_path = os.path.join(output_folder, "animation.gif")
     imageio.mimsave(gif_path, images, duration=0.5)
     
-    print(f"Animation saved at {gif_path}")
+    print(f"\n✅ Animation saved at: {gif_path}")
     return gif_path
 
 if __name__ == "__main__":
-    prompt = input("Enter your prompt: ")
+    prompt = get_prompt()  # Get user input safely
     generate_images(prompt)
+