Merge pull request #134 from vanderbilt-data-science/122-save-drafts

app.py

@@ -212,6 +212,14 @@ if authentication_status:
                 key='draft_response_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:
                 st.session_state.is_regenerating = False