Update app.py

app.py

@@ -79,41 +79,38 @@ def chat_with_model(prompt, document_section, model_choice='gpt-3.5-turbo'):
     conversation.append({'role': 'user', 'content': prompt})
     if len(document_section)>0:
         conversation.append({'role': 'assistant', 'content': document_section})
-        
     start_time = time.time()
     report = []
     res_box = st.empty()
     collected_chunks = []
     collected_messages = []
-    
     key = os.getenv('OPENAI_API_KEY')
     openai.api_key = key
+    
     for chunk in openai.ChatCompletion.create(
         model='gpt-3.5-turbo',
         messages=conversation,
         temperature=0.5,
         stream=True  
     ):
-        
         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")
-        
         try:
             report.append(content)
             if len(content) > 0:
                 result = "".join(report).strip()
-                #result = result.replace("\n", "")        
                 res_box.markdown(f'*{result}*') 
         except:
             st.write(' ')
-        
+            
     full_reply_content = ''.join([m.get('content', '') for m in collected_messages])
     st.write("Elapsed time:")
     st.write(time.time() - start_time)
     readitaloud(full_reply_content)
+    filename = generate_filename(full_reply_content, choice)
+    create_file(filename, prompt, full_reply_content, should_save)
     return full_reply_content
 
 def chat_with_file_contents(prompt, file_content, model_choice='gpt-3.5-turbo'):