Refresh front

app.py

@@ -157,47 +157,15 @@ if submit:
 
 st.write('##')
 
-# Optional: Add a subtitle for extra description
-st.subheader("Compare the two images below:")
-
-# Use custom CSS to make the image comparison widget responsive
-# Use custom CSS to make the image comparison widget responsive
-st.markdown(
-    """
-    <style>
-    /* Style the image comparison container */
-    .stImageComparison > div {
-        width: 100%;             /* Make the container responsive */
-        max-width: 90vw;         /* Max width as 90% of the viewport width */
-        margin-left: auto;       /* Center the widget horizontally */
-        margin-right: auto;
-        padding: 10px;           /* Optional padding around the images */
-        border: 3px solid #000;  /* Add a black border around the widget */
-        border-radius: 10px;     /* Optional: Rounded corners */
-        background-color: #f7f7f7;  /* Optional: Background color for contrast */
-    }
-
-    /* Target the images inside the comparison */
-    .stImageComparison > div img {
-        width: 100%;      /* Images take up 100% of the container's width */
-        height: auto;     /* Maintain aspect ratio */
-        border-radius: 8px; /* Optional: Rounded corners for images */
-    }
-    </style>
-    """, 
-    unsafe_allow_html=True
-)
-
-# Display the image comparison widget with the resized images
-image_comparison(
-    st.session_state["output_1"],
-    st.session_state["output_2"]
-)
-
-#col1, col2, col3 = st.columns([1, 10, 1], gap='small')
-#with col2:  
-    #st.markdown(f"#### Object Detection Result")
-    #with st.container(border = True):
+col1, col2, col3 = st.columns([1, 8, 1], gap='small')
+with col2:  
+    st.markdown(f"#### Object Detection Result")
+    with st.container(border = True):
+        tab1, tab2 = st.tabs(['Original Image','Inference Prediction'])
+        with tab1:
+            st.image(st.session_state["output_1"])
+        with tab2:
+            st.image(st.session_state["output_2"])
     #    static_component = image_comparison(
     #    img1=st.session_state["output_1"],
     #    img2=st.session_state["output_2"],