Update app.py

app.py

@@ -46,8 +46,55 @@ def get_base64_from_image(image):
     return base64.b64encode(buffered.getvalue()).decode('utf-8')
 
 def generate_tryon(cloth_input, user_input, background):
-    # ... (keep the existing generate_tryon function)
+    if isinstance(cloth_input, str):  # URL selected
+        cloth_base64 = get_base64_from_url(cloth_input)
+    else:  # Image uploaded
+        cloth_base64 = get_base64_from_image(cloth_input)
+    
+    if isinstance(user_input, str):  # URL selected
+        user_base64 = get_base64_from_url(user_input)
+    else:  # Image uploaded
+        user_base64 = get_base64_from_image(user_input)
+
+    input_data = {
+        "user_image": user_base64,
+        "product_image": cloth_base64,
+        "background": background
+    }
+
+    response = requests.post(
+        f"{API_URL}/run",
+        headers={
+            "Authorization": f"Bearer {API_KEY}",
+            "Content-Type": "application/json"
+        },
+        data=json.dumps({"input": input_data})
+    )
 
+    if not response.ok:
+        error_text = response.text
+        raise Exception(f"Failed to upload image: {response.status_code} {response.reason} - {error_text}")
+
+    job_id = response.json()['id']
+    
+    while True:
+        status_response = requests.get(
+            f"{API_URL}/status/{job_id}",
+            headers={"Authorization": f"Bearer {API_KEY}"}
+        )
+        
+        if not status_response.ok:
+            raise Exception(f"Status check failed: {status_response.status_code} {status_response.reason} - {status_response.text}")
+        
+        status_data = status_response.json()
+        if status_data['status'] == 'COMPLETED':
+            output_base64 = status_data['output']['output']
+            output_image = Image.open(io.BytesIO(base64.b64decode(output_base64)))
+            return output_image
+        elif status_data['status'] == 'FAILED':
+            raise Exception(f"Job processing failed: {status_data}")
+        
+        time.sleep(2)
 def tryon_interface(cloth_selected, cloth_upload, user_selected, user_upload, scene_selection, custom_scene):
     cloth = cloth_selected if cloth_selected else cloth_upload
     user = user_selected if user_selected else user_upload