📽️ AI Video Story Generator

This project utilizes Ollama LLM to create video text ideas based on user prompts. It takes a prompt from the user and generates a story, which is then converted into audio using AI. Subsequently, captions are generated from the audio, and finally, these captions are merged into a video.

🚀 How It Works

1. Prompt Input: Users provide a prompt to the system.
2. Text to Text: AI generates a story based on the provided prompt.
3. Text to Speech: The story is converted into audio using AI text-to-speech technology.
4. Caption Generation: AI generates captions from the audio.
5. Video Creation: Captions are merged into a video.

🛠️ How to Run

Prerequisites

• Docker installed on your system
• At least ~5.5GB of RAM to run all applications and llm models

Installation

1. Clone the repository:

   git clone https://github.com/ccallazans/ai-video-generator.git
   cd ai-video-generator

2. Start the Docker containers:

   make start

Once the containers are up and running, you can access the AI video story generator through its provided interface or API endpoints. The first time may be slow because it has to download the Ollama model and Python Torch requirements, but subsequent runs will be faster because Docker keeps it in cache.

Usage

Send a POST request with the content you want to be generated

curl --location 'http://localhost:8080/api/v1/generate' \
--header 'Content-Type: application/json' \
--data '{
    "message": "Tell me a story about Bahia"
}'

The generated videos are saved on the "generated" folder.

🖥️ To Run Locally

1. Install Ollama: Instructions on: https://ollama.com/download

2. Create a Python 3 environment and install its dependencies:

   python3 -m venv venv
   source venv/bin/activate
   pip install -r requirements.txt

3. Run the Golang application:

   go run cmd/*

📂 Additional Notes

• To add more background videos, put it inside resources/videos folder. The video is random selected, so change on code if want.
• Ensure that Docker is properly configured and running on your system.
• Adjust any necessary configurations in the Dockerfile or docker-compose.yml file according to your requirements.
• For production use, ensure proper security measures and scalability considerations are implemented.
• Fell free to change the ollama model. Choose one from https://ollama.com/library. Im using orca-mini on this project

Feel free to explore and modify the project according to your needs! If you encounter any issues or have suggestions for improvement, please don't hesitate to open an issue or submit a pull request.