Update app.py

app.py

@@ -12,7 +12,7 @@ app.prepare(ctx_id=0, det_size=(640, 640))  # Ensure the model runs on the CPU i
 # Load the face swapping model (downloading on-demand)
 swapper = insightface.model_zoo.get_model('insightface/inswapper_128.onnx', download=True)
 
-# Face swapping function
+# Face swapping function with added smoothing
 def swap_faces(destination_image, source_image):
     # Load the destination and source images from Streamlit inputs
     img = cv2.cvtColor(np.array(destination_image), cv2.COLOR_RGB2BGR)
@@ -41,10 +41,13 @@ def swap_faces(destination_image, source_image):
                        [0, -1, 0]])
     sharpened_res = cv2.filter2D(res, -1, kernel)
 
+    # Add smoothing (blurring) for a more natural look
+    smoothed_res = cv2.GaussianBlur(sharpened_res, (5, 5), 0)
+
     # Convert the result to RGB for display in Streamlit
-    sharpened_res_rgb = cv2.cvtColor(sharpened_res, cv2.COLOR_BGR2RGB)
+    smoothed_res_rgb = cv2.cvtColor(smoothed_res, cv2.COLOR_BGR2RGB)
     
-    return sharpened_res_rgb
+    return smoothed_res_rgb
 
 # Streamlit app layout
 def main():
@@ -72,15 +75,8 @@ def main():
     st.header("Face Swapping App :camera:")
     st.write("Welcome to the face swapping app! Upload two images and watch the magic happen!")
 
-    # Sidebar for instructions and file uploads
+    # Sidebar for file uploads and instructions
     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"])
@@ -102,6 +98,14 @@ def main():
                         st.session_state.result = result  # Save the result in session state
                 st.success("Face swapping complete!")
 
+        # Instructions on how to use the app (below upload buttons and swap button)
+        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.
+        """)
+
     # 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)