Update requirements.txt

requirements.txt

@@ -1,29 +1,68 @@
-# Core ML frameworks
-torch>=2.0.0
-torchvision
-transformers>=4.35.0
-diffusers>=0.24.0
-accelerate>=0.24.0
-safetensors
-
-# Image and video processing  
-Pillow>=9.0.0
-opencv-python
-numpy
-imageio
-imageio-ffmpeg
+# Core ML frameworks for professional AI generation
+torch>=2.4.0
+torchvision>=0.19.0
+torchaudio>=2.4.0
+transformers>=4.45.0
+diffusers>=0.30.0
+accelerate>=0.34.0
+safetensors>=0.4.0
+
+# FLUX and LoRA support
+xformers>=0.0.27
+flash-attn>=2.6.0
+peft>=0.7.0
+bitsandbytes>=0.41.0
+
+# Professional image and video processing  
+Pillow>=10.4.0
+opencv-python>=4.9.0
+numpy>=1.26.0
+matplotlib>=3.8.0
+scikit-image>=0.22.0
+moviepy>=1.0.3
+imageio>=2.31.0
+imageio-ffmpeg>=0.4.9
+ffmpeg-python>=0.2.0
 
 # Audio processing (open source)
-TTS>=0.20.0
-soundfile
-librosa
-
-# Gradio for interface
-gradio>=4.0.0
-spaces
-
-# Utilities
-requests
-tqdm
-huggingface-hub
-xformers  # For memory optimization 
+soundfile>=0.12.1
+librosa>=0.10.1
+pydub>=0.25.1
+
+# Professional Gradio interface
+gradio>=4.40.0
+spaces>=0.28.0
+
+# Enhanced utilities
+requests>=2.31.0
+tqdm>=4.66.0
+huggingface-hub>=0.25.0
+aiohttp>=3.9.0
+websockets>=12.0
+psutil>=5.9.0
+rich>=13.7.0
+
+# Advanced text processing
+nltk>=3.8.0
+textblob>=0.17.0
+regex>=2023.10.0
+
+# Performance optimization
+einops>=0.7.0
+omegaconf>=2.3.0
+timm>=0.9.0
+triton>=2.1.0
+
+# API support
+python-multipart>=0.0.6
+uvicorn>=0.24.0
+fastapi>=0.104.0
+pydantic>=2.4.0
+typing-extensions>=4.8.0
+
+# Async operations
+aiofiles>=23.2.0
+
+# Memory and GPU monitoring
+py3nvml>=0.2.7
+GPUtil>=1.4.0