adding logging in a different way

app.py

@@ -4,7 +4,9 @@ from swarm import Swarm, Agent
 from datasets import Dataset
 from datetime import datetime
 import pandas as pd
+from datetime import datetime
 import os
+
 hf_token = os.environ.get("HF_TOKEN")
 # Initialize OpenAI client outside of any function to use the same instance throughout the app
 openai_client = OpenAI(
@@ -46,7 +48,7 @@ st.title("Market Strategist")
 if "messages" not in st.session_state:
     st.session_state.messages = []
 if "dataset" not in st.session_state:
-    st.session_state.dataset = Dataset.from_dict({"role": [], "content": []})
+    st.session_state.dataset = Dataset.from_dict({"conversation": [], "timestamp": []})
 
 # Display all previous messages
 for message in st.session_state.messages:
@@ -57,12 +59,18 @@ for message in st.session_state.messages:
 if prompt := st.chat_input("Type your message (or 'exit' to clear history):"):
     # If user types 'exit', clear the history
     if prompt.strip().lower() == "exit":
-        if len(st.session_state.dataset) > 0:
-            dataset_name = "gl-kp/market-strategist-logging"  # Replace with your dataset name
-            st.session_state.dataset.push_to_hub(
-                dataset_name,
-                token=hf_token,
-                private=True
+        conversation_str = "\n".join(
+            [f"{msg['role']}: {msg['content']}" for msg in st.session_state.messages]
+        )
+        st.session_state.dataset = st.session_state.dataset.add_item({
+            "conversation": conversation_str,
+            "timestamp": datetime.now().isoformat()
+        })
+        dataset_name = "gl-kp/market-strategist-logging"  # Replace with your dataset name
+        st.session_state.dataset.push_to_hub(
+            dataset_name,
+            token=hf_token,
+            private=True
             )
         st.session_state.messages = []
         st.session_state.clear()
@@ -70,10 +78,6 @@ if prompt := st.chat_input("Type your message (or 'exit' to clear history):"):
     else:
         # Append user message to session state
         st.session_state.messages.append({"role": "user", "content": prompt})
-        st.session_state.dataset = st.session_state.dataset.add_item({
-            "role": "user",
-            "content": prompt
-        })
         
         with st.chat_message("user"):
             st.markdown(prompt)
@@ -96,9 +100,6 @@ if prompt := st.chat_input("Type your message (or 'exit' to clear history):"):
             # Display assistant response and append to history
             st.markdown(business_report)
             st.session_state.messages.append({"role": "assistant", "content": business_report})
-            st.session_state.dataset = st.session_state.dataset.add_item({
-                "role": "assistant",
-                "content": business_report
-            })
+