file upload gallery

app.py

@@ -688,7 +688,7 @@ def load_prompt_yaml(filename):
         # Placeholder:
         st.session_state['assigned_columns'] = list(chain.from_iterable(st.session_state['mapping'].values())) 
 
-def save_prompt_yaml(filename):
+def save_prompt_yaml(filename, col_right_save):
     yaml_content = {
         'instructions': st.session_state['instructions'],
         'json_formatting_instructions': st.session_state['json_formatting_instructions'],
@@ -705,6 +705,9 @@ def save_prompt_yaml(filename):
 
     st.success(f"Prompt saved as '{filename}.yaml'.")
 
+    with col_right_save:
+        create_download_button(filepath, filename)
+
 def check_unique_mapping_assignments():
     if len(st.session_state['assigned_columns']) != len(set(st.session_state['assigned_columns'])):
         st.error("Each column name must be assigned to only one category.")
@@ -829,33 +832,34 @@ The desired null value is also given. Populate the field with the null value of
 
         dir_prompt = os.path.join(st.session_state.dir_home, 'custom_prompts')
         yaml_files = [f for f in os.listdir(dir_prompt) if f.endswith('.yaml')]
-        col_upload_yaml, col_upload_btn = st.columns([8,2])
+        col_upload_yaml, col_upload_yaml_2 = st.columns([4,4])
         with col_upload_yaml:
             # Upload a prompt from your computer
             upload_local_prompt_to_server(dir_prompt)
 
-        with col_upload_btn:
-            st.write('##')
-            # Button to load the selected prompt
-            st.button('Load Prompt into Builder', on_click=btn_load_prompt, args=[st.session_state['selected_yaml_file'] , dir_prompt])
-
-        col_select_yaml, col_download_btn = st.columns([8,2])
+        col_select_yaml, col_upload_btn, col_download_btn = st.columns([6,2,2])
         with col_select_yaml:
             # Dropdown for selecting a YAML file
             st.session_state['selected_yaml_file'] = st.selectbox('Select a prompt .YAML file to load:', [''] + yaml_files)
 
+        with col_upload_btn:
+            st.write('##')
+            # Button to load the selected prompt
+            st.button('Load Selected Prompt into Builder', on_click=btn_load_prompt, args=[st.session_state['selected_yaml_file'] , dir_prompt])
+
         with col_download_btn:
             if st.session_state['selected_yaml_file']:
                 # Construct the full path to the file
                 download_file_path = os.path.join(dir_prompt, st.session_state['selected_yaml_file'] )
                 # Create the download button
+                st.write('##')
                 create_download_button(download_file_path, st.session_state['selected_yaml_file'] )
 
 
         # Define the options for the dropdown
         llm_options = ['gpt', 'palm']
         # Create the dropdown and set the value to session_state['LLM']
-        st.session_state['LLM'] = st.selectbox('Set LLM:', llm_options, index=llm_options.index(st.session_state.get('LLM', 'gpt')))
+        st.session_state['LLM'] = st.selectbox('Set LLM type:', llm_options, index=llm_options.index(st.session_state.get('LLM', 'gpt')))
 
         
 
@@ -1048,7 +1052,11 @@ The desired null value is also given. Populate the field with the null value of
             st.json(st.session_state['mapping'])
 
 
-        col_left_save, col_right_save = st.columns([6,4])
+        st.write('---')
+        st.header("Save and Download Custom Prompt")
+        st.write('Provide a name for your custom prompt. It can only conatin letters, numbers, and underscores.')
+        st.write('Once you click save, validation checks will verify the formatting and then a download button will appear so that you can ***save a local copy of your custom prompt.***')
+        col_left_save, col_right_save, _ = st.columns([2,2,8])
         with col_left_save:
             # Input for new file name
             new_filename = st.text_input("Enter filename to save your prompt as a configuration YAML:",placeholder='my_prompt_name')
@@ -1057,7 +1065,7 @@ The desired null value is also given. Populate the field with the null value of
                 if new_filename:
                     if check_unique_mapping_assignments():
                         if check_prompt_yaml_filename(new_filename):
-                            save_prompt_yaml(new_filename)
+                            save_prompt_yaml(new_filename, col_right_save)
                         else:
                             st.error("File name can only contain letters, numbers, underscores, and dashes. Cannot contain spaces.")
                     else:
@@ -1065,10 +1073,13 @@ The desired null value is also given. Populate the field with the null value of
                 else:
                     st.error("Please enter a filename.")
         
-            if st.button('Exit'):
-                st.session_state.proceed_to_build_llm_prompt = False
-                st.session_state.proceed_to_main = True
-                st.rerun()
+        st.write('---')
+        st.header("Return to VoucherVision")
+        if st.button('Exit'):
+            st.session_state.proceed_to_build_llm_prompt = False
+            st.session_state.proceed_to_main = True
+            st.rerun()
+
     with col_prompt_main_right:
         st.subheader('All Prompt Components')
         st.session_state['prompt_info'] = {
@@ -1518,6 +1529,7 @@ def main():
         if st.button("Build Custom LLM Prompt"):
             st.session_state.proceed_to_build_llm_prompt = True
             st.rerun()
+        st.write('When opening the Prompt Builder, it take a moment for the page to refresh.')
         
     with tab_component:
         content_tab_component()