Fix streaming

app.py

@@ -125,8 +125,9 @@ with Chat_Col:
             time_to_completion = round(perf_counter() - start, 2)
 
             # Writing answer progressively
-            chunks = [chunk for chunk in stream]
-            st.write_stream(chunks)
+            stream, stream_copy = itertools.tee(stream)
+            st.write_stream(stream)
+            chunks = [chunk for chunk in stream_copy]
 
             # Computing metrics
             last_chunk = chunks[-1]