IBSEN is an original collaborative framework in the field of drama playing and script generation. Powered by LLM-based generative agents and the surveillance of the director agent, actor agents in the drama play could act towards the desired plot objectives and present their personality at the same time. This framework also allows human players to involve in the play and interact with the actors, while keeping the plot progressing towards the objective. Our paper is accepted by ACL 2024 Main.
This repository provides a simple terminal frontend to interact with IBSEN.
NOTE: We recommend creating a virtual environment to install dependencies because some of the package versions we use are relatively old and their behavior may be incompatible with newer versions of the packages.
pip install -r requirements.txt
By default we use the gpt-3.5-turbo-1106 as the backbone LLM. You should provide your OpenAI API key in api_key.py first, then directly launch the terminal_frontend.py:
python terminal_frontend.py
The data directory contains all the script settings required by the IBSEN. You may customize these files to use IBSEN in your own drama scenarios.
data/scriptincludes the script settings of the whole stage. You can follow the format of our script to create your own stage.data/profileincludes the character profiles used by the actor agents. The profile file names should be the same as the character names defined in the script settings. You can follow the format of our profile to create a new character profile.- You can set the event monologue with an empty string; in this case, the actor agent will generate a monologue for the event.
- If you have dialogue corpora for certain characters, you can place them in
data/corpusto provide references for the actor agents. The corpus file names should be the same as the character names defined in the script settings. You can follow the format of our corpus to build the corpus.
Coming soon