Update app.py

app.py

@@ -25,13 +25,28 @@ gen_kwargs = dict(
 )
 prompt = f"Write instructions to teach anyone to write a discharge plan. List the entities, features and relationships to CCDA and FHIR objects in boldface."
 stream = client.text_generation(prompt, stream=True, details=True, **gen_kwargs)
+report=[]
+res_box = st.empty()
+collected_chunks=[]
+collected_messages=[]
 for r in stream:
     if r.token.special:
         continue
     if r.token.text in gen_kwargs["stop_sequences"]:
         break
+    collected_chunks.append(r.token.text)
+    #chunk_message = r.token.text
+    collected_messages.append(chunk_message)
+
+    try:
+        report.append(content)
+        if len(r.token.text) > 0:
+            result="".join(report).strip()
+            res_box.markdown(f'*{result}*')
+
+    #full_reply = ''.join()
     #st.markdown(r.token.text, end = "")
-    st.write(r.token.text)
+    #st.write(r.token.text)
 
 
 def query(payload):