Update app.py

app.py

@@ -140,6 +140,15 @@ def delete_image(image_path):
     except Exception as e:
         st.error(f"Error al borrar la imagen: {e}")
 
+def swap_faces(image_path):
+    try:
+        client = Client("deepfakes/faceswap")  
+        result = client.predict(input_image=handle_file(image_path))
+        return result[1] if isinstance(result, list) and len(result) > 1 else None
+    except Exception as e:
+        st.error(f"Error en el faceswap: {e}")
+        return None
+
 def main():
     st.set_page_config(layout="wide")
     prompt = st.sidebar.text_input("Descripción de la imagen", max_chars=900)
@@ -196,12 +205,19 @@ def main():
             
             if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
                 try:
-                    os.remove(file)  
-                    if prompt_file:
-                        os.remove(prompt_file)
-                    st.success(f"Imagen {idx+1} y su prompt fueron borrados.")
+                    os.remove(prompt_file)  
+                    st.success(f"Imagen {idx+1} borrada.")
                 except Exception as e:
-                    st.error(f"Error al borrar la imagen o prompt: {e}")
+                    st.error(f"Error al borrar la imagen: {e}")
+
+            if st.button(f"Swap Face {idx+1}", key=f"swap_{idx}"):
+                with st.spinner(f"Swapping faces en Imagen {idx+1}..."):
+                    swapped_image_path = swap_faces(file)
+                    if swapped_image_path:
+                        swapped_image = Image.open(swapped_image_path)
+                        st.image(swapped_image, caption=f"Imagen con Face Swap {idx+1}")
+                    else:
+                        st.error("Error en el face swap.")
 
 if __name__ == "__main__":
-    main() 
+    main()