Update app.py

app.py

@@ -96,14 +96,19 @@ def generate_tryon(cloth_input, user_input, background):
         
         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
+    # Use selected cloth if available, otherwise use uploaded cloth
+    cloth = cloth_selected if cloth_selected is not None else cloth_upload
+    
+    # Use selected user image if available, otherwise use uploaded user image
+    user = user_selected if user_selected is not None else user_upload
+    
+    # Use custom scene if provided, otherwise use selected scene
     background = custom_scene if custom_scene else scene_selection
 
     if cloth is None:
-        return None, "Please select a clothing image from the gallery or upload one."
+        return None, "Please select or upload a clothing image."
     if user is None:
-        return None, "Please select a user image from the gallery or upload one."
+        return None, "Please select or upload a user image."
     if not background:
         return None, "Please select or enter a background scene."
 
@@ -113,6 +118,7 @@ def tryon_interface(cloth_selected, cloth_upload, user_selected, user_upload, sc
     except Exception as e:
         return None, f"Error: {str(e)}"
 
+
 with gr.Blocks() as demo:
     gr.Markdown("# TryItOut.AI")
     
@@ -174,10 +180,12 @@ with gr.Blocks() as demo:
         outputs=[user_upload]
     )
 
+
     generate_button.click(
         tryon_interface,
         inputs=[cloth_selected, cloth_upload, user_selected, user_upload, scene_selection, custom_scene],
         outputs=[output_image, output_text]
     )
 
+
 demo.launch()