Added Video code

Pages/bgImages.py

@@ -31,7 +31,7 @@ def BGIMAGES():
 
         # Generate the image
         with st.spinner("Generating image..."):
-            image_prompt = f"{prompt}, Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+            image_prompt = f"{prompt}, muted colors, detailed, 8k"
             image = pipeline(image_prompt, generator=generator).images[0]
 
         # Save the image

Pages/createVideo.py

@@ -1,50 +1,27 @@
-from PIL import Image
 import streamlit as st
-import os
+import torch
+from diffusers import StableVideoDiffusionPipeline
+from diffusers.utils import load_image, export_to_video
 
 # Main function
 def CREATEGIF():
-    st.title("Create photoshoot visual")
-
-    # Set the directory where the uploaded images will be saved
-    UPLOAD_DIR = 'uploaded_images'
-
-    # Create the directory if it doesn't exist
-    if not os.path.exists(UPLOAD_DIR):
-        os.makedirs(UPLOAD_DIR)
-
-    # Streamlit app title
-    st.title("Image Upload and Save App")
-
-    # File uploader allows user to upload an image
-    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
-
-    if uploaded_file is not None:
-        # Open the uploaded image
-        image = Image.open(uploaded_file)
-
-        # Display the uploaded image
-        st.image(image, caption='Uploaded Image.', use_column_width=True)
-        
-        # Save the uploaded image to the specified directory
-        image_path = os.path.join(UPLOAD_DIR, uploaded_file.name)
-        image.save(image_path)
-        
-        st.write(f"Image saved at: {image_path}")
-        
-    else:
-        st.write("No image uploaded yet.")
-
-    image_files = [f for f in os.listdir("bgImages") if os.path.isfile(os.path.join("bgImages", f))]
-
-    # Create a dropdown with the list of image files
-    selected_image = st.selectbox("Select an image file", image_files)
-
-    if selected_image:
-        # Display the selected image
-        image_path = os.path.join("bgImages", selected_image)
-        image = Image.open(image_path)
-        st.image(image, caption=f"Selected image: {selected_image}")
+    st.title("Create photoshoot visual video")
+
+    pipeline = StableVideoDiffusionPipeline.from_pretrained(
+        "stabilityai/stable-video-diffusion-img2vid-xt", torch_dtype=torch.float16, variant="fp16"
+    )
+    pipeline.enable_model_cpu_offload()
+
+    image = load_image("output/overlay_image.png")
+    image = image.resize((1024, 576))
+
+    generator = torch.manual_seed(42)
+    frames = pipeline(image, decode_chunk_size=8, generator=generator).frames[0]
+    export_to_video(frames, "output/generated.mp4", fps=7)
+    video_file = open('output/generated.mp4', 'rb')
+    video_bytes = video_file.read()
+
+    st.video(video_bytes)
 
 
 if __name__ == "__main__":

Pages/imageBB.py

@@ -1,4 +1,4 @@
-from segment_anything import sam_model_registry, SamAutomaticMaskGenerator, SamPredictor
+from segment_anything import sam_model_registry, SamPredictor
 import torch
 import streamlit as st
 from Pages.streamlit_img_label import st_img_label

requirements.txt

@@ -7,3 +7,4 @@ pascal_voc_writer
 roboflow 
 dataclasses-json 
 supervision
+torchvision