add error display

app.py

@@ -49,6 +49,7 @@ def url_to_base64(url):
         print(f"Error converting URL to base64: {str(e)}")
         return None
 
+
 def run_viton(model_image_path: str = None, 
               garment_image_path: str = None, 
               model_url: str = None, 
@@ -59,53 +60,43 @@ def run_viton(model_image_path: str = None,
             ):
     """
     Run the Virtual Try-On model with provided images path or URLs.
-
-    Args:
-        model_image_path (str): Path to the model image file.
-        garment_image_path (str): Path to the garment image file.
-        model_url (str): URL of the model image.
-        garment_url (str): URL of the garment image.
-        n_steps (int): Number of steps for the model.
-        image_scale (float): Scale for the generated images.
-        seed (int): Random seed for reproducibility.
-
-    Returns:
-        list: List of generated images in base64 format.
     """
     if not model_image_path and not model_url:
-        print("Error: No model image provided")
-        return []
+        raise gr.Error("❌ Please provide either a model image file or URL")
     if not garment_image_path and not garment_url:
-        print("Error: No garment image provided")
-        return []
+        raise gr.Error("❌ Please provide either a garment image file or URL")
+    
     try:
         api_url = os.environ.get("SERVER_URL")
-        print(f"Using API URL: {api_url}")  # Add this to debug
+        if not api_url:
+            raise gr.Error("❌ SERVER_URL not configured in environment variables")
         
-        # Determine which inputs to use (file upload or URL)
-        model_b64 = None
-        garment_b64 = None
+        print(f"Using API URL: {api_url}")
         
         # Handle model image
+        model_b64 = None
         if model_url and model_url.strip():
             print(f"Using model URL: {model_url}")
             model_b64 = url_to_base64(model_url.strip())
+            if not model_b64:
+                raise gr.Error("❌ Failed to load model image from URL. Please check the URL is valid.")
         elif model_image_path:
             print(f"Using model file: {model_image_path}")
             model_b64 = image_to_base64(model_image_path)
         
         # Handle garment image
+        garment_b64 = None
         if garment_url and garment_url.strip():
             print(f"Using garment URL: {garment_url}")
             garment_b64 = url_to_base64(garment_url.strip())
+            if not garment_b64:
+                raise gr.Error("❌ Failed to load garment image from URL. Please check the URL is valid.")
         elif garment_image_path:
             print(f"Using garment file: {garment_image_path}")
             garment_b64 = image_to_base64(garment_image_path)
         
-        # Check if we have both images
         if not model_b64 or not garment_b64:
-            print("Error: Missing model or garment image")
-            return []
+            raise gr.Error("❌ Failed to process images. Please try again.")
 
         # Prepare request
         request_data = {
@@ -128,25 +119,28 @@ def run_viton(model_image_path: str = None,
         if response.status_code == 200:
             result = response.json()
             if result.get("error"):
-                print(f"Error: {result['error']}")
-                return []
+                raise gr.Error(f"❌ Server error: {result['error']}")
             
             generated_images = []
             for i, img_b64 in enumerate(result.get("images_base64", [])):
                 output_path = f"ootd_output_{i}.png"
-                img = base64_to_image(img_b64, output_path)  # Remove 'self.'
+                img = base64_to_image(img_b64, output_path)
                 generated_images.append(img)
             
+            if not generated_images:
+                raise gr.Error("❌ No images were generated. Please try again.")
+            
             print(f"Successfully generated {len(generated_images)} images")
             return generated_images
         else:
-            print(f"Request failed with status code: {response.status_code}")
-            return []  
+            raise gr.Error(f"❌ Request failed with status code: {response.status_code}")
             
+    except gr.Error:
+        raise  # Re-raise Gradio errors
     except Exception as e:
-        print(f"Exception occurred: {str(e)}")  # Add this
-        return []  
-    
+        print(f"Exception occurred: {str(e)}")
+        raise gr.Error(f"❌ An unexpected error occurred: {str(e)}")
+
 def run_new_garment(model_image_path: str = None, 
               garment_prompt: str = None, 
               model_url: str = None, 
@@ -155,45 +149,33 @@ def run_new_garment(model_image_path: str = None,
               seed=-1
             ):
     """
-    Run the Virtual Try-On model with provided model image and garment image generated using FLUX.1-dev
-    based on description of the garment obtained from the user
-
-    Args:
-        model_image_path (str): Path to the model image file.
-        garment_prompt (str): Description of the garment.
-        model_url (str): URL of the model image.
-        n_steps (int): Number of steps for the model.
-        image_scale (float): Scale for the generated images.
-        seed (int): Random seed for reproducibility.
-
-    Returns:
-        list: List of generated images in base64 format.
+    Run the Virtual Try-On model with provided model image and garment prompt.
     """
     if not model_image_path and not model_url:
-        print("Error: No model image provided")
-        return []
+        raise gr.Error("❌ Please provide either a model image file or URL")
     if not garment_prompt or not garment_prompt.strip():
-        print("Error: No garment description provided")
-        return []
+        raise gr.Error("❌ Please provide a garment description")
+    
     try:
         api_url = os.environ.get("SERVER_URL")
-        print(f"Using API URL: {api_url}")  # Add this to debug
+        if not api_url:
+            raise gr.Error("❌ SERVER_URL not configured in environment variables")
         
-        # Determine which inputs to use (file upload or URL)
-        model_b64 = None
+        print(f"Using API URL: {api_url}")
         
         # Handle model image
+        model_b64 = None
         if model_url and model_url.strip():
             print(f"Using model URL: {model_url}")
             model_b64 = url_to_base64(model_url.strip())
+            if not model_b64:
+                raise gr.Error("❌ Failed to load model image from URL. Please check the URL is valid.")
         elif model_image_path:
             print(f"Using model file: {model_image_path}")
             model_b64 = image_to_base64(model_image_path)
         
-        # Check if we have both images
-        if not model_b64 or not garment_prompt:
-            print("Error: Missing model or garment description")
-            return []
+        if not model_b64:
+            raise gr.Error("❌ Failed to process model image. Please try again.")
 
         # Prepare request
         request_data = {
@@ -216,24 +198,27 @@ def run_new_garment(model_image_path: str = None,
         if response.status_code == 200:
             result = response.json()
             if result.get("error"):
-                print(f"Error: {result['error']}")
-                return []
+                raise gr.Error(f"❌ Server error: {result['error']}")
             
             generated_images = []
             for i, img_b64 in enumerate(result.get("images_base64", [])):
                 output_path = f"flux_output_{i}.png"
-                img = base64_to_image(img_b64, output_path)  # Remove 'self.'
+                img = base64_to_image(img_b64, output_path)
                 generated_images.append(img)
             
+            if not generated_images:
+                raise gr.Error("❌ No images were generated. Please try again.")
+            
             print(f"Successfully generated {len(generated_images)} images")
             return generated_images
         else:
-            print(f"Request failed with status code: {response.status_code}")
-            return []  # Fix: was missing 'return'
+            raise gr.Error(f"❌ Request failed with status code: {response.status_code}")
             
+    except gr.Error:
+        raise  # Re-raise Gradio errors
     except Exception as e:
-        print(f"Exception occurred: {str(e)}")  # Add this
-        return []  # Fix: should return list, not dict for gallery
+        print(f"Exception occurred: {str(e)}")
+        raise gr.Error(f"❌ An unexpected error occurred: {str(e)}")
 
 block = gr.Blocks().queue()
 with block: