Update app.py

app.py

@@ -68,12 +68,19 @@ def main():
         </style>
     """, unsafe_allow_html=True)
 
-    # Page Header
+    # Page Header (outside sidebar)
     st.header("Face Swapping App :camera:")
     st.write("Welcome to the face swapping app! Upload two images and watch the magic happen!")
 
-    # Sidebar for file uploads
+    # Sidebar for instructions and file uploads
     with st.sidebar:
+        st.subheader("How to Use the App:")
+        st.markdown("""
+            1. **Upload the destination and source images**: Select the images you want to use for face swapping.
+            2. **Click on the "Swap Faces" button**: The app will process the images and swap the faces.
+            3. **View the output**: The swapped face image will appear below after processing.
+        """)
+        
         st.subheader("Upload Images for Face Swapping")
         destination_image = st.file_uploader("Upload Destination Image", type=["jpg", "png", "jpeg"])
         source_image = st.file_uploader("Upload Source Image", type=["jpg", "png", "jpeg"])
@@ -92,16 +99,12 @@ def main():
                     if isinstance(result, str):
                         st.error(result)  # Display error message if no faces detected or memory error
                     else:
-                        st.image(result, caption="Swapped Face", use_container_width=True)
+                        st.session_state.result = result  # Save the result in session state
                 st.success("Face swapping complete!")
 
-    # Instructions on how to use the app
-    st.markdown("### How to Use the App:")
-    st.markdown("""
-        1. **Upload the destination and source images**: Select the images you want to use for face swapping.
-        2. **Click on the "Swap Faces" button**: The app will process the images and swap the faces.
-        3. **View the output**: The swapped face image will appear below after processing.
-    """)
+    # Main content area for displaying the swapped face image
+    if 'result' in st.session_state:
+        st.image(st.session_state.result, caption="Swapped Face", use_container_width=True)
 
     # Footer with credits
     st.markdown("Developed with ❤ by [Your Name]")