Update app.py

app.py

@@ -169,11 +169,12 @@ def chat_with_model(prompt, document_section, model_choice='gpt-3.5-turbo'):
         messages=conversation,
         stream=True,
     )
+    all_content = ""  # Initialize an empty string to hold all content
     for part in stream:
         chunk_message = (part.choices[0].delta.content or "")
         collected_messages.append(chunk_message)  # save the message
-        #content=part["choices"][0].get("delta",{}).get("content")
         content=part.choices[0].delta.content
+        all_content += content  
         try:
             report.append(content)
             if len(content) > 0:
@@ -181,7 +182,7 @@ def chat_with_model(prompt, document_section, model_choice='gpt-3.5-turbo'):
                 res_box.markdown(f'*{result}*') 
         except:
             st.write(' ')
-    full_reply_content = ''.join([m.get('content', '') for m in collected_messages])
+    full_reply_content = all_content
 
     
     #for chunk in client.chat.completions.create(