Update app.py

app.py

@@ -2,43 +2,120 @@ import os
 import sys
 import subprocess
 
-# Ensure we're in the right directory and install diffusers
-try:
-    os.chdir('diffusers_repo')
-    subprocess.check_call([sys.executable, '-m', 'pip', 'install', '-e', '.'])
-    print("Successfully installed diffusers from local repository")
-except Exception as e:
-    print(f"Installation error: {e}")
-    # Fallback to standard diffusers installation
-    subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'diffusers'])
+# Debug: Print environment information
+def print_env_info():
+    print("=== ENVIRONMENT DEBUG INFO ===")
+    print(f"Python executable: {sys.executable}")
+    print(f"Python version: {sys.version}")
+    print(f"Current working directory: {os.getcwd()}")
+    print(f"Python path: {sys.path[:3]}...")  # First 3 entries
+    
+    # Check if we can find pip
+    try:
+        pip_result = subprocess.run([sys.executable, '-m', 'pip', '--version'], 
+                                  capture_output=True, text=True)
+        print(f"Pip version: {pip_result.stdout.strip()}")
+    except Exception as e:
+        print(f"Pip check failed: {e}")
+    
+    # List contents of current directory
+    print(f"Directory contents: {os.listdir('.')}")
+    if os.path.exists('diffusers_repo'):
+        print("✓ diffusers_repo found")
+    else:
+        print("✗ diffusers_repo NOT found")
+    print("===============================")
 
-# Add the current directory to Python path if needed
-if os.getcwd() not in sys.path:
-    sys.path.insert(0, os.getcwd())
+print_env_info()
 
+# Install diffusers from local repository
+def install_diffusers():
+    try:
+        # Change to diffusers repo directory
+        original_dir = os.getcwd()
+        
+        if not os.path.exists('diffusers_repo'):
+            print("ERROR: diffusers_repo directory not found!")
+            return False
+            
+        os.chdir('diffusers_repo')
+        print(f"Changed to directory: {os.getcwd()}")
+        
+        # Use the exact same Python executable for pip
+        python_exe = sys.executable
+        print(f"Using Python executable: {python_exe}")
+        
+        # Install in editable mode with verbose output
+        cmd = [python_exe, '-m', 'pip', 'install', '-e', '.', '--verbose']
+        print(f"Running command: {' '.join(cmd)}")
+        
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        print("Installation stdout:", result.stdout[-1000:])  # Last 1000 chars
+        if result.stderr:
+            print("Installation stderr:", result.stderr[-1000:])  # Last 1000 chars
+        
+        # Change back to original directory
+        os.chdir(original_dir)
+        
+        # Add diffusers_repo to Python path
+        diffusers_path = os.path.join(original_dir, 'diffusers_repo')
+        if diffusers_path not in sys.path:
+            sys.path.insert(0, diffusers_path)
+            print(f"Added to Python path: {diffusers_path}")
+        
+        # Test if we can import after installation
+        try:
+            import diffusers
+            print(f"✓ Successfully imported diffusers from: {diffusers.__file__}")
+            return True
+        except ImportError as e:
+            print(f"✗ Failed to import diffusers after installation: {e}")
+            return False
+            
+    except Exception as e:
+        print(f"Installation failed with exception: {e}")
+        return False
+
+# Attempt installation
+print("Starting diffusers installation...")
+if not install_diffusers():
+    print("Failed to install local diffusers, exiting...")
+    sys.exit(1)
+else:
+    print("✓ Local diffusers installation successful!")
+
+# Now import required modules
+print("Importing required modules...")
 try:
     import gradio as gr
-    import spaces
+    print("✓ Gradio imported")
+    import spaces  
+    print("✓ Spaces imported")
     import torch
+    print("✓ Torch imported")
     from diffusers import CosmosTextToImagePipeline
+    print("✓ CosmosTextToImagePipeline imported")
     import random
-    print("All imports successful")
+    print("✓ Random imported")
+    print("All imports successful!")
 except ImportError as e:
-    print(f"Import error: {e}")
-    print("Please ensure all required packages are installed")
+    print(f"✗ Import failed: {e}")
+    print("Checking installed packages...")
+    try:
+        result = subprocess.run([sys.executable, '-m', 'pip', 'list'], 
+                              capture_output=True, text=True)
+        print("Installed packages:")
+        print(result.stdout)
+    except:
+        pass
     sys.exit(1)
 
 # Available checkpoints: nvidia/Cosmos-Predict2-2B-Text2Image, nvidia/Cosmos-Predict2-14B-Text2Image
 model_id = "diffusers-internal-dev/ct2i-14B"
 
 # Load the pipeline once to avoid repeated loading
-try:
-    pipe = CosmosTextToImagePipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
-    pipe.to("cuda")
-    print("Pipeline loaded successfully")
-except Exception as e:
-    print(f"Error loading pipeline: {e}")
-    sys.exit(1)
+pipe = CosmosTextToImagePipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
+pipe.to("cuda")
 
 @spaces.GPU
 def generate_image(prompt, negative_prompt, seed, randomize_seed):