Update vton.py

vton.py

@@ -1,6 +1,7 @@
 import replicate
 import requests
 import os
+import base64
 
 def virtual_try_on(garm_img_url, human_img_url, garment_des):
     """
@@ -14,15 +15,13 @@ def virtual_try_on(garm_img_url, human_img_url, garment_des):
     Returns:
         str: Path to the saved output image
     """
-    # Convert local files to data URLs if needed
+    # Convert local files to data URLs
     def file_to_data_url(file_path):
-        if file_path.startswith('http'):
-            response = requests.get(file_path)
-            img_data = response.content
-        else:
-            with open(file_path, 'rb') as f:
-                img_data = f.read()
-        import base64
+        if not os.path.exists(file_path):
+            raise ValueError(f"File not found: {file_path}")
+            
+        with open(file_path, 'rb') as f:
+            img_data = f.read()
         encoded = base64.b64encode(img_data).decode('utf-8')
         return f"data:image/png;base64,{encoded}"
     
@@ -56,10 +55,11 @@ def virtual_try_on(garm_img_url, human_img_url, garment_des):
 
 # Example usage
 if __name__ == "__main__":
-    # Example inputs
-    garm_img = "https://replicate.delivery/pbxt/KgwTlZyFx5aUU3gc5gMiKuD5nNPTgliMlLUWx160G4z99YjO/sweater.webp"
-    human_img = "https://replicate.delivery/pbxt/KgwTlhCMvDagRrcVzZJbuozNJ8esPqiNAIJS3eMgHrYuHmW4/KakaoTalk_Photo_2024-04-04-21-44-45.png"
-    garment_des = "cute pink top"
+    print("Please use this module by importing it in your application.")
+    print("Example usage:")
+    print("from vton import virtual_try_on")
+    print("output_path = virtual_try_on(garment_image_path, person_image_path, garment_description)")
+    print("\nNote: Both garment_image_path and person_image_path should be paths to local image files.")
     
     # Check if REPLICATE_API_TOKEN is set
     if "REPLICATE_API_TOKEN" not in os.environ: