Add first commit

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2021 indonesian-nlp
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,37 +1,2 @@
----
-title: Gpt2 App
-emoji: 🏢
-colorFrom: gray
-colorTo: gray
-sdk: streamlit
-app_file: app.py
-pinned: false
----
-
-# Configuration
-
-`title`: _string_  
-Display title for the Space
-
-`emoji`: _string_  
-Space emoji (emoji-only character allowed)
-
-`colorFrom`: _string_  
-Color for Thumbnail gradient (red, yellow, green, blue, indigo, purple, pink, gray)
-
-`colorTo`: _string_  
-Color for Thumbnail gradient (red, yellow, green, blue, indigo, purple, pink, gray)
-
-`sdk`: _string_  
-Can be either `gradio` or `streamlit`
-
-`sdk_version` : _string_  
-Only applicable for `streamlit` SDK.  
-See [doc](https://hf.co/docs/hub/spaces) for more info on supported versions.
-
-`app_file`: _string_  
-Path to your main application file (which contains either `gradio` or `streamlit` Python code).  
-Path is relative to the root of the repository.
-
-`pinned`: _boolean_  
-Whether the Space stays on top of your list.
+# gpt2-app
+Streamlit app showing some Indonesian GPT-2 models

app/SessionState.py

@@ -0,0 +1,107 @@
+"""Hack to add per-session state to Streamlit.
+Usage
+-----
+>>> import SessionState
+>>>
+>>> session_state = SessionState.get(user_name='', favorite_color='black')
+>>> session_state.user_name
+''
+>>> session_state.user_name = 'Mary'
+>>> session_state.favorite_color
+'black'
+Since you set user_name above, next time your script runs this will be the
+result:
+>>> session_state = get(user_name='', favorite_color='black')
+>>> session_state.user_name
+'Mary'
+"""
+try:
+    import streamlit.ReportThread as ReportThread
+    from streamlit.server.Server import Server
+except Exception:
+    # Streamlit >= 0.65.0
+    import streamlit.report_thread as ReportThread
+    from streamlit.server.server import Server
+
+
+class SessionState(object):
+    def __init__(self, **kwargs):
+        """A new SessionState object.
+        Parameters
+        ----------
+        **kwargs : any
+            Default values for the session state.
+        Example
+        -------
+        >>> session_state = SessionState(user_name='', favorite_color='black')
+        >>> session_state.user_name = 'Mary'
+        ''
+        >>> session_state.favorite_color
+        'black'
+        """
+        for key, val in kwargs.items():
+            setattr(self, key, val)
+
+
+def get(**kwargs):
+    """Gets a SessionState object for the current session.
+    Creates a new object if necessary.
+    Parameters
+    ----------
+    **kwargs : any
+        Default values you want to add to the session state, if we're creating a
+        new one.
+    Example
+    -------
+    >>> session_state = get(user_name='', favorite_color='black')
+    >>> session_state.user_name
+    ''
+    >>> session_state.user_name = 'Mary'
+    >>> session_state.favorite_color
+    'black'
+    Since you set user_name above, next time your script runs this will be the
+    result:
+    >>> session_state = get(user_name='', favorite_color='black')
+    >>> session_state.user_name
+    'Mary'
+    """
+    # Hack to get the session object from Streamlit.
+
+    ctx = ReportThread.get_report_ctx()
+
+    this_session = None
+
+    current_server = Server.get_current()
+    if hasattr(current_server, '_session_infos'):
+        # Streamlit < 0.56
+        session_infos = Server.get_current()._session_infos.values()
+    else:
+        session_infos = Server.get_current()._session_info_by_id.values()
+
+    for session_info in session_infos:
+        s = session_info.session
+        if (
+            # Streamlit < 0.54.0
+            (hasattr(s, '_main_dg') and s._main_dg == ctx.main_dg)
+            or
+            # Streamlit >= 0.54.0
+            (not hasattr(s, '_main_dg') and s.enqueue == ctx.enqueue)
+            or
+            # Streamlit >= 0.65.2
+            (not hasattr(s, '_main_dg') and s._uploaded_file_mgr == ctx.uploaded_file_mgr)
+        ):
+            this_session = s
+
+    if this_session is None:
+        raise RuntimeError(
+            "Oh noes. Couldn't get your Streamlit Session object. "
+            'Are you doing something fancy with threads?')
+
+    # Got the session object! Now let's attach some state into it.
+
+    if not hasattr(this_session, '_custom_session_state'):
+        this_session._custom_session_state = SessionState(**kwargs)
+
+    return this_session._custom_session_state
+
+__all__ = ['get']

app/app.py

@@ -0,0 +1,176 @@
+import streamlit as st
+import SessionState
+from mtranslate import translate
+from prompts import PROMPT_LIST
+import random
+import time
+from transformers import pipeline, set_seed
+import psutil
+import codecs
+import streamlit.components.v1 as stc
+import shutil
+import pathlib
+
+# st.set_page_config(page_title="Indonesian Story Generator")
+
+MODELS = {
+    "Indonesian Literature - GPT-2 Small": {
+        "name": "cahya/gpt2-small-indonesian-story",
+        "text_generator": None
+    },
+    "Indonesian Literature - GPT-2 Medium": {
+        "name": "cahya/gpt2-medium-indonesian-story",
+        "text_generator": None
+    },
+    "Indonesian Persona Chatbot": {
+        "name": "",
+        "text_generator": None
+    },
+}
+
+
+def stc_chatbot(html_file, width=700, height=900):
+    html = codecs.open(html_file, "r")
+    page = html.read()
+    stc.html(page, width=width, height=height, scrolling=True)
+
+
+model = st.sidebar.selectbox('Model', (MODELS.keys()))
+
+
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
+def get_generator(model_name: str):
+    st.write(f"Loading the GPT2 model {model_name}, please wait...")
+    text_generator = pipeline('text-generation', model=model_name)
+    return text_generator
+
+
+# Disable the st.cache for this function due to issue on newer version of streamlit
+# @st.cache(suppress_st_warning=True, hash_funcs={tokenizers.Tokenizer: id})
+def process(text_generator, text: str, max_length: int = 100, do_sample: bool = True, top_k: int = 50, top_p: float = 0.95,
+            temperature: float = 1.0, max_time: float = 60.0, seed=42):
+    # st.write("Cache miss: process")
+    set_seed(seed)
+    result = text_generator(text, max_length=max_length, do_sample=do_sample,
+                            top_k=top_k, top_p=top_p, temperature=temperature,
+                            max_time=max_time)
+    return result
+
+
+st.title("Indonesian GPT-2 Applications")
+prompt_group_name = ""
+if model.find("Indonesian Literature") != -1:
+    st.subheader("Indonesian Literature")
+    prompt_group_name = "Indonesian Literature"
+    st.markdown(
+        """
+        This application is a demo for Indonesian Literature Generator using GPT2.
+        """
+    )
+    session_state = SessionState.get(prompt=None, prompt_box=None, text=None)
+    ALL_PROMPTS = list(PROMPT_LIST[prompt_group_name].keys())+["Custom"]
+
+    prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
+
+    # Update prompt
+    if session_state.prompt is None:
+        session_state.prompt = prompt
+    elif session_state.prompt is not None and (prompt != session_state.prompt):
+        session_state.prompt = prompt
+        session_state.prompt_box = None
+        session_state.text = None
+    else:
+        session_state.prompt = prompt
+
+    # Update prompt box
+    if session_state.prompt == "Custom":
+        session_state.prompt_box = "Enter your text here"
+    else:
+        print(f"# prompt: {session_state.prompt}")
+        print(f"# prompt_box: {session_state.prompt_box}")
+        if session_state.prompt is not None and session_state.prompt_box is None:
+            session_state.prompt_box = random.choice(PROMPT_LIST[prompt_group_name][session_state.prompt])
+
+    session_state.text = st.text_area("Enter text", session_state.prompt_box)
+
+    max_length = st.sidebar.number_input(
+        "Maximum length",
+        value=100,
+        max_value=512,
+        help="The maximum length of the sequence to be generated."
+    )
+
+    temperature = st.sidebar.slider(
+        "Temperature",
+        value=1.0,
+        min_value=0.0,
+        max_value=10.0
+    )
+
+    do_sample = st.sidebar.checkbox(
+        "Use sampling",
+        value=True
+    )
+
+    top_k = 40
+    top_p = 0.95
+
+    if do_sample:
+        top_k = st.sidebar.number_input(
+            "Top k",
+            value=top_k
+        )
+        top_p = st.sidebar.number_input(
+            "Top p",
+            value=top_p
+        )
+
+    seed = st.sidebar.number_input(
+        "Random Seed",
+        value=25,
+        help="The number used to initialize a pseudorandom number generator"
+    )
+
+    for group_name in MODELS:
+        if group_name.find("Indonesian Literature") != -1:
+            MODELS[group_name]["text_generator"] = get_generator(MODELS[group_name]["name"])
+    # text_generator = get_generator()
+    if st.button("Run"):
+        with st.spinner(text="Getting results..."):
+            memory = psutil.virtual_memory()
+            st.subheader("Result")
+            time_start = time.time()
+            # text_generator = MODELS[model]["text_generator"]
+            result = process(MODELS[model]["text_generator"], text=session_state.text, max_length=int(max_length),
+                             temperature=temperature, do_sample=do_sample,
+                             top_k=int(top_k), top_p=float(top_p), seed=seed)
+            time_end = time.time()
+            time_diff = time_end-time_start
+            result = result[0]["generated_text"]
+            st.write(result.replace("\n", "  \n"))
+            st.text("Translation")
+            translation = translate(result, "en", "id")
+            st.write(translation.replace("\n", "  \n"))
+            # st.write(f"*do_sample: {do_sample}, top_k: {top_k}, top_p: {top_p}, seed: {seed}*")
+            info = f"""
+            *Memory: {memory.total/(1024*1024*1024):.2f}GB, used: {memory.percent}%, available: {memory.available/(1024*1024*1024):.2f}GB*        
+            *Text generated in {time_diff:.5} seconds*
+            """
+            st.write(info)
+
+            # Reset state
+            session_state.prompt = None
+            session_state.prompt_box = None
+            session_state.text = None
+elif model == "Indonesian Persona Chatbot":
+    st.subheader("Indonesian GPT-2 Persona Chatbot")
+    STREAMLIT_STATIC_PATH = pathlib.Path(st.__path__[0]) / 'static'
+    # We create a videos directory within the streamlit static asset directory
+    # and we write output files to it
+    ASSETS_PATH = STREAMLIT_STATIC_PATH/"gpt2-app"
+    if not ASSETS_PATH.is_dir():
+        ASSETS_PATH.mkdir()
+        shutil.copytree("app/css", ASSETS_PATH/"css")
+        shutil.copytree("app/js", ASSETS_PATH/"js")
+
+    stc_chatbot("app/chatbot.html")

app/chatbot.html

@@ -0,0 +1,134 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <!-- Required meta tags -->
+    <meta charset="utf-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1">
+
+    <!-- Bootstrap CSS -->
+    <link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css" rel="stylesheet" integrity="sha384-F3w7mX95PdgyTmZZMECAngseQB83DfGTowi0iMjiWaeVhAn4FJkqJByhZMI3AhiU" crossorigin="anonymous">
+
+    <title>Indonesian GPT2 Chatbot</title>
+    <link rel="stylesheet" href="gpt2-app/css/main.css">
+    <script src="gpt2-app/js/main.js"></script>
+</head>
+<body onload="pageSetup();">
+    <script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/js/bootstrap.bundle.min.js" integrity="sha384-/bQdsTh/da6pkI1MST/rWKFNjaCP5gBSY4sEBT38Q/9RBh9AH40zEOg7Hlq2THRZ" crossorigin="anonymous"></script>
+
+    <div class="buttons" style="display: none">
+        <div class="minus button">-</div>
+        <div class="value">?</div>
+        <div class="plus button">+</div>
+    </div>
+    <div class="state" style="display: none">
+        <span class="users"></span>
+    </div>
+
+    <div class="container">
+        <div class="chat-container">
+            <div class="chat-messages">
+                <div class="messages">
+                </div>
+            </div>
+        </div>
+        <div class="chat-input input-group mb-3">
+            <input type="text" class="form-control user-input" placeholder="Type a message..." aria-label="User message" aria-describedby="basic-addon2">
+            <span class="input-group-text btn btn-primary user-input-button" id="basic-addon2">Send</span>
+        </div>
+        <!--
+        <div class="chat-suggestion">
+            Suggestion: <span class="js-loading">Loading…</span> <a class="js-suggestion hide">Kenapa kamu sedih?</a>
+        </div>
+        -->
+        <div class="server-message">
+            <span class="server-message-value"></span>
+        </div>
+        <div class="accordion" id="accordionExample">
+            <div class="accordion-item">
+                <h2 class="accordion-header" id="headingOne">
+                    <button class="accordion-button" type="button" data-bs-toggle="collapse" data-bs-target="#collapseOne" aria-expanded="true" aria-controls="collapseOne">
+                        Bots Personalities
+                    </button>
+                </h2>
+                <div id="collapseOne" class="accordion-collapse collapse show" aria-labelledby="headingOne" data-bs-parent="#accordionExample">
+                    <form class="bot-personality">
+                        <div class="mb-3">
+                            <div class="row g-2 align-items-center">
+                                <div class="col-auto">
+                                    <label for="inputPersonality1" class="col-form-label">Personality 1:</label>
+                                </div>
+                                <div class="col-auto">
+                                    <input type="text" class="form-control" id="inputPersonality1">
+                                </div>
+                            </div>
+                            <div class="row g-2 align-items-center">
+                                <div class="col-auto">
+                                    <label for="inputPersonality2" class="col-form-label">Personality 2:</label>
+                                </div>
+                                <div class="col-auto">
+                                    <input type="text" class="form-control" id="inputPersonality2">
+                                </div>
+                            </div>
+                            <div class="row g-2 align-items-center">
+                                <div class="col-auto">
+                                    <label for="inputPersonality3" class="col-form-label">Personality 3:</label>
+                                </div>
+                                <div class="col-auto">
+                                    <input type="text" class="form-control" id="inputPersonality3">
+                                </div>
+                            </div>
+                            <div class="row g-2 align-items-center">
+                                <div class="col-auto">
+                                    <label for="inputPersonality4" class="col-form-label">Personality 4:</label>
+                                </div>
+                                <div class="col-auto">
+                                    <input type="text" class="form-control" id="inputPersonality4">
+                                </div>
+                            </div>
+                            <div class="row g-2 align-items-center">
+                                <div class="col-auto">
+                                    <label for="inputPersonality5" class="col-form-label">Personality 5:</label>
+                                </div>
+                                <div class="col-auto">
+                                    <input type="text" class="form-control" id="inputPersonality5">
+                                </div>
+                            </div>
+                        </div>
+                        <button id="updatePersonality" class="btn btn-primary" type="button" data-bs-toggle="collapse" data-bs-target="#collapseExample" aria-expanded="false" aria-controls="collapseExample">
+                            Update Personality
+                        </button>
+                    </form>
+                </div>
+            </div>
+            <div class="accordion-item">
+                <h2 class="accordion-header" id="headingThree">
+                    <button class="accordion-button collapsed" type="button" data-bs-toggle="collapse" data-bs-target="#collapseThree" aria-expanded="false" aria-controls="collapseThree">
+                        Parameters
+                    </button>
+                </h2>
+                <div id="collapseThree" class="accordion-collapse collapse" aria-labelledby="headingThree" data-bs-parent="#accordionExample">
+
+                    <div class="chat-parameter card card-body">
+                        <div class="form-check">
+                            <input class="form-check-input" type="checkbox" value="" id="doSample" checked>
+                            <label class="form-check-label" for="doSample">
+                                Do Sample
+                            </label>
+                        </div>
+                        <label for="minLength" class="form-label">Minimal Length: <span id="minLengthValue">1</span></label>
+                        <input type="range" class="form-range" min="1" max="10" value="1" id="minLength" onmousemove="updateValue('minLengthValue', this.value);">
+                        <label for="maxLength" class="form-label">Maximal Length: <span id="maxLengthValue">20</span></label>
+                        <input type="range" class="form-range" min="20" max="50" value="20" id="maxLength" onmousemove="updateValue('maxLengthValue', this.value);">
+                        <label for="temperature" class="form-label">Temperature: <span id="temperatureValue">0.7</span></label>
+                        <input type="range" class="form-range" min="0.5" max="10" value="0.7" step="0.1" id="temperature" onmousemove="updateValue('temperatureValue', this.value);">
+                        <label for="topK" class="form-label">Top k: <span id="topKValue">0</span></label>
+                        <input type="range" class="form-range" min="0" max="50" value="0" id="topK" onmousemove="updateValue('topKValue', this.value);">
+                        <label for="topP" class="form-label">Top p: <span id="topPValue">0.9</span></label>
+                        <input type="range" class="form-range" min="0.1" max="1.0" value="0.9" step="0.01" id="topP" onmousemove="updateValue('topPValue', this.value);">
+                    </div>
+                </div>
+            </div>
+        </div>
+    </div>
+</body>
+</html>

app/css/main.css

@@ -0,0 +1,109 @@
+body {
+    font-family: "Arial, Courier New", sans-serif;
+    text-align: center;
+}
+
+h1 {
+    margin: 10px 10px;
+}
+
+.buttons {
+    font-size: 2em;
+    display: flex;
+    justify-content: center;
+}
+
+.button, .value {
+    line-height: 1;
+    padding: 1rem;
+    margin: 1rem;
+    border: medium solid;
+    min-height: 1em;
+    min-width: 1em;
+}
+
+.button {
+    cursor: pointer;
+    user-select: none;
+}
+
+.minus {
+    color: red;
+}
+
+.plus {
+    color: green;
+}
+
+.value {
+    min-width: 2em;
+}
+
+.state {
+    font-size: 2em;
+}
+
+.container {
+    min-width: 30em;
+    max-width: 40em;
+}
+
+.accordion-collapse {
+    padding: 5px 5px 0 5px;
+}
+
+.chat-container {
+    margin: 10px 0;
+    min-height: 300px;
+    max-height: 600px;
+    overflow: auto;
+}
+
+.bot-personality {
+    text-align: left;
+    margin: 0 0 5px 0;
+}
+
+.chat-parameter {
+    text-align: left;
+    margin: 5px 0 5px 0;
+}
+
+.bot-personality input {
+    margin: 5px 0 0 0;
+    min-width: 20em;
+}
+
+.message {
+    margin: 5px 0;
+}
+
+.message-inner {
+    font-size: 16px;
+}
+
+.outgoing {
+    text-align: right;
+}
+
+.outgoing .badge {
+    text-align: right;
+}
+
+.botPersonality, .incoming, .incoming .badge, .chat-suggestion, .server-message, .parameters {
+    text-align: left;
+}
+
+.chat-suggestion, .server-message
+{
+    padding-left: 5px;
+}
+
+.server-message-value {
+    font-style: italic;
+}
+
+#collapseParameter {
+    width: 300px;
+    margin: 8px 0px;
+}

app/js/main.js

@@ -0,0 +1,122 @@
+updateValue = function(id, value) {
+    document.getElementById(id).innerText = value;
+}
+
+htmlToElement = function(html) {
+    let template = document.createElement('template');
+    html = html.trim(); // Never return a text node of whitespace as the result
+    template.innerHTML = html;
+    return template.content.firstChild;
+}
+
+pageSetup = function() {
+    const minus = document.querySelector('.minus');
+    const plus = document.querySelector('.plus');
+    const value = document.querySelector('.value');
+    // const users = document.querySelector('.users');
+    const userInput = document.querySelector('.user-input');
+    const userInputButton = document.querySelector('.user-input-button');
+    const serverMessageValue = document.querySelector('.server-message-value');
+    const messages = document.querySelector('.messages');
+    const updatePersonality = document.getElementById("updatePersonality")
+    const websocket = new WebSocket("wss://gpt2-chat.ai-research.id/");
+    //const websocket = new WebSocket("ws://localhost:8502/");
+
+    minus.onclick = function () {
+        websocket.send(JSON.stringify({action: 'minus'}));
+    }
+
+    plus.onclick = function () {
+        websocket.send(JSON.stringify({action: 'plus'}));
+    }
+
+    updatePersonality.onclick = function () {
+        const elements = document.querySelectorAll(".bot-personality input")
+        let data = {
+            "action": "personality",
+            "message": []
+        }
+        for (let i = 0; i < Math.min(elements.length, 5); i++) {
+            if(elements[i].value.length >0)
+                data.message.push(elements[i].value);
+        }
+        websocket.send(JSON.stringify(data));
+    }
+
+    let getParameters = function() {
+        return {
+            "do_sample": document.getElementById("doSample").checked,
+            "min_length": parseInt(document.getElementById("minLength").value),
+            "max_length": parseInt(document.getElementById("maxLength").value),
+            "temperature": parseFloat(document.getElementById("temperature").value),
+            "top_k": parseInt(document.getElementById("topK").value),
+            "top_p": parseFloat(document.getElementById("topP").value),
+        };
+    }
+
+    let processUserInput = function (userInput) {
+        let parameters = getParameters();
+        parameters["action"] = "talk";
+        parameters["utterance"] = userInput.value;
+        websocket.send(JSON.stringify(parameters));
+        const element = htmlToElement("<div class=\"message outgoing\"><div class=\"message-inner badge bg-primary text-wrap\">"
+            + userInput.value + "</div></div>");
+        userInput.value = "";
+        messages.appendChild(element);
+        messages.scrollIntoView(false)
+    }
+
+    userInputButton.onclick = function () {
+        processUserInput(userInput);
+    }
+
+    userInput.addEventListener("keyup", function(event) {
+        if (event.keyCode === 13) {
+            // Cancel the default action, if needed
+            event.preventDefault();
+            processUserInput(userInput);
+        }
+    });
+
+    websocket.onmessage = function (event) {
+        let data = JSON.parse(event.data);
+        switch (data.type) {
+            case 'connection':
+                console.log(data.value)
+                websocket.send(JSON.stringify({action: 'dialog', personality: []}));
+                break;
+            case 'state':
+                value.textContent = data.value;
+                break;
+            case 'users':
+                serverMessageValue.textContent = (
+                    data.count.toString() + " user" +
+                    (data.count === 1 ? "" : "s") + " online");
+                break;
+            case 'dialog':
+                console.log(data.message)
+                break;
+            case 'talk':
+                const element = htmlToElement("<div class=\"message incoming\"><div class=\"message-inner badge bg-success text-wrap\">"
+                    + data.message+ "</div></div>");
+                messages.appendChild(element);
+                messages.scrollIntoView(false)
+                break;
+            case 'personality':
+                const elements = document.querySelectorAll(".bot-personality input")
+                for (let i = 0; i < Math.min(elements.length, data.message.length); i++) {
+                    elements[i].value = data.message[i];
+                }
+                break;
+            case 'personality_reply':
+                serverMessageValue.textContent = data.message
+                setTimeout(function() {
+                    websocket.send(JSON.stringify({action: 'get_users'}));
+                }, 3000);
+                break;
+            default:
+                console.error(
+                    "unsupported event", data);
+        }
+    };
+}

app/prompts.py

@@ -0,0 +1,20 @@
+PROMPT_LIST = {
+    "Indonesian Literature": {
+        "Adult Romance": [
+            "Ini adalah kisah tentang seorang laki-laki yang berusaha memperjuangkan cintanya",
+            "Alunan musik terdengar memenuhi ruangan kantor, cowok itu duduk di balik meja kerjanya sambil memejamkan mata. Berusaha meresapi nada per nada",
+            "Aku mencari dan terus mencari\nDimana bahagia akan kutemui\nKumencari terus mencari\nHingga ku tak mengerti arti hari-hari",
+            "Gadis itu mengharuskan dirinya tegar, dan kuat dalam menghadapi masalah. Menahan air matanya jatuh setiap kali ingin menangis"
+        ],
+        "Horror": [
+            "Ditengah-tengah perbincangan mereka berdua, datanglah sesosok mahluk tinggi hitam dan besar",
+            "Sesosok hantu perempuan seperti kuntilanak yang melayang keluar dan bergerak perlahan dari pintu kamar kecil tadi yang tertutup.",
+            "Sejak pertemuannya dengan leak, yang ternyata tinggal satu atap dengannya, hidupnya terus dihantui oleh berbagai sosok seram."
+        ],
+        "Poetry": [
+            "Aku ingin menulis sajak\nyang melesat dalam kejap\nmenembus hati yang pejam\nmemaksa mimpimu terjaga\ndari semu",
+            "Malam ini langitku lengang\ntiada hujan yang membasuh rindu\npun awan yang biasanya temani seruput kopimu",
+            "Di sisimu waktu menjelma\nsetangkai kembang api\ngelora membakar tanpa jeda\nmemercik pijar binar kita."
+        ]
+    }
+}

requirements.txt

@@ -0,0 +1,10 @@
+numpy
+torch
+tokenizers
+transformers
+datasets
+mtranslate
+# streamlit version 0.67.1 is needed due to issue with caching
+# streamlit==0.67.1
+streamlit
+psutil