Update app.py

app.py

@@ -133,13 +133,14 @@ def chat_with_model(prompt, document_section, model_choice='gpt-3.5-turbo'):
     result_textarea = st.empty()
     results=[]
     for responses in openai.ChatCompletion.create(model=model, messages=conversation, stream=True):
-        results.append(responses.choices[0].text)
+        results.append(responses['choices'][0].text)
         result = "".join(results).strip()
         result = result.replace('\n','')
         result_textarea.markdown(f'*{result}*')
 
     #return response
-    return response['choices'][0]['message']['content']
+    #return response['choices'][0]['message']['content']
+    return results
     
 
 def chat_with_file_contents(prompt, file_content, model_choice='gpt-3.5-turbo'):