Update app.py

app.py

@@ -142,11 +142,10 @@ def delete_image(image_path):
 
 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
+        swapped_image_path = image_path 
+        return swapped_image_path
     except Exception as e:
-        st.error(f"Error en el faceswap: {e}")
+        st.error(f"Error en el face swap: {e}")
         return None
 
 def main():
@@ -203,21 +202,13 @@ def main():
             st.image(image, caption=f"Imagen {idx+1}")
             st.write(f"Prompt: {prompt_text}")
             
-            if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
-                try:
-                    os.remove(prompt_file)  
-                    st.success(f"Imagen {idx+1} borrada.")
-                except Exception as 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 st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx+1}"):
+                delete_image(file)
+            
+            if st.button(f"Swap Face {idx+1}", key=f"swap_{idx+1}"):
+                swapped_image_path = swap_faces(file)
+                if swapped_image_path:
+                st.image(swapped_image_path, caption=f"Imagen {idx+1} con Face Swap")
 
 if __name__ == "__main__":
-    main()
+    main()