Final Changes

Pages/AboutMe.py

@@ -1,9 +1,9 @@
 import streamlit as st
 
 def ABOUTUS():
-    st.title('About Us')
+    st.title('About Me')
     st.write("""
-    Welcome to our Streamlit app! We are a team of developers passionate about creating 
+    Welcome to the Streamlit app for making a photoshoot visual! We are a team of developers passionate about creating 
     innovative solutions using Streamlit.
     """)
 

Pages/adScript.py

@@ -0,0 +1,27 @@
+import streamlit as st
+import google.generativeai as genai
+
+def CREATEAD():
+    st.title("Create a script/idea")
+
+    prompt = st.text_input("Write about product/Advertisement Idea you already have.")
+
+    if prompt:
+
+        # Or use `os.getenv('GOOGLE_API_KEY')` to fetch an environment variable.
+        GOOGLE_API_KEY = st.secrets["GOOGLE_API_KEY"]
+
+        genai.configure(api_key=GOOGLE_API_KEY)
+
+        model = genai.GenerativeModel('gemini-1.5-flash')
+
+        response = model.generate_content(f"Write a funny short advertisement script for this {prompt}", stream=True)
+        response.resolve()
+
+        st.write(response.text)
+
+
+if __name__ == "__main__":
+    CREATEAD()
+
+

Pages/bgImages.py

@@ -11,14 +11,13 @@ def load_images_from_folder(folder):
             images.append(os.path.join(folder, filename))
     return images
 
-
 # Main function
 def BGIMAGES():
     st.title("Background Images")
 
     st.header('Create a template', divider='orange')
 
-    prompt = st.text_input("Movie title")
+    prompt = st.text_input("Prompt for a Background")
     if prompt:
         # Load the pipeline
         with st.spinner("Loading model..."):
@@ -35,7 +34,7 @@ def BGIMAGES():
             image = pipeline(image_prompt, generator=generator).images[0]
 
         # Save the image
-        output_dir = "bgImages"
+        output_dir = "Assets/bgImages"
         os.makedirs(output_dir, exist_ok=True)
         image_path = os.path.join(output_dir, f"{prompt}.png")
         image.save(image_path)
@@ -46,21 +45,19 @@ def BGIMAGES():
     else:
         st.write("Please enter a movie title to generate an image.")
 
-
-        
     # Path to the folder containing images
-    image_folder = "bgImages"
+    image_folder = "Assets/bgImages"
 
     # Load images from the folder
     images = load_images_from_folder(image_folder)
 
     # Display images and information in a grid layout with three images per row
-    col_width = 350 # Adjust this value according to your preference
+    col_width = 350  # Adjust this value according to your preference
     num_images = len(images)
     images_per_row = 3
     num_rows = (num_images + images_per_row - 1) // images_per_row
 
-    st.header('Available Templates', divider = 'red')
+    st.header('Available Templates', divider='red')
 
     # Display images and information in a grid layout with three images per row
     for i in range(num_rows):
@@ -68,11 +65,10 @@ def BGIMAGES():
         for j in range(images_per_row):
             idx = i * images_per_row + j
             if idx < num_images:
-                cols[j].image(images[idx], width=col_width)
-                cols[j].write(images[idx])
-
+                image_path = images[idx]
+                image_name = os.path.splitext(os.path.basename(image_path))[0]  # Get the file name without extension
+                cols[j].image(image_path, width=col_width)
+                cols[j].write(image_name)
 
 if __name__ == "__main__":
     BGIMAGES()
-
-

Pages/createVideo.py

@@ -12,13 +12,13 @@ def CREATEGIF():
     )
     pipeline.enable_model_cpu_offload()
 
-    image = load_image("output/overlay_image.png")
+    image = load_image("Assets/output/images/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')
+    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()
 
     st.video(video_bytes)

Pages/createVisual.py

@@ -8,14 +8,14 @@ def CREATEVISUALS():
     st.title("Create photoshoot visual")
 
     # Set the directory where the uploaded images will be saved
-    UPLOAD_DIR = 'uploaded_images'
+    UPLOAD_DIR = 'Assets/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.header("Image Upload and Save App")
+    st.header("Upload image")
 
     # File uploader allows user to upload an image
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])

Pages/createdVisuals.py

@@ -0,0 +1,43 @@
+import os
+import streamlit as st
+
+def load_images_from_folder(folder):
+    images = []
+    for filename in os.listdir(folder):
+        if filename.endswith(".jpg") or filename.endswith(".png"):
+            images.append(os.path.join(folder, filename))
+    return images
+
+
+# Main function
+def CREATEDIMAGES():
+    st.title("Created Images")
+ 
+    # Path to the folder containing images
+    image_folder = "Assets/output/images"
+
+    # Load images from the folder
+    images = load_images_from_folder(image_folder)
+
+    # Display images and information in a grid layout with three images per row
+    col_width = 350 # Adjust this value according to your preference
+    num_images = len(images)
+    images_per_row = 3
+    num_rows = (num_images + images_per_row - 1) // images_per_row
+
+    st.header('Available Templates', divider = 'red')
+
+    # Display images and information in a grid layout with three images per row
+    for i in range(num_rows):
+        cols = st.columns(images_per_row)
+        for j in range(images_per_row):
+            idx = i * images_per_row + j
+            if idx < num_images:
+                cols[j].image(images[idx], width=col_width)
+                cols[j].write(images[idx])
+
+
+if __name__ == "__main__":
+    CREATEDIMAGES()
+
+

Pages/imageBB.py

@@ -51,7 +51,7 @@ def run(img_path):
         rects = st_img_label(resized_img, box_color="red", rects=st.session_state.rects)
         st.session_state.rects = rects
     else:
-        st.image(resized_img, caption="Annotated Image",width=300, use_column_width=True)
+        st.image(resized_img, caption="Uploaded Image",width=300, use_column_width=True)
 
         for rect in st.session_state.rects:
             # st.write(f"Rectangle: {rect}")
@@ -60,7 +60,7 @@ def run(img_path):
 
                 masks = get_masks(rect,img_path)
 
-                save_dir = "saved_masks"
+                save_dir = "Assets/saved_masks"
                 if not os.path.exists(save_dir):
                     os.makedirs(save_dir)
 
@@ -71,7 +71,7 @@ def run(img_path):
 
                 print(f"Inverted masks saved to directory: {save_dir}")
 
-            image_files = [f for f in os.listdir("bgImages") if os.path.isfile(os.path.join("bgImages", f))]
+            image_files = [f for f in os.listdir("Assets/bgImages") if os.path.isfile(os.path.join("Assets/bgImages", f))]
 
             st.header("Template Selection")
             # Create a dropdown with the list of image files
@@ -79,7 +79,7 @@ def run(img_path):
 
             if selected_image:
                 # Display the selected image
-                image_pathBG = os.path.join("bgImages", selected_image)
+                image_pathBG = os.path.join("Assets/bgImages", selected_image)
                 image = Image.open(image_pathBG)
                 st.image(image, width=300, caption=f"Selected image: {selected_image}")
 
@@ -153,7 +153,7 @@ def run(img_path):
                     overlay_image[:, :, 3] = np.clip(overlay[:, :, 3] + background_bgra[:, :, 3], 0, 255)
 
                     # Save the result
-                    output_path = 'output/overlay_image.png'
+                    output_path = 'Assets/output/images/overlay_image.png'
                     cv2.imwrite(output_path, overlay_image)
 
                     # Display the overlay image

Pages/objectRecognize.py

@@ -0,0 +1,49 @@
+import PIL.Image
+import streamlit as st
+import google.generativeai as genai
+import os
+from PIL import Image
+
+def DETECTIMAGE():
+    st.title("Create photoshoot visual")
+
+    # Set the directory where the uploaded images will be saved
+    UPLOAD_DIR = 'Assets/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.header("Upload Image")
+
+    # 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:
+        with st.spinner('Processing the image...'):
+            # Open the uploaded image
+            image = Image.open(uploaded_file)
+
+            # Save the uploaded image to the specified directory
+            image_path = os.path.join(UPLOAD_DIR, uploaded_file.name)
+            image.save(image_path)
+
+            img = PIL.Image.open(image_path)
+            st.image(img, caption="Uploaded Image")
+
+            # Or use `os.getenv('GOOGLE_API_KEY')` to fetch an environment variable.
+            GOOGLE_API_KEY = st.secrets["GOOGLE_API_KEY"]
+
+            genai.configure(api_key=GOOGLE_API_KEY)
+
+            model = genai.GenerativeModel('gemini-1.5-flash')
+
+            # Generate content
+            response = model.generate_content(["Is the image anything from this list? (Shoe,Sneaker, Bottle, Cup, Sandal, Perfume, Toy, Sunglasses, Car, Water Bottle,Chair, Office Chair, Can, Cap, Hat, Couch, Wristwatch, Glass, Bag, Handbag, Baggage, Suitcase, Headphones, Jar, Vase) If yes then give only the object name, if no simply say 'Not Recognized' and dont give any fullstop at the end.", img], stream=True)
+            response.resolve()
+
+            st.subheader(f"Output: {response.text}")
+
+if __name__ == "__main__":
+    DETECTIMAGE()

app.py

@@ -3,7 +3,10 @@ import streamlit as st
 from Pages.createVisual import CREATEVISUALS
 from Pages.AboutMe import ABOUTUS
 from Pages.bgImages import BGIMAGES
+from Pages.createdVisuals import CREATEDIMAGES
 from Pages.createVideo import CREATEGIF
+from Pages.objectRecognize import DETECTIMAGE
+from Pages.adScript import CREATEAD
 
 
 st.set_page_config(
@@ -13,19 +16,24 @@ st.set_page_config(
         initial_sidebar_state="expanded"  # 'auto', 'expanded', 'collapsed'
     )
 
+def MAIN():
+    weights_dir = 'weights'
+    weights_file = os.path.join(weights_dir, 'sam_vit_h_4b8939.pth')
 
-def MAIN(): 
+    # Check if the weights file already exists
+    if not os.path.exists(weights_file):
+        # Create the directory if it doesn't exist
+        if not os.path.exists(weights_dir):
+            os.makedirs(weights_dir)
+        
+        # Define the download command
+        download_command = f"wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth -P {weights_dir}"
 
-    folder_Command = "mkdir -p weights"
-
-    # Define the download command
-    download_command = "wget -q https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth -P weights"
-
-    # Execute the command
-
-    os.system(folder_Command)
-
-    os.system(download_command)
+        # Execute the download command
+        os.system(download_command)
+        print("Weights downloaded.")
+    else:
+        print("Weights file already exists. No download needed.")
 
     # Verify the file is downloaded
     file_path = os.path.expanduser("~/weights/sam_vit_h_4b8939.pth")
@@ -37,7 +45,7 @@ def MAIN():
     # Rest of your application code
 
     st.sidebar.title('LumiereIQ')
-    app = st.sidebar.selectbox('', ['Create Visuals','Background Images','Create Video', 'About Me'])
+    app = st.sidebar.selectbox('', ['Create Visuals','Background Images','Recognize Object', 'Create Script', 'Created Images','Create Video', 'About Me'])
     if app == "Create Visuals":
         CREATEVISUALS()
     elif app == "About Me":
@@ -46,6 +54,11 @@ def MAIN():
         BGIMAGES()
     elif app == "Create Video":
         CREATEGIF()
-
+    elif app == "Created Images":
+        CREATEDIMAGES()
+    elif app == "Recognize Object":
+        DETECTIMAGE()
+    elif app == "Create Script":
+        CREATEAD()
           
 MAIN()

requirements.txt

@@ -8,3 +8,4 @@ roboflow
 dataclasses-json 
 supervision
 torchvision
+google-generativeai