Merge pull request #164 from vanderbilt-data-science/save-response-db-patch

app.py

@@ -225,20 +225,24 @@ if authentication_status:
                 key='draft_response_key'
             )
 
-            # Assessment of the response
+
+            # Add option to save responses
+            if st.button("Save This Response", help="Save this response to your writing style."):
+                #user_data["writing_style"] += st.session_state.draft_response
+                writing_style = user_data["writing_style"] + "\n" + st.session_state.draft_response
+                principles = user_data["principles"] if user_data and "principles" in user_data else ""
+                encrypted_api_key = str(encrypt(st.session_state.api_key, fernet))
+                
+                db.put({"key": username,  "writing_style": writing_style, "principles": principles,"api_key": encrypted_api_key})
+                st.write("Response saved to your writing style.")
+
+                            # Assessment of the response
             response_assessment = st.text_area(
                 label="Assessment of the Response",
                 value=st.session_state.response_assessment if 'response_assessment' in st.session_state else '',
                 height=350,
                 key='response_assessment_key'
             )
-            # Add option to save responses
-            if st.button("Save This Response"):
-                #user_data["writing_style"] += st.session_state.draft_response
-                writing_style = user_data["writing_style"] + "\n" + st.session_state.draft_response
-                
-                db.put({"key": username,  "writing_style": writing_style})
-                st.write("Response saved to your writing style.")
 
             # Initialization of the regeneration flag
             if 'is_regenerating' not in st.session_state: