Update app.py

app.py

@@ -50,16 +50,22 @@ except Exception as e:
 def generate_image(prompt):
     global pipeline
     if pipeline is None:
+        print("Error: Pipeline is None (model not loaded)")  # Log this specifically
         return "Error: Model not loaded!"
 
     try:
-        image = pipeline(prompt).images[0]  # Try to generate the image
-        print("Image generated successfully!") # Print success message (for debugging)
-        return image # Return the image if successful
+        print("Starting image generation...")  # Log before the image generation
+        image = pipeline(prompt).images[0]
+        print("Image generated successfully!")
+        return image
     except Exception as e:
-        error_message = f"Error during image generation: {e}" # Capture error
-        print(error_message) # Print error message to console
-        return error_message  # Return the error message to Gradio (so it shows up)
+        error_message = f"Error during image generation: {type(e).__name__}: {e}"  # Include exception type
+        print(f"Full Error Details:\n{error_message}")  # Print full details
+        return error_message  # Return error message to Gradio
+    except RuntimeError as re:
+        error_message = f"Runtime Error during image generation: {type(re).__name__}: {re}"  # Include exception type
+        print(f"Full Runtime Error Details:\n{error_message}")  # Print full details
+        return error_message  # Return error message to Gradio
 
 # 5. Gradio interface
 with gr.Blocks() as demo: