app.py

app.py

@@ -5,23 +5,33 @@ import moviepy.editor as mp
 import os
 
 def generate_image(prompt):
-    model_id = "runwayml/stable-diffusion-v1-5"
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
-    pipe.to("cpu")
-    image = pipe(prompt).images[0]
-    image_path = "generated_image.png"
-    image.save(image_path)
-    return image_path
+    try:
+        model_id = "runwayml/stable-diffusion-v1-5"
+        pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+        pipe.to("cpu")
+        image = pipe(prompt).images[0]
+        image_path = "generated_image.png"
+        image.save(image_path)
+        return image_path
+    except Exception as e:
+        return f"Error generating image: {str(e)}"
 
 def create_video(image_path, output_video="output.mp4", duration=5):
-    clip = mp.ImageClip(image_path, duration=duration)
-    clip = clip.set_fps(24)
-    clip.write_videofile(output_video, codec='libx264')
-    return output_video
+    try:
+        clip = mp.ImageClip(image_path, duration=duration)
+        clip = clip.set_fps(24)
+        clip.write_videofile(output_video, codec='libx264')
+        return output_video
+    except Exception as e:
+        return f"Error creating video: {str(e)}"
 
 def text_to_video(prompt):
     image_path = generate_image(prompt)
+    if "Error" in image_path:
+        return image_path
     video_path = create_video(image_path)
+    if "Error" in video_path:
+        return video_path
     return video_path
 
 demo = gr.Interface(