better display

app.py

@@ -52,24 +52,34 @@ def sum_chain_l1(text, p_bar):
                             top_k=top_k,
                             top_p=top_p,
                             temperature=temperature):
-                yield answer
+                yield f"{'='*8} {i}/{len(text)//2000+1} {'='*8}\n{answer}"
             docs.append(answer)
     return docs
 
-def sum_chain_l2_deprecated(docs):
-  hist = ''
-  for doc in tqdm(docs):
-    hist = model.response(prompt+"\n"+hist+"\n"+doc)
-  return hist
+def sum_chain_l2_deprecated(docs, pbar):
+    hist = docs[0]
+    for doc in p_bar(docs[1:]):
+        for answer in model.generate_iterate(prompt+"\n"+hist+"\n"+doc,
+                    max_generated_tokens=max_tokens,
+                    top_k=top_k,
+                    top_p=top_p,
+                    temperature=temperature):
+            yield answer
+        hist = answer
+    return hist
 
 import gradio as gr
 
-def greet(x, progress=gr.Progress()):
-    progress(0, desc="Starting...")
+def greet(text, progress=gr.Progress()):
+    progress(0, desc="Reading")
     docs = []
-    for doc in sum_chain_l1(x, progress.tqdm):
+    for doc in sum_chain_l1(text, progress.tqdm):
         yield doc
-    return '===== summarized parts ====='.join(doc)
+    # yield '\n===== summarized parts =====\n'.join(doc)
+    progress(0, desc="Refinement")
+    for ans in sum_chain_l2_deprecated(doc, progress.tqdm):
+        yield '# drafting and refining summary =\n' + ans
+    return ans
 
 iface = gr.Interface(fn=greet,
                      inputs=gr.Textbox(lines=20,