Update pages/📸_InstaMuse.py

pages/📸_InstaMuse.py

@@ -20,34 +20,48 @@ with st.sidebar:
     st.write("Start turning heads with your posts. Use InstaMuse now and watch your likes soar!")
 
 # Main page content
-st.markdown('<p class="big-font">InstaMuse Photo Caption Generator</p>', unsafe_allow_html=True)
+st.markdown('## InstaMuse Photo Caption Generator')
 st.write("""### Upload your photo below and spark some caption magic! ଘ(੭ˊᵕˋ)੭* ੈ✩‧₊""")
 
 st.write(
-'**Notice**: *Due to model load time, it may take a moment to load the application. This only occurs once.* *\n*Subsequent uploads will be faster.*')
-
-
+f'**Notice**: *Due to model load time, it may take a moment to load the application. This only occurs once.* \n*Subsequent uploads will be faster.*'
+)
 
 # Load the model for image description
 with st.spinner('Loading Application, this make take a minute :hourglass_flowing_sand:'):
     # Load the model for image captioning
     model, processor = utils.init_model()
 
+
+def clear_cache():
+    """
+    Function to clear the session state cache.
+    """
+    if 'file' in st.session_state:
+        del st.session_state['file']
+    if 'captions' in st.session_state:
+        del st.session_state['captions']
+    if 'caption_list' in st.session_state:
+        del st.session_state['caption_list']
+    if 'img_description' in st.session_state:
+        del st.session_state['img_description']
+    st.rerun()
+
 # Upload image file and process image
 uploaded_file = st.file_uploader(
-    "Upload your image here:", type=["jpg", "png"], help="Only jpg and png images are supported"
+    "Upload your image here:", type=["jpg", "png"],
+    help="Only jpg and png images are supported"
 )
 
 
-if uploaded_file is not None:
-    if 'file' not in st.session_state or uploaded_file != st.session_state['file']:
+if uploaded_file:
+    if 'file' not in st.session_state:
         st.session_state['file'] = uploaded_file
+        image = Image.open(uploaded_file)
+        image.thumbnail((400, 400), Image.Resampling.LANCZOS)
+        st.session_state['image'] = image
 
-    image = Image.open(st.session_state['file'])
-    image.thumbnail((400, 400), Image.Resampling.LANCZOS)
-    st.session_state['image'] = image
     col1, col2 = st.columns(2)
-
     with st.container():
         with col1:
             st.markdown("## 📸 Your Image:")
@@ -61,7 +75,18 @@ if uploaded_file is not None:
                     st.session_state['captions'] = captions
                     st.session_state['caption_list'] = caption_list
                     st.session_state['img_description'] = img_description
-                    st.empty()
+                    st.markdown("## 📝 Generated Captions:")
+                    for caption in st.session_state['caption_list']:
+                        if caption.strip() != "":
+                            st.info(f"##### {caption}")
+
+                elif st.session_state['file'] != uploaded_file:
+                    clear_cache()
+                    desc = caption_generator.image_2_text(image, model, processor)
+                    captions, caption_list, img_description = caption_generator.text_2_caption(desc)
+                    st.session_state['captions'] = captions
+                    st.session_state['caption_list'] = caption_list
+                    st.session_state['img_description'] = img_description
                     st.markdown("## 📝 Generated Captions:")
                     for caption in st.session_state['caption_list']:
                         if caption.strip() != "":