Update app.py

app.py

@@ -8,12 +8,6 @@ st.markdown("""
             background-color: #f7f9fc;
             font-family: 'Arial', sans-serif;
         }
-        .main {
-            background-color: #ffffff;
-            padding: 30px;
-            border-radius: 15px;
-            box-shadow: 0px 0px 15px rgba(0, 0, 0, 0.1);
-        }
         .title {
             color: #2c3e50;
             font-size: 40px;
@@ -68,49 +62,30 @@ 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)
 
-# Task prompt input
-task_prompt = st.text_input("Task Prompt", value="Describe the image in detail:")
+
 
 # Image upload area
 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)
 
 # Display the uploaded image and process it if available
 if uploaded_image is not None:
+    # Task prompt input
+    task_prompt = st.text_input("Task Prompt", value="Describe the image in detail:")
+    # Additional text input (optional)
+    text_input = st.text_area("Additional Text Input (Optional)", height=150)
     image = Image.open(uploaded_image)
     st.image(image, caption="Uploaded Image", use_column_width=True)
 
     # Generate Caption button
-    if st.button("Generate Caption", key="generate") and not st.session_state.has_run:
+    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)
-
-        # Display the generated caption
-        st.markdown('<div class="result-section">', unsafe_allow_html=True)
-        st.markdown('<div class="result-title">Generated Caption</div>', unsafe_allow_html=True)
-        st.markdown(f'<div class="success">{result["text"]}</div>', unsafe_allow_html=True)
-
-        # Display bounding boxes or polygons if available
-        if "bboxes" in result:
-            st.markdown("### Detected Objects")
-            fig = plot_bbox(image, result)
-            st.pyplot(fig)
-
-        if "polygons" in result:
-            st.markdown("### Image with Polygons")
-            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"):