Added folder saving logics

Pages/createVideo.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import os
 import torch
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import load_image, export_to_video
@@ -17,6 +18,9 @@ def CREATEGIF():
 
     generator = torch.manual_seed(42)
     frames = pipeline(image, decode_chunk_size=8, generator=generator).frames[0]
+
+    if not os.path.exists("Assets/output/videos"):
+        os.makedirs("Assets/output/videos")
     export_to_video(frames, "Assets/output/videos/generated.mp4", fps=7)
     video_file = open('Assets/output/videos/generated.mp4', 'rb')
     video_bytes = video_file.read()

Pages/imageBB.py

@@ -152,6 +152,8 @@ def run(img_path):
             # Prompt user for the filename
             filename = st.text_input("Enter a name to save the image:")
             if filename and st.button("Save Image"):
+                if not os.path.exists("Assets/output/images"):
+                    os.makedirs("Assets/output/images")
                 output_path = f'Assets/output/images/{filename}.png'
                 cv2.imwrite(output_path, overlay_image)