Merge pull request #111 from vanderbilt-data-science/issues-10567

app.py

@@ -58,7 +58,7 @@ if authentication_status:
     st.write(f'Welcome *{name}*')
 
     # Sidebar for navigation
-    page = st.sidebar.radio("Choose a page", ["Account Setup", "OpenAI API Key Setup", "Respond to Post"])
+    page = st.sidebar.radio("Choose a page", ["Account Setup", "API Key Help", "Respond to Post"])
 
     # Fetch user data from the database
     user_data = get_user_data(username)
@@ -77,15 +77,15 @@ if authentication_status:
         encrypted_api_key = str(encrypt(api_input, fernet))
         st.session_state.api_key = api_input
         
-        principles = st.text_input("My Principles (Paste Principles here)", placeholder = "Enter the main principles of your life you wish this response to uphold", value=user_data["principles"] if user_data and "principles" in user_data else "")
-        writing_style = st.text_input("My Writing Style (Paste Examples)", placeholder = "Provide examples of your writing style here", value=user_data["writing_style"] if user_data and "writing_style" in user_data else "")
+        principles = st.text_area("My Principles (Paste Principles here)", height = 100, placeholder = "Enter the main principles of your life you wish this response to uphold", value=user_data["principles"] if user_data and "principles" in user_data else "")
+        writing_style = st.text_area("My Writing Style (Paste Examples)", height = 100, placeholder = "Provide examples of your writing style here", value=user_data["writing_style"] if user_data and "writing_style" in user_data else "")
         #sources = st.text_area("Sources (This autopopulates for your reference)", value=st.session_state.sources if 'sources' in st.session_state else '', key = 'sources_key', height = 100)
     
         # Update button
         if st.button("Save Changes"):
             db.put({"key": username, "principles": principles, "writing_style": writing_style, "api_key": encrypted_api_key})
         
-    if page == "OpenAI API Key Setup":
+    if page == "API Key Help":
         st.title("OpenAI API Key Setup")
 
         st.header('What is an API key?')
@@ -117,10 +117,11 @@ if authentication_status:
         # Input boxes
 
         with right_col: 
-            background_info = st.text_area("Background information on original post (This autopopulates for your reference)", height = 700, value=st.session_state.background_info if 'background_info' in st.session_state else '', key = 'background_info_key')
+            background_info = st.text_area("Background information on original post (This autopopulates for your reference)", height = 780, value=st.session_state.background_info if 'background_info' in st.session_state else '', key = 'background_info_key')
 
         with left_col:
             original_post = st.text_area("Paste Original Post Here \n", height=100)
+            word_limit = st.text_input("Word Limit for Response", placeholder = "Please provide a word limit for the response")
             
             chat_mdl = None
             draft_response = ''
@@ -136,7 +137,7 @@ if authentication_status:
         
                 if chat_mdl is not None:
                     if user_data is None:
-                        draft_response, background_text = generate_custom_response(original_post, chat_mdl, "", "")
+                        draft_response, background_text = generate_custom_response(original_post, chat_mdl, "", "", word_limit)
                         st.session_state.draft_response = draft_response.content
                         st.session_state.background_text = background_text
                         #st.session_state.sources_text = sources_text
@@ -144,7 +145,7 @@ if authentication_status:
                         #st.session_state.sources = sources_text
                         st.rerun()
                     else:
-                        draft_response, background_text = generate_custom_response(original_post, chat_mdl, user_data['principles'], user_data['writing_style'])
+                        draft_response, background_text = generate_custom_response(original_post, chat_mdl, user_data['principles'], user_data['writing_style'], word_limit)
                         st.session_state.draft_response = draft_response.content
                         st.session_state.background_text = background_text
                         #st.session_state.sources_text = sources_text