First commit

app.py

@@ -0,0 +1,379 @@
+import os
+import multiprocessing
+import subprocess
+import nltk
+import torch
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from moviepy.editor import VideoFileClip
+import moviepy.editor as mpy
+from PIL import Image, ImageDraw, ImageFont
+from mutagen.mp3 import MP3
+from gtts import gTTS
+from pydub import AudioSegment
+import textwrap
+import gradio as gr
+import matplotlib.pyplot as plt
+import gc
+from huggingface_hub import snapshot_download
+from typing import List
+import shutil
+import numpy as np
+import random
+from diffusers import DiffusionPipeline
+
+# Initialize FLUX pipeline outside of the function
+dtype = torch.bfloat16
+device = "cuda" if torch.cuda.is_available() else "cpu"
+flux_pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
+
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 2048
+
+nltk.download('punkt')
+# Ensure proper multiprocessing start method
+multiprocessing.set_start_method("spawn", force=True)
+
+# GPU Fallback Setup
+if os.environ.get("SPACES_ZERO_GPU") is not None:
+    import spaces
+else:
+    class spaces:
+        @staticmethod
+        def GPU(func=None, duration=None):
+            def wrapper(fn):
+                return fn
+            return wrapper if func is None else wrapper(func)
+
+# Download necessary NLTK data
+def setup_nltk():
+    """Ensure required NLTK data is available."""
+    try:
+        nltk.data.find('tokenizers/punkt')
+    except LookupError:
+        nltk.download('punkt')
+
+setup_nltk()
+
+# Constants
+DESCRIPTION = (
+    "Video Story Generator with Audio\n"
+    "PS: Generation of video by using Artificial Intelligence via FLUX, distilbart, and GTTS."
+)
+TITLE = "Video Story Generator with Audio by using FLUX, distilbart, and GTTS."
+
+# Load Tokenizer and Model for Text Summarization
+def load_text_summarization_model():
+    """Load the tokenizer and model for text summarization."""
+    print("Loading text summarization model...")
+    tokenizer = AutoTokenizer.from_pretrained("sshleifer/distilbart-cnn-12-6")
+    model = AutoModelForSeq2SeqLM.from_pretrained("sshleifer/distilbart-cnn-12-6")
+    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+    print(f"Using device: {device}")
+    model.to(device)
+    return tokenizer, model, device
+
+tokenizer, model, device = load_text_summarization_model()
+
+# Log GPU Memory (optional, for debugging)
+def log_gpu_memory():
+    """Log GPU memory usage."""
+    if torch.cuda.is_available():
+        print(subprocess.check_output('nvidia-smi').decode('utf-8'))
+    else:
+        print("CUDA is not available. Cannot log GPU memory.")
+
+# Check GPU Availability
+def check_gpu_availability():
+    """Print GPU availability and device details."""
+    if torch.cuda.is_available():
+        print(f"CUDA devices: {torch.cuda.device_count()}")
+        print(f"Current device: {torch.cuda.current_device()}")
+        print(torch.cuda.get_device_properties(torch.cuda.current_device()))
+    else:
+        print("CUDA is not available. Running on CPU.")
+
+check_gpu_availability()
+
+@spaces.GPU()
+def generate_image_with_flux(
+    text: str,
+    seed: int = 42,
+    width: int = 1024,
+    height: int = 1024,
+    num_inference_steps: int = 4,
+    randomize_seed: bool = True
+):
+    """
+    Generates an image from text using FLUX.
+
+    Args:
+        text: The text prompt to generate the image from.
+        seed: The random seed for image generation. -1 for random.
+        width: Width of the generated image.
+        height: Height of the generated image.
+        num_inference_steps: Number of inference steps.
+        randomize_seed: Whether to randomize the seed.
+
+    Returns:
+        A PIL Image object.
+    """
+    print(f"DEBUG: Generating image with FLUX for text: '{text}'")
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
+    generator = torch.Generator().manual_seed(seed)
+    image = flux_pipe(
+        prompt=text,
+        width=width,
+        height=height,
+        num_inference_steps=num_inference_steps,
+        generator=generator,
+        guidance_scale=0.0
+    ).images[0]
+
+    print("DEBUG: Image generated successfully.")
+    return image
+
+# --------- End of MinDalle Functions ---------
+# Merge audio files
+
+def merge_audio_files(mp3_names: List[str]) -> str:
+    """
+    Merges a list of MP3 files into a single MP3 file.
+
+    Args:
+        mp3_names: List of paths to MP3 files.
+
+    Returns:
+        Path to the merged MP3 file.
+    """
+    combined = AudioSegment.empty()
+    for f_name in mp3_names:
+        audio = AudioSegment.from_mp3(f_name)
+        combined += audio
+    export_path = "result.mp3"
+    combined.export(export_path, format="mp3")
+    print(f"DEBUG: Audio files merged and saved to {export_path}")
+    return export_path
+
+
+
+# Function to generate video from text
+def get_output_video(text, seed, randomize_seed, width, height, num_inference_steps):
+    print("DEBUG: Starting get_output_video function...")
+    
+    # Summarize the input text
+    print("DEBUG: Summarizing text...")
+    inputs = tokenizer(
+        text,
+        max_length=1024,
+        truncation=True,
+        return_tensors="pt"
+    ).to(device)
+    summary_ids = model.generate(inputs["input_ids"])
+    summary = tokenizer.batch_decode(
+        summary_ids,
+        skip_special_tokens=True,
+        clean_up_tokenization_spaces=False
+    )
+    plot = list(summary[0].split('.'))
+    print(f"DEBUG: Summary generated: {plot}")
+
+    image_system ="Generate a realistic picture about this: "
+
+    # Generate images for each sentence in the plot
+    generated_images = []
+    for i, senten in enumerate(plot[:-1]):
+        print(f"DEBUG: Generating image {i+1} of {len(plot)-1}...")
+        image_dir = f"image_{i}"
+        os.makedirs(image_dir, exist_ok=True)
+        image = generate_image_with_flux(
+            text= image_system + senten,
+            seed=seed,
+            randomize_seed=randomize_seed,
+            width=width,  
+            height=height, 
+            num_inference_steps=num_inference_steps
+        )
+        generated_images.append(image)
+        image_path = os.path.join(image_dir, "generated_image.png")
+        image.save(image_path)
+        print(f"DEBUG: Image generated and saved to {image_path}")
+
+        #del min_dalle_model # No need to delete the model here
+        # torch.cuda.empty_cache() # No need to empty cache here
+        # gc.collect() # No need to collect garbage here
+
+    # Create subtitles from the plot
+    sentences = plot[:-1]
+    print("DEBUG: Creating subtitles...")
+    assert len(generated_images) == len(sentences), "Mismatch in number of images and sentences."
+    sub_names = [nltk.tokenize.sent_tokenize(sentence) for sentence in sentences]
+
+    # Add subtitles to images with dynamic adjustments
+    def get_dynamic_wrap_width(font, text, image_width, padding):
+        # Estimate the number of characters per line dynamically
+        avg_char_width = sum(font.getbbox(c)[2] for c in text) / len(text)
+        return max(1, (image_width - padding * 2) // avg_char_width)
+
+    def draw_multiple_line_text(image, text, font, text_color, text_start_height, padding=10):
+        draw = ImageDraw.Draw(image)
+        image_width, _ = image.size
+        y_text = text_start_height
+        lines = textwrap.wrap(text, width=get_dynamic_wrap_width(font, text, image_width, padding))
+        for line in lines:
+            line_width, line_height = font.getbbox(line)[2:]
+            draw.text(((image_width - line_width) / 2, y_text), line, font=font, fill=text_color)
+            y_text += line_height + padding
+
+    def add_text_to_img(text1, image_input):
+        print(f"DEBUG: Adding text to image: '{text1}'")
+        # Scale font size dynamically
+        base_font_size = 30
+        image_width, image_height = image_input.size
+        scaled_font_size = max(10, int(base_font_size * (image_width / 800)))
+        path_font = "/usr/share/fonts/truetype/liberation/LiberationSans-Bold.ttf"
+        if not os.path.exists(path_font):
+            path_font = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"
+        font = ImageFont.truetype(path_font, scaled_font_size)
+
+        text_color = (255, 255, 0)
+        padding = 10
+
+        # Estimate starting height dynamically
+        line_height = font.getbbox("A")[3] + padding
+        total_text_height = len(textwrap.wrap(text1, get_dynamic_wrap_width(font, text1, image_width, padding))) * line_height
+        text_start_height = image_height - total_text_height - 20
+
+        draw_multiple_line_text(image_input, text1, font, text_color, text_start_height, padding)
+        return image_input
+
+
+    # Process images with subtitles
+    generated_images_sub = []
+    for k, image in enumerate(generated_images):
+        text_to_add = sub_names[k][0]
+        result = add_text_to_img(text_to_add, image.copy())
+        generated_images_sub.append(result)
+        result.save(f"image_{k}/generated_image_with_subtitles.png")
+
+
+
+    # Generate audio for each subtitle
+    mp3_names = []
+    mp3_lengths = []
+    for k, text_to_add in enumerate(sub_names):
+        print(f"DEBUG: Generating audio for: '{text_to_add[0]}'")
+        f_name = f'audio_{k}.mp3'
+        mp3_names.append(f_name)
+        myobj = gTTS(text=text_to_add[0], lang='en', slow=False)
+        myobj.save(f_name)
+        audio = MP3(f_name)
+        mp3_lengths.append(audio.info.length)
+        print(f"DEBUG: Audio duration: {audio.info.length} seconds")
+
+    # Merge audio files
+    export_path = merge_audio_files(mp3_names)
+
+    # Create video clips from images
+    clips = []
+    for k, img in enumerate(generated_images_sub):
+        duration = mp3_lengths[k]
+        print(f"DEBUG: Creating video clip {k+1} with duration: {duration} seconds")
+        clip = mpy.ImageClip(f"image_{k}/generated_image_with_subtitles.png").set_duration(duration + 0.5)
+        clips.append(clip)
+
+    # Concatenate video clips
+    print("DEBUG: Concatenating video clips...")
+    concat_clip = mpy.concatenate_videoclips(clips, method="compose")
+    concat_clip.write_videofile("result_no_audio.mp4", fps=24, logger=None)
+
+    # Combine video and audio
+    movie_name = 'result_no_audio.mp4'
+    movie_final = 'result_final.mp4'
+
+    def combine_audio(vidname, audname, outname, fps=24):
+        print(f"DEBUG: Combining audio for video: '{vidname}'")
+        my_clip = mpy.VideoFileClip(vidname)
+        audio_background = mpy.AudioFileClip(audname)
+        final_clip = my_clip.set_audio(audio_background)
+        final_clip.write_videofile(outname, fps=fps, logger=None)
+
+    combine_audio(movie_name, export_path, movie_final)
+
+    # Clean up
+    print("DEBUG: Cleaning up files...")
+    for i in range(len(generated_images_sub)):
+        shutil.rmtree(f"image_{i}")
+        os.remove(f"audio_{i}.mp3")
+    os.remove("result.mp3")
+    os.remove("result_no_audio.mp4")
+
+    print("DEBUG: Cleanup complete.")
+    print("DEBUG: get_output_video function completed successfully.")
+    return 'result_final.mp4'
+
+# Example text (can be changed by user in Gradio interface)
+text = 'Once, there was a girl called Laura who went to the supermarket to buy the ingredients to make a cake. Because today is her birthday and her friends come to her house and help her to prepare the cake.'
+
+# Create Gradio interface
+demo = gr.Blocks()
+with demo:
+    gr.Markdown("# Video Generator from stories with Artificial Intelligence")
+    gr.Markdown("A story can be input by user. The story is summarized using DistilBART model. Then, the images are generated by using FLUX, and the subtitles and audio are created using gTTS. These are combined to generate a video.")
+    with gr.Row():
+        with gr.Column():
+            input_start_text = gr.Textbox(value=text, label="Type your story here, for now a sample story is added already!")
+            with gr.Accordion("Advanced Settings", open=False):
+                seed = gr.Slider(
+                    label="Seed",
+                    minimum=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=42,
+                )
+                
+                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+                
+                with gr.Row():
+                    
+                    width = gr.Slider(
+                        label="Width",
+                        minimum=256,
+                        maximum=MAX_IMAGE_SIZE,
+                        step=32,
+                        value=512,
+                    )
+                    
+                    height = gr.Slider(
+                        label="Height",
+                        minimum=256,
+                        maximum=MAX_IMAGE_SIZE,
+                        step=32,
+                        value=512,
+                    )
+                
+                with gr.Row():
+                    
+    
+                    num_inference_steps = gr.Slider(
+                        label="Number of inference steps",
+                        minimum=1,
+                        maximum=50,
+                        step=1,
+                        value=4,
+                    )
+            with gr.Row():
+                button_gen_video = gr.Button("Generate Video")
+        with gr.Column():
+            #output_interpolation = gr.Video(label="Generated Video")
+            output_interpolation = gr.Video(value="test.mp4", label="Generated Video")  # Set default video
+    gr.Markdown("<h3>Future Works </h3>")
+    gr.Markdown("This program is a text-to-video AI software generating videos from any prompt! AI software to build an art gallery. The future version will use more advanced image generation models. For more info visit [ruslanmv.com](https://ruslanmv.com/) ")
+    button_gen_video.click(
+        fn=get_output_video, 
+        inputs=[input_start_text, seed, randomize_seed, width, height, num_inference_steps], 
+        outputs=output_interpolation
+    )
+
+# Launch the Gradio app
+demo.launch(debug=True, share=False)

requirements.txt

@@ -0,0 +1,590 @@
+absl-py==1.4.0
+accelerate==1.3.0
+aiofiles==23.2.1
+aiohappyeyeballs==2.4.4
+aiohttp==3.11.11
+aiosignal==1.3.2
+alabaster==1.0.0
+albucore==0.0.19
+albumentations==1.4.20
+ale-py==0.10.1
+altair==5.5.0
+annotated-types==0.7.0
+anyio==3.7.1
+argon2-cffi==23.1.0
+argon2-cffi-bindings==21.2.0
+array_record==0.6.0
+arviz==0.20.0
+astropy==6.1.7
+astropy-iers-data==0.2025.1.27.0.32.44
+astunparse==1.6.3
+atpublic==4.1.0
+attrs==25.1.0
+audioread==3.0.1
+autograd==1.7.0
+babel==2.16.0
+backcall==0.2.0
+beautifulsoup4==4.12.3
+bigframes==1.34.0
+bigquery-magics==0.5.0
+bleach==6.2.0
+blinker==1.9.0
+blis==0.7.11
+blosc2==3.0.0
+bokeh==3.6.2
+Bottleneck==1.4.2
+bqplot==0.12.44
+branca==0.8.1
+CacheControl==0.14.2
+cachetools==5.5.1
+catalogue==2.0.10
+certifi==2024.12.14
+cffi==1.17.1
+chardet==5.2.0
+charset-normalizer==3.4.1
+chex==0.1.88
+clarabel==0.9.0
+click==8.1.8
+cloudpathlib==0.20.0
+cloudpickle==3.1.1
+cmake==3.31.4
+cmdstanpy==1.2.5
+colorcet==3.1.0
+colorlover==0.3.0
+colour==0.1.5
+community==1.0.0b1
+confection==0.1.5
+cons==0.4.6
+contourpy==1.3.1
+cramjam==2.9.1
+cryptography==43.0.3
+cuda-python==12.6.0
+cudf-cu12 @ https://pypi.nvidia.com/cudf-cu12/cudf_cu12-24.12.0-cp311-cp311-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl
+cufflinks==0.17.3
+cupy-cuda12x==13.3.0
+cvxopt==1.3.2
+cvxpy==1.6.0
+cycler==0.12.1
+cyipopt==1.5.0
+cymem==2.0.11
+Cython==3.0.11
+dask==2024.10.0
+datascience==0.17.6
+db-dtypes==1.4.0
+dbus-python==1.2.18
+debugpy==1.8.0
+decorator==4.4.2
+defusedxml==0.7.1
+Deprecated==1.2.18
+diffusers @ git+https://github.com/huggingface/diffusers.git@9f28f1abbaf1de21454644ea5391389dabe9a14a
+distro==1.9.0
+dlib==19.24.2
+dm-tree==0.1.8
+docker-pycreds==0.4.0
+docstring_parser==0.16
+docutils==0.21.2
+dopamine_rl==4.1.2
+duckdb==1.1.3
+earthengine-api==1.4.6
+easydict==1.13
+editdistance==0.8.1
+eerepr==0.1.0
+einops==0.8.0
+emoji==2.14.1
+en-core-web-sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.7.1/en_core_web_sm-3.7.1-py3-none-any.whl#sha256=86cc141f63942d4b2c5fcee06630fd6f904788d2f0ab005cce45aadb8fb73889
+entrypoints==0.4
+et_xmlfile==2.0.0
+etils==1.11.0
+etuples==0.3.9
+eval_type_backport==0.2.2
+Farama-Notifications==0.0.4
+fastai==2.7.18
+fastapi==0.115.8
+fastcore==1.7.28
+fastdownload==0.0.7
+fastjsonschema==2.21.1
+fastprogress==1.0.3
+fastrlock==0.8.3
+ffmpy==0.5.0
+filelock==3.17.0
+firebase-admin==6.6.0
+Flask==3.1.0
+flatbuffers==25.1.24
+flax==0.10.2
+folium==0.19.4
+fonttools==4.55.7
+frozendict==2.4.6
+frozenlist==1.5.0
+fsspec==2024.10.0
+future==1.0.0
+gast==0.6.0
+gcsfs==2024.10.0
+GDAL==3.6.4
+gdown==5.2.0
+geemap==0.35.1
+gensim==4.3.3
+geocoder==1.38.1
+geographiclib==2.0
+geopandas==1.0.1
+geopy==2.4.1
+gin-config==0.5.0
+gitdb==4.0.12
+GitPython==3.1.44
+glob2==0.7
+google==2.0.3
+google-ai-generativelanguage==0.6.15
+google-api-core==2.19.2
+google-api-python-client==2.155.0
+google-auth==2.27.0
+google-auth-httplib2==0.2.0
+google-auth-oauthlib==1.2.1
+google-cloud-aiplatform==1.74.0
+google-cloud-bigquery==3.25.0
+google-cloud-bigquery-connection==1.17.0
+google-cloud-bigquery-storage==2.27.0
+google-cloud-bigtable==2.28.1
+google-cloud-core==2.4.1
+google-cloud-datastore==2.20.2
+google-cloud-firestore==2.19.0
+google-cloud-functions==1.19.0
+google-cloud-iam==2.17.0
+google-cloud-language==2.16.0
+google-cloud-pubsub==2.25.0
+google-cloud-resource-manager==1.14.0
+google-cloud-spanner==3.51.0
+google-cloud-storage==2.19.0
+google-cloud-translate==3.19.0
+google-colab @ file:///colabtools/dist/google_colab-1.0.0.tar.gz
+google-crc32c==1.6.0
+google-genai==0.3.0
+google-generativeai==0.8.4
+google-pasta==0.2.0
+google-resumable-media==2.7.2
+googleapis-common-protos==1.66.0
+googledrivedownloader==0.4
+gradio==5.14.0
+gradio_client==1.7.0
+graphviz==0.20.3
+greenlet==3.1.1
+grpc-google-iam-v1==0.14.0
+grpc-interceptor==0.15.4
+grpcio==1.70.0
+grpcio-status==1.62.3
+gspread==6.1.4
+gspread-dataframe==4.0.0
+gTTS==2.5.4
+gym==0.25.2
+gym-notices==0.0.8
+gymnasium==1.0.0
+h11==0.14.0
+h5netcdf==1.5.0
+h5py==3.12.1
+highspy==1.9.0
+holidays==0.65
+holoviews==1.20.0
+html5lib==1.1
+httpcore==1.0.7
+httpimport==1.4.0
+httplib2==0.22.0
+httpx==0.28.1
+huggingface-hub==0.27.1
+humanize==4.11.0
+hyperopt==0.2.7
+ibis-framework==9.2.0
+idna==3.10
+imageio==2.36.1
+imageio-ffmpeg==0.6.0
+imagesize==1.4.1
+imbalanced-learn==0.13.0
+imgaug==0.4.0
+immutabledict==4.2.1
+importlib_metadata==8.6.1
+importlib_resources==6.5.2
+imutils==0.5.4
+inflect==7.5.0
+iniconfig==2.0.0
+intel-cmplr-lib-ur==2025.0.4
+intel-openmp==2025.0.4
+ipyevents==2.0.2
+ipyfilechooser==0.6.0
+ipykernel==5.5.6
+ipyleaflet==0.19.2
+ipyparallel==8.8.0
+ipython==7.34.0
+ipython-genutils==0.2.0
+ipython-sql==0.5.0
+ipytree==0.2.2
+ipywidgets==7.7.1
+itsdangerous==2.2.0
+jax==0.4.33
+jax-cuda12-pjrt==0.4.33
+jax-cuda12-plugin==0.4.33
+jaxlib==0.4.33
+jeepney==0.7.1
+jellyfish==1.1.0
+jieba==0.42.1
+Jinja2==3.1.5
+jiter==0.8.2
+joblib==1.4.2
+jsonpatch==1.33
+jsonpickle==4.0.1
+jsonpointer==3.0.0
+jsonschema==4.23.0
+jsonschema-specifications==2024.10.1
+jupyter-client==6.1.12
+jupyter-console==6.1.0
+jupyter-leaflet==0.19.2
+jupyter-server==1.24.0
+jupyter_core==5.7.2
+jupyterlab_pygments==0.3.0
+jupyterlab_widgets==3.0.13
+kaggle==1.6.17
+kagglehub==0.3.6
+keras==3.8.0
+keras-hub==0.18.1
+keras-nlp==0.18.1
+keyring==23.5.0
+kiwisolver==1.4.8
+langchain==0.3.16
+langchain-core==0.3.32
+langchain-text-splitters==0.3.5
+langcodes==3.5.0
+langsmith==0.3.2
+language_data==1.3.0
+launchpadlib==1.10.16
+lazr.restfulclient==0.14.4
+lazr.uri==1.0.6
+lazy_loader==0.4
+libclang==18.1.1
+libcudf-cu12 @ https://pypi.nvidia.com/libcudf-cu12/libcudf_cu12-24.12.0-py3-none-manylinux_2_28_x86_64.whl
+libkvikio-cu12==24.12.1
+librosa==0.10.2.post1
+lightgbm==4.5.0
+linkify-it-py==2.0.3
+llvmlite==0.43.0
+locket==1.0.0
+logical-unification==0.4.6
+lxml==5.3.0
+marisa-trie==1.2.1
+Markdown==3.7
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.10.0
+matplotlib-inline==0.1.7
+matplotlib-venn==1.1.1
+mdit-py-plugins==0.4.2
+mdurl==0.1.2
+min-dalle==0.4.11
+miniKanren==1.0.3
+missingno==0.5.2
+mistune==3.1.1
+mizani==0.13.1
+mkl==2025.0.1
+ml-dtypes==0.4.1
+mlxtend==0.23.4
+more-itertools==10.5.0
+moviepy==1.0.3
+mpmath==1.3.0
+msgpack==1.1.0
+multidict==6.1.0
+multipledispatch==1.0.0
+multitasking==0.0.11
+murmurhash==1.0.12
+music21==9.3.0
+mutagen==1.47.0
+namex==0.0.8
+narwhals==1.24.1
+natsort==8.4.0
+nbclassic==1.2.0
+nbclient==0.10.2
+nbconvert==7.16.6
+nbformat==5.10.4
+ndindex==1.9.2
+nest-asyncio==1.6.0
+networkx==3.4.2
+nibabel==5.3.2
+nltk==3.9.1
+notebook==6.5.5
+notebook_shim==0.2.4
+numba==0.60.0
+numba-cuda==0.0.17.1
+numexpr==2.10.2
+numpy==1.26.4
+nvidia-cublas-cu11==11.11.3.6
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu11==11.8.87
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvcc-cu12==12.5.82
+nvidia-cuda-nvrtc-cu11==11.8.89
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu11==11.8.89
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu11==9.1.0.70
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu11==10.9.0.58
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu11==10.3.0.86
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu11==11.4.1.48
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu11==11.7.5.86
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-cusparselt-cu12==0.6.2
+nvidia-nccl-cu11==2.21.5
+nvidia-nccl-cu12==2.21.5
+nvidia-nvcomp-cu12==4.1.0.6
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu11==11.8.86
+nvidia-nvtx-cu12==12.4.127
+nvtx==0.2.10
+nx-cugraph-cu12 @ https://pypi.nvidia.com/nx-cugraph-cu12/nx_cugraph_cu12-24.12.0-py3-none-any.whl
+oauth2client==4.1.3
+oauthlib==3.2.2
+openai==1.59.9
+opencv-contrib-python==4.10.0.84
+opencv-python==4.10.0.84
+opencv-python-headless==4.11.0.86
+openpyxl==3.1.5
+opentelemetry-api==1.16.0
+opentelemetry-sdk==1.16.0
+opentelemetry-semantic-conventions==0.37b0
+opt_einsum==3.4.0
+optax==0.2.4
+optree==0.14.0
+orbax-checkpoint==0.6.4
+orjson==3.10.15
+osqp==0.6.7.post3
+packaging==24.2
+pandas==2.2.2
+pandas-datareader==0.10.0
+pandas-gbq==0.26.1
+pandas-stubs==2.2.2.240909
+pandocfilters==1.5.1
+panel==1.6.0
+param==2.2.0
+parso==0.8.4
+parsy==2.1
+partd==1.4.2
+pathlib==1.0.1
+patsy==1.0.1
+peewee==3.17.8
+peft==0.14.0
+pexpect==4.9.0
+pickleshare==0.7.5
+pillow==11.1.0
+platformdirs==4.3.6
+plotly==5.24.1
+plotnine==0.14.5
+pluggy==1.5.0
+ply==3.11
+polars==1.9.0
+pooch==1.8.2
+portpicker==1.5.2
+preshed==3.0.9
+prettytable==3.13.0
+proglog==0.1.10
+progressbar2==4.5.0
+prometheus_client==0.21.1
+promise==2.3
+prompt_toolkit==3.0.50
+propcache==0.2.1
+prophet==1.1.6
+proto-plus==1.26.0
+protobuf==4.25.6
+psutil==5.9.5
+psycopg2==2.9.10
+ptyprocess==0.7.0
+py-cpuinfo==9.0.0
+py4j==0.10.9.7
+pyarrow==17.0.0
+pyasn1==0.6.1
+pyasn1_modules==0.4.1
+pycocotools==2.0.8
+pycparser==2.22
+pydantic==2.10.6
+pydantic_core==2.27.2
+pydata-google-auth==1.9.1
+pydot==3.0.4
+pydotplus==2.0.2
+PyDrive==1.3.1
+PyDrive2==1.21.3
+pydub==0.25.1
+pyerfa==2.0.1.5
+pygame==2.6.1
+pygit2==1.16.0
+Pygments==2.18.0
+PyGObject==3.42.1
+PyJWT==2.10.1
+pylibcudf-cu12 @ https://pypi.nvidia.com/pylibcudf-cu12/pylibcudf_cu12-24.12.0-cp311-cp311-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl
+pylibcugraph-cu12==24.12.0
+pylibraft-cu12==24.12.0
+pymc==5.19.1
+pymystem3==0.2.0
+pynvjitlink-cu12==0.5.0
+pyogrio==0.10.0
+Pyomo==6.8.2
+PyOpenGL==3.1.9
+pyOpenSSL==24.2.1
+pyparsing==3.2.1
+pyperclip==1.9.0
+pyproj==3.7.0
+pyshp==2.3.1
+PySocks==1.7.1
+pyspark==3.5.4
+pytensor==2.26.4
+pytest==8.3.4
+python-apt==0.0.0
+python-box==7.3.2
+python-dateutil==2.8.2
+python-louvain==0.16
+python-multipart==0.0.20
+python-slugify==8.0.4
+python-snappy==0.7.3
+python-utils==3.9.1
+pytz==2024.2
+pyviz_comms==3.0.4
+PyYAML==6.0.2
+pyzmq==24.0.1
+qdldl==0.1.7.post5
+ratelim==0.1.6
+referencing==0.36.2
+regex==2024.11.6
+requests==2.32.3
+requests-oauthlib==1.3.1
+requests-toolbelt==1.0.0
+requirements-parser==0.9.0
+rich==13.9.4
+rmm-cu12==24.12.1
+rpds-py==0.22.3
+rpy2==3.4.2
+rsa==4.9
+ruff==0.9.4
+safehttpx==0.1.6
+safetensors==0.5.2
+scikit-image==0.25.1
+scikit-learn==1.6.1
+scipy==1.13.1
+scooby==0.10.0
+scs==3.2.7.post2
+seaborn==0.13.2
+SecretStorage==3.3.1
+semantic-version==2.10.0
+Send2Trash==1.8.3
+sentence-transformers==3.3.1
+sentencepiece==0.2.0
+sentry-sdk==2.20.0
+setproctitle==1.3.4
+shap==0.46.0
+shapely==2.0.6
+shellingham==1.5.4
+simple-parsing==0.1.7
+six==1.17.0
+sklearn-compat==0.1.3
+sklearn-pandas==2.2.0
+slicer==0.0.8
+smart-open==7.1.0
+smmap==5.0.2
+sniffio==1.3.1
+snowballstemmer==2.2.0
+soundfile==0.13.1
+soupsieve==2.6
+soxr==0.5.0.post1
+spaces==0.32.0
+spacy==3.7.5
+spacy-legacy==3.0.12
+spacy-loggers==1.0.5
+spanner-graph-notebook==1.0.9
+Sphinx==8.1.3
+sphinxcontrib-applehelp==2.0.0
+sphinxcontrib-devhelp==2.0.0
+sphinxcontrib-htmlhelp==2.1.0
+sphinxcontrib-jsmath==1.0.1
+sphinxcontrib-qthelp==2.0.0
+sphinxcontrib-serializinghtml==2.0.0
+SQLAlchemy==2.0.37
+sqlglot==25.6.1
+sqlparse==0.5.3
+srsly==2.5.1
+stanio==0.5.1
+starlette==0.45.3
+statsmodels==0.14.4
+stringzilla==3.11.3
+sympy==1.13.1
+tables==3.10.2
+tabulate==0.9.0
+tbb==2022.0.0
+tcmlib==1.2.0
+tenacity==9.0.0
+tensorboard==2.18.0
+tensorboard-data-server==0.7.2
+tensorflow==2.18.0
+tensorflow-datasets==4.9.7
+tensorflow-hub==0.16.1
+tensorflow-io-gcs-filesystem==0.37.1
+tensorflow-metadata==1.16.1
+tensorflow-probability==0.24.0
+tensorflow-text==2.18.1
+tensorstore==0.1.71
+termcolor==2.5.0
+terminado==0.18.1
+text-unidecode==1.3
+textblob==0.17.1
+tf-slim==1.1.0
+tf_keras==2.18.0
+thinc==8.2.5
+threadpoolctl==3.5.0
+tifffile==2025.1.10
+timm==1.0.14
+tinycss2==1.4.0
+tokenizers==0.21.0
+toml==0.10.2
+tomlkit==0.13.2
+toolz==0.12.1
+torch==2.6.0+cu118
+torchaudio==2.6.0+cu118
+torchsummary==1.5.1
+torchvision==0.21.0+cu118
+tornado==6.4.2
+tqdm==4.67.1
+traitlets==5.7.1
+traittypes==0.2.1
+transformers==4.48.2
+triton==3.2.0
+tweepy==4.14.0
+typeguard==4.4.1
+typer==0.15.1
+types-pytz==2024.2.0.20241221
+types-setuptools==75.8.0.20250110
+typing_extensions==4.12.2
+tzdata==2025.1
+tzlocal==5.2
+uc-micro-py==1.0.3
+umf==0.9.1
+uritemplate==4.1.1
+urllib3==2.3.0
+uvicorn==0.34.0
+vega-datasets==0.9.0
+wadllib==1.3.6
+wandb==0.19.5
+wasabi==1.1.3
+wcwidth==0.2.13
+weasel==0.4.1
+webcolors==24.11.1
+webencodings==0.5.1
+websocket-client==1.8.0
+websockets==14.2
+Werkzeug==3.1.3
+widgetsnbextension==3.6.10
+wordcloud==1.9.4
+wrapt==1.17.2
+xarray==2025.1.1
+xarray-einstats==0.8.0
+xformers==0.0.29.post2
+xgboost==2.1.3
+xlrd==2.0.1
+xyzservices==2025.1.0
+yarl==1.18.3
+yellowbrick==1.5
+yfinance==0.2.52
+zipp==3.21.0
+zstandard==0.23.0