Merge pull request #113 from vanderbilt-data-science/issue-90

app.py

@@ -10,14 +10,14 @@ from free_speech_app.DataLoadDb import *
 from free_speech_app.FreeSpeechPromptsResponses import *
 from langchain.chat_models import ChatOpenAI
 
-#connect to/create Deta user database
+# connect to/create Deta user database
 deta = Deta(st.secrets["deta_key"])
 db = deta.Base("user_data")
 
-#fernet key (generated locally, stored in streamlit secrets)
-fernet = Fernet(bytes(st.secrets["fernet_key"],'utf-8'))
+# fernet key (generated locally, stored in streamlit secrets)
+fernet = Fernet(bytes(st.secrets["fernet_key"], 'utf-8'))
 
-#activeloop_token
+# activeloop_token
 os.environ["ACTIVELOOP_TOKEN"] = st.secrets["deeplake_key"]
 
 config_drive = deta.Drive("passwords")
@@ -34,6 +34,7 @@ authenticator = stauth.Authenticate(
     config['preauthorized']
 )
 
+
 def get_user_data(user):
     data = db.fetch().items
     for person in data:
@@ -41,14 +42,24 @@ def get_user_data(user):
             return person
     return None
 
+
 def encrypt(api_key: str, fernet) -> bytes:
     """Encrypt the API key."""
     return fernet.encrypt(api_key.encode())
 
+
 def decrypt(encrypted_api_key: bytes, fernet) -> str:
     """Decrypt the encrypted API key."""
     return fernet.decrypt(encrypted_api_key).decode()
 
+
+def add_logos():
+    st.markdown(
+        '![image](free_speech_app/logos/Future-of-Free-Speech-logo.png)')
+    st.markdown(
+        '![image](free_speech_app/logos/Vanderbilt-University-Logo.png)')
+
+
 # Render the login module
 name, authentication_status, username = authenticator.login('Login', 'main')
 
@@ -60,23 +71,25 @@ if authentication_status:
     # Sidebar for navigation
     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)
 
     if page == "Account Setup":
+        add_logos()
 
         st.title("Account Setup")
         st.markdown("Please use this page to provide your OpenAI API Key, Principles and Writing Style. **Please make sure to press the Save Changes button after providing the information.**")
-        
-    
+
         # Input boxes with existing data
-    
+
         if 'api_key' not in st.session_state:
             st.session_state.api_key = ""
-        api_input = st.text_input("Paste OpenAI API Key", value=decrypt(user_data["api_key"].encode()[2:-1], fernet) if user_data and "api_key" in user_data else "", type="password")
+        api_input = st.text_input("Paste OpenAI API Key", value=decrypt(user_data["api_key"].encode()[
+                                  2:-1], fernet) if user_data and "api_key" in user_data else "", type="password")
         encrypted_api_key = str(encrypt(api_input, fernet))
         st.session_state.api_key = api_input
-        
+
         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)
@@ -86,6 +99,8 @@ if authentication_status:
             db.put({"key": username, "principles": principles, "writing_style": writing_style, "api_key": encrypted_api_key})
         
     if page == "API Key Help":
+        add_logos()
+
         st.title("OpenAI API Key Setup")
 
         st.header('What is an API key?')
@@ -96,10 +111,11 @@ if authentication_status:
 
         image = 'apikeyex.png'
         st.header('How to get an OpenAI API key:')
-        st.write('1. Go to https://platform.openai.com/account/api-keys') 
+        st.write('1. Go to https://platform.openai.com/account/api-keys')
         st.write('2. Log in or create an OpenAI account if you do not have one')
         st.write('3. Click "Create new secret key" and give your key a name')
-        st.image(image, caption=None, width=None, use_column_width=None, clamp=False, channels="RGB", output_format="auto")
+        st.image(image, caption=None, width=None, use_column_width=None,
+                 clamp=False, channels="RGB", output_format="auto")
         st.write('4. Copy the generated API key and keep it private like a password')
 
         st.header('Using your API key')
@@ -108,14 +124,15 @@ if authentication_status:
 
         st.warning('Treat your API key like a secret! Do not share it publicly.')
 
-
     elif page == "Respond to Post":
+        add_logos()
         st.title("Respond to Post")
-        
+
         left_col, right_col = st.columns(2)
-        
+
         # Input boxes
 
+
         with right_col: 
             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')
 
@@ -125,7 +142,7 @@ if authentication_status:
             
             chat_mdl = None
             draft_response = ''
-    
+
             # Check if the "Submit" button is clicked
             if st.button("Submit"):
                 if st.session_state.api_key:
@@ -134,32 +151,35 @@ if authentication_status:
                     if (os.environ["OPENAI_API_KEY"] == st.secrets["secret_passphrase"]):
                         os.environ["OPENAI_API_KEY"] = st.secrets["dsi_openai_key"]
                     chat_mdl = ChatOpenAI(model_name='gpt-4', temperature=0.1)
-        
+
                 if chat_mdl is not None:
                     if user_data is None:
+
                         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
+                        # st.session_state.sources_text = sources_text
                         st.session_state.background_info = background_text
-                        #st.session_state.sources = sources_text
+                        # 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'], word_limit)
+
                         st.session_state.draft_response = draft_response.content
                         st.session_state.background_text = background_text
-                        #st.session_state.sources_text = sources_text
+                        # st.session_state.sources_text = sources_text
                         st.session_state.background_info = background_text
-                        #st.session_state.sources = sources_text
+                        # st.session_state.sources = sources_text
                         st.rerun()
-                        
-            
+
             # Ensure session state variables are initialized
             if 'draft_response' not in st.session_state:
                 st.session_state.draft_response = ''
             if 'regenerate_prompt' not in st.session_state:
                 st.session_state.regenerate_prompt = ''
-            
+
             # Output from function
             response_textarea = st.text_area(
                 label="Draft Response. Please edit here or prompt suggestions in the box below.",
@@ -167,11 +187,11 @@ if authentication_status:
                 height=350,
                 key='draft_response_key'
             )
-                        
+
             # Initialization of the regeneration flag
             if 'is_regenerating' not in st.session_state:
                 st.session_state.is_regenerating = False
-            
+
             # Check if the app is in the "regeneration" phase
             if st.session_state.is_regenerating:
                 # Display the regenerated response explicitly
@@ -199,23 +219,27 @@ if authentication_status:
                         os.environ['OPENAI_API_KEY'] = st.session_state.api_key
                         # add condition to check for passphrase to allow use of DSI api key stored in secrets
                         if (os.environ["OPENAI_API_KEY"] == st.secrets["secret_passphrase"]):
-                            os.environ["OPENAI_API_KEY"] = st.secrets["dsi_openai_key"] #umang key
+                            # umang key
+                            os.environ["OPENAI_API_KEY"] = st.secrets["dsi_openai_key"]
                         elif (os.environ["OPENAI_API_KEY"] == st.secrets["secret_passphrase2"]):
-                            os.environ["OPENAI_API_KEY"] = st.secrets["dsi_openai_key2"] #abbie key
-                        chat_mdl = ChatOpenAI(model_name='gpt-4', temperature=0.1)
-                    
+                            # abbie key
+                            os.environ["OPENAI_API_KEY"] = st.secrets["dsi_openai_key2"]
+                        chat_mdl = ChatOpenAI(
+                            model_name='gpt-4', temperature=0.1)
+
                     if chat_mdl is not None:
-                        updated_response = regenerate_custom_response(chat_mdl, regenerate_prompt, st.session_state.draft_response).content    
+                        updated_response = regenerate_custom_response(
+                            chat_mdl, regenerate_prompt, st.session_state.draft_response).content
                         st.session_state.draft_response = updated_response
                         st.session_state.is_regenerating = True
 
                     st.rerun()
-                        
-                
+
+
 elif authentication_status is False:
     st.error('Username/password is incorrect')
-    #added registration module if username/password is incorrect
-    
+    # added registration module if username/password is incorrect
+
     try:
         if authenticator.register_user('New User Registration', preauthorization=False):
             st.success('User Registered Successfully! Please log in above.')
@@ -224,7 +248,7 @@ elif authentication_status is False:
 
 elif authentication_status is None:
     st.warning('Please enter your username and password')
-    
+
     try:
         if authenticator.register_user('New User Registration', preauthorization=False):
             st.success('User Registered Successfully! Please log in above.')
@@ -234,4 +258,4 @@ elif authentication_status is None:
 with open('config.yaml', 'w') as file:
     yaml.dump(config, file, default_flow_style=False)
 
-config_drive.put("config.yaml", path = "config.yaml")
+config_drive.put("config.yaml", path="config.yaml")