Update app.py

app.py

@@ -1,114 +1,14 @@
 import os
-import sys
 import subprocess
 
-# 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("===============================")
-
-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', '.', '--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!")
+os.chdir('diffusers_repo') 
+subprocess.check_call(['pip', 'install', '.'])
 
-# Now import required modules
-print("Importing required modules...")
-try:
-    import gradio as gr
-    print("✓ Gradio imported")
-    import spaces  
-    print("✓ Spaces imported")
-    import torch
-    print("✓ Torch imported")
-    from diffusers import CosmosTextToImagePipeline
-    print("✓ CosmosTextToImagePipeline imported")
-    import random
-    print("✓ Random imported")
-    print("All imports successful!")
-except ImportError as e:
-    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)
+import gradio as gr
+import spaces
+import torch
+from diffusers import CosmosTextToImagePipeline
+import random
 
 # Available checkpoints: nvidia/Cosmos-Predict2-2B-Text2Image, nvidia/Cosmos-Predict2-14B-Text2Image
 model_id = "diffusers-internal-dev/ct2i-14B"
@@ -162,7 +62,7 @@ with gr.Blocks() as demo:
             with gr.Row():
                 randomize_seed_checkbox = gr.Checkbox(
                     label="Randomize Seed",
-                    value=True  # Fixed: was "TRue"
+                    value=TRue
                 )
                 seed_input = gr.Slider(
                     minimum=0,