removed column deprecation param

pages/Semantic_Search.py

@@ -287,97 +287,97 @@ if(search_all_type==True):
 
    
 def generate_images(tab,inp_):
-        #write_top_bar()
-        seed = random.randint(1, 10)
-        request = json.dumps(
-                    {
-                        "taskType": "TEXT_IMAGE",
-                        "textToImageParams": {"text": st.session_state.image_prompt},
-                        "imageGenerationConfig": {
-                            "numberOfImages": 3,
-                            "quality": "standard",
-                            "cfgScale": 8.0,
-                            "height": 512,
-                            "width": 512,
-                            "seed": seed,
-                        },
-                    }
-                )
-
-        if(inp_!=st.session_state.image_prompt):
-            print("call bedrocck")
-            response = bedrock_.invoke_model(
-            modelId="amazon.titan-image-generator-v1", body=request
+    #write_top_bar()
+    seed = random.randint(1, 10)
+    request = json.dumps(
+                {
+                    "taskType": "TEXT_IMAGE",
+                    "textToImageParams": {"text": st.session_state.image_prompt},
+                    "imageGenerationConfig": {
+                        "numberOfImages": 3,
+                        "quality": "standard",
+                        "cfgScale": 8.0,
+                        "height": 512,
+                        "width": 512,
+                        "seed": seed,
+                    },
+                }
             )
-            
-            response_body = json.loads(response["body"].read())
-            st.session_state.img_gen = response_body["images"]
-        gen_images_dir = os.path.join(parent_dirname, "gen_images")
-        if os.path.exists(gen_images_dir):
-            shutil.rmtree(gen_images_dir)
-        os.mkdir(gen_images_dir)
-        width_ = 200
-        height_ = 200
-        index_ = 0
-        #if(inp_!=st.session_state.image_prompt):
+
+    if(inp_!=st.session_state.image_prompt):
+        print("call bedrocck")
+        response = bedrock_.invoke_model(
+        modelId="amazon.titan-image-generator-v1", body=request
+        )
         
-        if(len(st.session_state.img_gen)==0 and st.session_state.clear_ == True):
-            #write_top_bar()
-            placeholder1 = st.empty()
+        response_body = json.loads(response["body"].read())
+        st.session_state.img_gen = response_body["images"]
+    gen_images_dir = os.path.join(parent_dirname, "gen_images")
+    if os.path.exists(gen_images_dir):
+        shutil.rmtree(gen_images_dir)
+    os.mkdir(gen_images_dir)
+    width_ = 200
+    height_ = 200
+    index_ = 0
+    #if(inp_!=st.session_state.image_prompt):
+    
+    if(len(st.session_state.img_gen)==0 and st.session_state.clear_ == True):
+        #write_top_bar()
+        placeholder1 = st.empty()
+        with tab:
+            with placeholder1.container():
+                st.empty()
+
+    images_dis = []
+    for image_ in st.session_state.img_gen:
+        st.session_state.radio_disabled  = False
+        if(index_==0):
+            # with tab:
+            #     rad1, rad2,rad3  = st.columns([98,1,1])
+            # if(st.session_state.input_rad_1 is None):
+            #     rand_ = ""
+            # else:
+            #     rand_ = st.session_state.input_rad_1
+            # if(inp_!=st.session_state.image_prompt+rand_):
+            #     with rad1:
+            #         sel_rad_1 = st.radio("Choose one image", ["1","2","3"],index=None, horizontal = True,key = 'input_rad_1')
+
             with tab:
-                with placeholder1.container():
-                    st.empty()
-
-        images_dis = []
-        for image_ in st.session_state.img_gen:
-            st.session_state.radio_disabled  = False
-            if(index_==0):
-                # with tab:
-                #     rad1, rad2,rad3  = st.columns([98,1,1])
-                # if(st.session_state.input_rad_1 is None):
-                #     rand_ = ""
-                # else:
-                #     rand_ = st.session_state.input_rad_1
-                # if(inp_!=st.session_state.image_prompt+rand_):
-                #     with rad1:
-                #         sel_rad_1 = st.radio("Choose one image", ["1","2","3"],index=None, horizontal = True,key = 'input_rad_1')
-
-                with tab:
-                    #sel_image = st.radio("", ["1","2","3"],index=None, horizontal = True)
-                    if(st.session_state.img_container!=""):
-                        st.session_state.img_container.empty()
-                    place_ = st.empty()
-                    img1, img2,img3  = place_.columns([30,30,30])
-                    st.session_state.img_container = place_
-                img_arr = [img1, img2,img3]
-            
-            base64_image_data = image_
+                #sel_image = st.radio("", ["1","2","3"],index=None, horizontal = True)
+                if(st.session_state.img_container!=""):
+                    st.session_state.img_container.empty()
+                place_ = st.empty()
+                img1, img2,img3  = place_.columns([30,30,30])
+                st.session_state.img_container = place_
+            img_arr = [img1, img2,img3]
+        
+        base64_image_data = image_
 
-            #st.session_state.gen_image_str = base64_image_data
+        #st.session_state.gen_image_str = base64_image_data
+
+        print("perform multimodal search")
+    
+        Image.MAX_IMAGE_PIXELS = 100000000
+        filename = st.session_state.image_prompt+"_gen_"+str(index_)
+        photo = parent_dirname+"/gen_images/"+filename+'.jpg' 
+        imgdata = base64.b64decode(base64_image_data)
+        with open(photo, 'wb') as f:
+            f.write(imgdata) 
 
-            print("perform multimodal search")
         
-            Image.MAX_IMAGE_PIXELS = 100000000
-            filename = st.session_state.image_prompt+"_gen_"+str(index_)
-            photo = parent_dirname+"/gen_images/"+filename+'.jpg'  # I assume you have a way of picking unique filenames
-            imgdata = base64.b64decode(base64_image_data)
-            with open(photo, 'wb') as f:
-                f.write(imgdata) 
+        
+        with Image.open(photo) as image:    
+            file_type = 'jpg'
+            path = image.filename.rsplit(".", 1)[0]
+            image.thumbnail((width_, height_))
+            image.save(parent_dirname+"/gen_images/"+filename+"-resized_display."+file_type)
 
-            
-            
-            with Image.open(photo) as image:    
-                file_type = 'jpg'
-                path = image.filename.rsplit(".", 1)[0]
-                image.thumbnail((width_, height_))
-                image.save(parent_dirname+"/gen_images/"+filename+"-resized_display."+file_type)
-
-            with img_arr[index_]:
-                placeholder_ = st.empty()
-                placeholder_.image(parent_dirname+"/gen_images/"+filename+"-resized_display."+file_type)
-
-            index_ = index_ + 1
-      
+        with img_arr[index_]:
+            placeholder_ = st.empty()
+            placeholder_.image(parent_dirname+"/gen_images/"+filename+"-resized_display."+file_type)
+
+        index_ = index_ + 1
+    
 
 
 def handle_input():