Update app.py

app.py

@@ -3,6 +3,7 @@ from gradio_client import Client, handle_file
 import re
 import time
 import os
+import traceback
 from dotenv import load_dotenv
 
 # Load environment variables
@@ -17,7 +18,6 @@ client = Client(
     hf_token=hf_token
 )
 
-
 def generate_outfit(model_image, garment_image, n_samples=1, n_steps=20, image_scale=2, seed=-1):
     if model_image is None or garment_image is None:
         return None, "Please upload both model and garment images"
@@ -59,6 +59,8 @@ def generate_outfit(model_image, garment_image, n_samples=1, n_steps=20, image_s
                     time.sleep(wait_seconds)  # Wait before retrying
                 return None, f"GPU quota exceeded. Please wait {wait_time} before trying again."
             else:
+                # Log the full traceback for debugging
+                traceback.print_exc()
                 return None, f"Error: {str(e)}"
 
 # Create Gradio interface
@@ -70,7 +72,6 @@ with gr.Blocks() as demo:
     - Followed by selecting or uploading the garment you want to try-on.
     - Next click Try Outfit button and wait
 
-  
     """)
 
     with gr.Row():
@@ -79,7 +80,6 @@ with gr.Blocks() as demo:
                 label="Upload Model Image (person wearing clothes)", 
                 type="filepath",
                 height=300
-                
             )
             model_examples = [
                 "https://levihsu-ootdiffusion.hf.space/file=/tmp/gradio/ba5ba7978e7302e8ab5eb733cc7221394c4e6faf/model_5.png",
@@ -104,7 +104,6 @@ with gr.Blocks() as demo:
             ]
             gr.Examples(examples=garment_examples, inputs=garment_image)
 
-        
         with gr.Column():
             output_image = gr.Image(label="Generated Output")
             error_text = gr.Markdown()  # Add error display