Update app.py

app.py

@@ -63,6 +63,10 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
+# Initialize session state to block re-running
+if 'has_run' not in st.session_state:
+    st.session_state.has_run = False
+
 # Main UI container
 st.markdown('<div class="main">', unsafe_allow_html=True)
 st.markdown('<div class="title">Florence-2 Image Captioning Demo</div>', unsafe_allow_html=True)
@@ -71,9 +75,9 @@ st.markdown('<div class="title">Florence-2 Image Captioning Demo</div>', unsafe_
 task_prompt = st.text_input("Task Prompt", value="Describe the image in detail:")
 
 # Image upload area
-st.markdown('<div class="upload-area">', unsafe_allow_html=True)
-uploaded_image = st.file_uploader("Upload your image here", type=["jpg", "jpeg", "png"])
-st.markdown('</div>', unsafe_allow_html=True)
+st.sidebar.markdown('<div class="upload-area">', unsafe_allow_html=True)
+uploaded_image = st.sidebar.file_uploader("Upload your image here", type=["jpg", "jpeg", "png"])
+st.sidebar.markdown('</div>', unsafe_allow_html=True)
 
 # Additional text input (optional)
 text_input = st.text_area("Additional Text Input (Optional)", height=150)
@@ -84,7 +88,10 @@ if uploaded_image is not None:
     st.image(image, caption="Uploaded Image", use_column_width=True)
 
     # Generate Caption button
-    if st.button("Generate Caption", key="generate"):
+    if st.button("Generate Caption", key="generate") and not st.session_state.has_run:
+        # Mark that the script has been run
+        st.session_state.has_run = True
+
         # Assuming `run_example` function is defined
         result = run_example(task_prompt, image, text_input)
 
@@ -104,4 +111,10 @@ if uploaded_image is not None:
             processed_image = draw_polygons(image.copy(), result)
             st.image(processed_image, caption="Image with Polygons", use_column_width=True)
 
+# Clear session state button
+if st.session_state.has_run:
+    if st.button("Clear and Upload New Image"):
+        st.session_state.has_run = False
+        st.experimental_rerun()
+
 st.markdown('</div>', unsafe_allow_html=True)