Please Note: This is a fork of d2clin's OG project for GPT Lab on Streamlit
GPT Lab is a user-friendly Streamlit app that lets users interact with and create their own AI Assistants powered by OpenAI's GPT language model. With GPT Lab, users can chat with pre-built AI Assistants or create their own by specifying a prompt and OpenAI model parameters. Our goal is to make AI accessible and easy to use for everyone, so users can focus on designing their Assistants without worrying about the underlying infrastructure.
GPT Lab is also featured in the Streamlit App Gallery among other impressive Streamlit apps.
For more insight into the development process and lessons learned while building GPT Lab, check out this blog post on the official Streamlit blog.
This README will cover:
- Code structure
- Data models
- Accessing the app
- Running the app Locally
- Contributions
- License
+----------------+ +-------------------+ +-------------------+ +------------+
| | | | | | | |
| Streamlit App |<--->| util_collections |<--->| api_util_firebase |<--->| Firestore |
| | | (users, sessions, | | | | |
| | | bots) | | | | |
+----------------+ +-------------------+ +-------------------+ +------------+
|
|
v
+-----------------+ +------------+
| | | |
| api_util_openai |<--->| OpenAI |
| | | |
+-----------------+ +------------+
Users Collection
|
| - id: (Firestore auto-ID)
| - user_hash: string (one-way hash value of OpenAI API key)
| - created_date: datetime
| - last_modified_date: datetime
| - sessions_started: number
| - sessions_ended: number
| - bots_created: number
User_hash Collection
|
| - id = one-way hash value of OpenAI API key
| - user_hash_type: string (open_ai_key)
| - created_date: datetime
Bots Collection
|
| - id: (Firestore auto-ID)
| - name: string
| - tag_line: string
| - description: string
| - session_type: number
| - creator_user_id: string
| - created_date: datetime
| - last_modified_date: datetime
| - active_initial_prompt_id: string
| - active_model_config_id: string
| - active_summary_prompt_id: string
| - showcased: boolean
| - is_active: boolean
|
v
|--> Model_configs subcollection
| |
| | - config: map
| | | - model: string
| | | - max_tokens: number
| | | - temperature: number
| | | - top_p: number
| | | - frequency_penalty: number
| | | - presence_penalty: number
| | - created_date: datetime
| | - is_active: boolean
|
v
|--> Prompts subcollection
|
| - message_type: string
| - message: string
| - created_date: datetime
| - is_active: boolean
| - sessions_started: number
| - sessions_ended: number
Sessions Collection
|
| - id: (Firestore auto-ID)
| - user_id: string
| - bot_id: string
| - bot_initial_prompt_msg: string
|
| - bot_model_config: map
| | - model: string
| | - max_tokens: number
| | - temperature: number
| | - top_p: number
| | - frequency_penalty: number
| | - presence_penalty: number
|
| - bot_session_type: number
| - bot_summary_prompt_msg: string
| - created_date: datetime
| - session_schema_version: number
| - status: number
| - message_count: number
| - messages_str: string (encrypted)
|
v
|--> Messages subcollection
|
| - created_date: datetime
| - message: string (encrypted)
| - role: string
You can access the app on the Streamlit Cloud community at gptlab.streamlit.app.
To use the app, you will need an OpenAI API key. Don't have one yet? Create one on the OpenAI website. Once you have your API key, enter it into the app when prompted.
For optimal chatting experience, we recommend using a pay-as-you-go API key. (Free trial API keys are limited to 3 requests a minute, not enough to chat with assistants.) You will need to enter your billing information here. You can learn more about OpenAI API rate limits here.
To run the app locally, you will need to:
- Set up your own Google Firestore database.
- GPT Lab uses four main collections:
users
,user_hash
,bots
, andsessions
. - You will need to manually set up a
users
collection before you can run the app locally. (All other collections will be set up by the app).
- GPT Lab uses four main collections:
- Clone this repository
- Get your FireBase JSON secret from https://console.firebase.google.com/u/0/project//settings/serviceaccounts/adminsdk and "Generate new private key"
- Run key-to-toml.py to convert the JSON secret to a TOML file.
- Confirm a .streamlit/secrets.toml file containing the following:
[firestore]
db-key = "YOUR GOOGLE SERVICE ACCOUNT TOML"
exists
- In your terminal, set up your local environment:
- Set up a Python virtual environment (using
venv
,conda
,virtualenv
, or any other tool you prefer) - Install the required Python dependencies (
pip install -r app/requirements.txt
) - Run
streamlit run app/home.py
- Set up a Python virtual environment (using
Contributions are welcomed. Simply open up an issue and create a pull request. If you are introducing new features, please provide a detailed description of the specific use case you are addressing and set up instructions to test.
Aside: I am new to open source, work full-time, and have young kids, please bear with me if I don't get back to you right away.
This project is licensed under the terms of the MIT License. See LICENSE for more details.