updated the prompts

app/abstract_dataset.py

@@ -1,62 +0,0 @@
-import torch
-import random
-from torch.utils.data import Dataset
-
-
-class AbstractDataset(Dataset):
-    special_tokens = {"bos_token": "<|BOS|>",
-                      "eos_token": "<|EOS|>",
-                      "unk_token": "<|UNK|>",
-                      "pad_token": "<|PAD|>",
-                      "sep_token": "<|SEP|>"}
-    max_length = 1024
-
-    def __init__(self, data, tokenizer, randomize=True):
-        title, text, keywords = [], [], []
-        for k, v in data.items():
-            title.append(v[0])
-            text.append(v[1])
-            keywords.append(v[2])
-
-        self.randomize = randomize
-        self.tokenizer = tokenizer
-        self.title = title
-        self.text = text
-        self.keywords = keywords
-
-    @staticmethod
-    def join_keywords(keywords, randomize=True):
-        N = len(keywords)
-
-        # random sampling and shuffle
-        if randomize:
-            # M = random.choice(range(N + 1))
-            # keywords = keywords[:M]
-            random.shuffle(keywords)
-
-        return ','.join(keywords)
-
-
-    def __len__(self):
-        return len(self.text)
-
-
-    def __getitem__(self, i):
-        keywords = self.keywords[i].copy()
-        kw = self.join_keywords(keywords, self.randomize)
-
-        input = self.special_tokens['bos_token'] + self.title[i] + \
-                self.special_tokens['sep_token'] + kw + self.special_tokens['sep_token'] + \
-                self.text[i] + self.special_tokens['eos_token']
-
-        encodings_dict = self.tokenizer(input,
-                                   truncation=True,
-                                   max_length=self.max_length,
-                                   padding="max_length")
-
-        input_ids = encodings_dict['input_ids']
-        attention_mask = encodings_dict['attention_mask']
-
-        return {'label': torch.tensor(input_ids),
-                'input_ids': torch.tensor(input_ids),
-                'attention_mask': torch.tensor(attention_mask)}

app/app.py

@@ -66,9 +66,7 @@ def process(title: str, keywords: str, text: str,
             penalty_alpha = 0.6):
     # st.write("Cache miss: process")
     url = 'https://news-api.uncool.ai/api/text_generator/v1'
-    print("news_api_auth_token:", news_api_auth_token)
     headers = {'Authorization': 'Bearer ' + news_api_auth_token}
-    print("Requesting to news-api.uncool.ai with headers: ", headers)
     data = {
         "title": title,
         "keywords": keywords,
@@ -122,7 +120,10 @@ if prompt_group_name in ["Indonesian Newspaper"]:
         session_state.keywords = ""
     else:
         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])
+            choice = random.choice(PROMPT_LIST[prompt_group_name][session_state.prompt])
+            session_state.title = choice["title"]
+            session_state.keywords = choice["keywords"]
+            session_state.prompt_box = choice["text"]
 
     session_state.title = st.text_input("Title", session_state.title)
     session_state.keywords = st.text_input("Keywords", session_state.keywords)

app/prompts.py

@@ -1,57 +1,38 @@
 PROMPT_LIST = {
-    "Indonesian GPT-2": {
-        "Resep masakan (recipe)": [
-            "Berikut adalah cara memasak sate ayam:\n",
-            "Langkah-langkah membuat nasi goreng:\n",
-            "Berikut adalah bahan-bahan membuat nastar:\n"
-        ],
-        "Puisi (poetry)": [
-            "Aku ingin jadi merpati\nTerbang di langit yang damai\nBernyanyi-nyanyi tentang masa depan\n",
-            "Terdiam aku satu persatu dengan tatapan binar\nSenyawa merasuk dalam sukma membuat lara\nKefanaan membentuk kelemahan"
-        ],
-        "Cerpen (short story)": [
-            "Putri memakai sepatunya dengan malas. Kalau bisa, selama seminggu ini ia bolos sekolah saja. Namun, Mama pasti akan marah. Ulangan tengah semester telah selesai. Minggu ini, di sekolah sedang berlangsung pekan olahraga.",
-            "\"Wah, hari ini cerah sekali ya,\" ucap Budi ketika ia keluar rumah.",
-            "Sewindu sudah kita tak berjumpa, rinduku padamu sudah tak terkira."
-        ],
-        "Sejarah (history)": [
-            "Mohammad Natsir adalah seorang ulama, politisi, dan pejuang kemerdekaan Indonesia.",
-            "Ir. H. Soekarno adalah Presiden pertama Republik Indonesia. Ia adalah seorang tokoh perjuangan yang memainkan peranan penting dalam memerdekakan bangsa Indonesia",
-            "Borobudur adalah sebuah candi Buddha yang terletak di sebelah barat laut Yogyakarta. Monumen ini merupakan model alam semesta dan dibangun sebagai tempat suci untuk memuliakan Buddha"
-        ],
-    },
-    "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"
+    "Indonesian Newspaper": {
+        "Politik": [
+            {
+                "title": "Wapres Berharap Muhammadiyah Berkontribusi Wujudkan Indonesia Emas",
+                "keywords": "Indonesia Emas, Muhammadiyah, Wapres",
+                "text": "Wakil Presiden (Wapres) Ma'ruf Amin berharap Muhammadiyah dapat berkontribusi dalam upaya mewujudkan Indonesia Emas pada tahun 2045 mendatang."
+            },
+            {
+                "title": "Jokowi: Pembangunan Infrastuktur Perlu Berdayakan Masyarakat Setempat",
+                "keywords": "Infrastuktur, Jokowi, Masyarakat",
+                "text": "Presiden Joko Widodo menekankan bahwa pembangunan infrastruktur perlu memberdayakan masyarakat dan ekonomi setempat agar memiliki rasa kepemilikan yang tinggi disertai dukungan bagi negara berkembang untuk membangun kapasitas dan kemampuan mandiri"
+            },
+            {
+                "title": "PKB Nilai Jokowi Masih Menimbang Beri Dukungan ke Prabowo atau Ganjar",
+                "keywords": "Joko Widodo, Prabowo Subianto, Ganjar Pranowo",
+                "text": "Wakil Sekretaris Jenderal (Wasekjen) Partai Kebangkitan Bangsa (PKB) Syaiful Huda menilai, Presiden Joko Widodo masih menimbang bakal memberikan dukungan pada Prabowo Subianto atau Ganjar Pranowo."
+            },
         ],
-        "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."
+        "Wisata": [
+            {
+                "title": "Kebun Raya Bogor sebagai Taman Hutan Raya Terbesar di Indonesia",
+                "keywords": "2022",
+                "text": "Pemerintah Kota (Pemkot) Bogor, Jawa Baat berencana membangun kebun raya seluas 87.3 hektar"
+            },
+            {
+                "title": "Kepulauan Raja Ampat",
+                "keywords": "Papua Barat, Wisata",
+                "text": "Kepulauan Raja Ampat merupakan rangkaian empat gugusan pulau yang berdekatan dan berlokasi di barat bagian Kepala Burung (Vogelkoop) Pulau Papua."
+            },
+            {
+                "title": "Ubud, kota indah di Bali",
+                "keywords": "Ubud, Wisata",
+                "text": "Ubud adalah sebuah desa adat sekaligus menjadi destinasi wisata di daerah kabupaten Gianyar, pulau Bali, Indonesia."
+            },
         ],
-        "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."
-        ]
-    },
-    "Indonesian Newspaper": {
-        "Biologi (biology)": [
-            "Tujuan  penelitian ini untuk menentukan keanekaragaman Arthropoda pada lahan pertanian kacang",
-            "Identifikasi spesies secara molekuler sangat diperlukan dalam mempelajari taksonomi",
-            "Penelitian ini bertujuan untuk menentukan identitas invertebrata laut dari Perairan Papua dengan teknik DNA barcoding"],
-        "Psikologi (psychology)": [
-            "Penelitian ini bertujuan untuk mengetahui perilaku wirausaha remaja yang diprediksi dari motivasi intrinsik",
-            "Tujuan dari penelitian ini adalah untuk mendapatkan data empiris mengenai gambaran peta bakat mahasiswa Fakultas Psikologi Unjani"],
-        "Ekonomi (economics)": [
-            "Faktor kepuasan dan kepercayaan konsumen merupakan dua faktor kunci dalam meningkatkan penetrasi e-commerce. Penelitian yang dilakukan",
-            "Penelitian ini bertujuan untuk menganalisis pola konsumsi pangan di Indonesia",
-            "Model GTAP diimplementasikan untuk melihat dampak yang ditimbulkan pada PDB"],
-        "Teknologi Informasi (IT)": [
-            "pembuatan aplikasi ini menggunakan pengembangan metode Waterfall dan dirancang mengguynakan Unified Modeling Language (UML) dengan bahasa pemrograman",
-            "Berdasarkan masalah tersebut, maka penulis termotivasi untuk membangun Pengembangan Sistem Informasi Manajemen"]
     },
 }