Update components/historical_personalities.py

components/historical_personalities.py

@@ -5,6 +5,11 @@ def show_historical_personalities():
     st.title("Interact with Historical Personalities")
     st.write("Engage in dynamic, educational dialogues with AI-generated historical figures.")
     
+    # Initialize the conversation history in session state
+    if 'conversation_history' not in st.session_state:
+        st.session_state.conversation_history = []
+
+    # User selects a personality and enters a question
     personality = st.selectbox("Choose a historical personality", ["Albert Einstein", "William Shakespeare", "Cleopatra", "Leonardo da Vinci", "Marie Curie"])
     user_input = st.text_input("Ask a question or start a conversation:")
     
@@ -12,14 +17,22 @@ def show_historical_personalities():
         if user_input:
             with st.spinner(f"Generating response from {personality}..."):
                 response = generate_historical_dialogue(personality, user_input)
-            st.write(f"{personality}: {response}")
+
+                # Update conversation history
+                st.session_state.conversation_history.append(f"You: {user_input}")
+                st.session_state.conversation_history.append(f"{personality}: {response}")
+                
+                # Display the response
+                st.write(f"{personality}: {response}")
         else:
             st.warning("Please enter a question or statement to start the conversation.")
     
-    if 'conversation_history' not in st.session_state:
-        st.session_state.conversation_history = []
-    
+    # Display the conversation history
     if len(st.session_state.conversation_history) > 0:
         st.subheader("Conversation History")
         for entry in st.session_state.conversation_history:
             st.text(entry)
+
+# Run the function to start the Streamlit app
+if __name__ == "__main__":
+    show_historical_personalities()