Update app.py

app.py

@@ -139,15 +139,17 @@ def chat_with_model(prompt, document_section, model_choice='gpt-3.5-turbo'):
     collected_chunks = []
     collected_messages = []
     for chunk in response:
-        chunk_time = time.time() - start_time  # calculate the time delay of the chunk
+        #chunk_time = time.time() - start_time  # calculate the time delay of the chunk
         collected_chunks.append(chunk)  # save the event response
         chunk_message = chunk['choices'][0]['delta']  # extract the message
         collected_messages.append(chunk_message)  # save the message
         content=chunk["choices"][0].get("delta",{}).get("content")
-        #st.write(f'*{content}*')
-    #st.write(f"Full response received {chunk_time:.2f} seconds after request")
+        st.write(f'*{content}*')
+    # st.write(f"Full response received {chunk_time:.2f} seconds after request")
     full_reply_content = ''.join([m.get('content', '') for m in collected_messages])
-    st.write(f"Full conversation received: {full_reply_content}")
+    # st.write(f"Full conversation received: {full_reply_content}")
+    st.write("Elapsed time:")
+    st.write(time.time() - start_time)
     return full_reply_content
 
 def chat_with_file_contents(prompt, file_content, model_choice='gpt-3.5-turbo'):