feat: add update states

app.py

@@ -6,13 +6,20 @@ gpt_tokenizer = IndoNLGTokenizer.from_pretrained("indobenchmark/indogpt")
 gpt_tokenizer.pad_token = gpt_tokenizer.eos_token
 kancilgpt = GPT2LMHeadModel.from_pretrained("abdiharyadi/kancilgpt")
 
-def generate_story(judul: str):
-    yield "...", "..."
-
+def generate_story():
     stop = False
+    ellipsis = "..."
     while not stop:
+        yield f"(memuat cerita {ellipsis})"
+        ellipsis += "."
+
         gpt_input = gpt_tokenizer('<s> awal cerita | judul:', return_tensors='pt')
-        gpt_out = kancilgpt.generate(**gpt_input, do_sample=True, max_length=512, pad_token_id=gpt_tokenizer.eos_token_id)
+        gpt_out = kancilgpt.generate(
+            **gpt_input,
+            do_sample=True,
+            max_length=512,
+            pad_token_id=gpt_tokenizer.eos_token_id
+        )
         result = gpt_tokenizer.decode(gpt_out[0])
         _, judul_prompt, isi, *end_part = result.split(" | ")
         end_part = "".join(end_part)
@@ -36,12 +43,12 @@ def generate_story(judul: str):
 
         stop = True
 
-    yield judul, isi + " ..."
-
     total_isi = isi
 
+    ellipsis = "..."
     while not end_part.startswith("tamat"):
-        yield judul, total_isi + " ..."
+        yield judul + "\n" + ("-" * len(judul)) + "\n" + total_isi + f" {ellipsis}"
+        ellipsis += "."
 
         i = 0
         in_quote = False
@@ -98,14 +105,14 @@ def generate_story(judul: str):
             stop = True
 
         total_isi += " " + isi[len(next_isi) + 1:]
+        ellipsis = "..."
 
-    yield judul, total_isi + "\n\ntamat."
+    yield judul + "\n" + ("-" * len(judul)) + "\n" + total_isi + "\n\ntamat."
 
 demo = gr.Interface(
     fn=generate_story,
     inputs=None,
     outputs=[
-        gr.Textbox(label="judul", lines=1),
         gr.Textbox(label="cerita", lines=7)
     ]
 )