Update processing/ai_background.py

processing/ai_background.py

@@ -258,10 +258,15 @@ def generate_ai_background(prompt: str, width: int = 1024, height: int = 768) ->
 
 # Example usage and testing
 if __name__ == "__main__":
-    # Test the background generator
+    # Quick test first
+    if not test_background_generation():
+        print("Exiting due to test failure")
+        sys.exit(1)
+    
+    # Full test suite
     test_prompts = [
         "professional office environment",
-        "peaceful ocean sunset",
+        "peaceful ocean sunset", 
         "dark forest at night",
         "modern corporate meeting room",
         "abstract blue technology background"
@@ -287,11 +292,13 @@ def generate_ai_background(prompt: str, width: int = 1024, height: int = 768) ->
         except Exception as e:
             print(f"✗ Error: {e}")
     
-    print(f"\nModule status:")
+    print(f"\nFinal module status:")
     print(f"- PIL Available: {PIL_AVAILABLE}")
     print(f"- PyTorch Available: {TORCH_AVAILABLE}")
     print(f"- Diffusers Available: {DIFFUSERS_AVAILABLE}")
     if generator.pipeline:
         print(f"- Device: {generator.device}")
+        print("- Mode: AI Generation (Stable Diffusion)")
     else:
-        print("- AI Pipeline: Not initialized (using gradients only)")
+        print("- AI Pipeline: Not initialized")
+        print("- Mode: Gradient Fallback Only")