Update logging...

app.py

@@ -25,7 +25,9 @@ if prompt := st.chat_input("What's up?"):
     st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user"):
         st.markdown(prompt)
-    with st.chat_message("assistant"):        
+    with st.chat_message("assistant"):
+        print(f"Prompt: {prompt}")
+        
         memory = pinecone_client.query_conversation(messages=st.session_state.messages[:-5], user=st.session_state['user'], top_k=3)
         print(f"Memory: {memory}")
         
@@ -38,6 +40,8 @@ if prompt := st.chat_input("What's up?"):
         
         stream = llm_client.response_generate(prompt, st.session_state.messages, memory, web_result)
         response = st.write_stream(stream)
+        
+        print(f"Response: {response}")
     st.session_state.messages.append({"role": "assistant", "content": response})
     
     # Update seesion in Pinecone