file upload gallery

app.py

@@ -727,7 +727,7 @@ def save_prompt_yaml(filename, col_right_save):
     upload_to_drive(filepath, filename)
 
     with col_right_save:
-        create_download_button(filepath, filename)
+        create_download_button_yaml(filepath, filename)
 
 # Function to upload files to Google Drive
 def upload_to_drive(filepath, filename):
@@ -833,7 +833,7 @@ def upload_local_prompt_to_server(dir_prompt):
         else:
             st.error("Please upload a .yaml file that you previously created using this Prompt Builder tool.")
 
-def create_download_button(file_path, selected_yaml_file):
+def create_download_button_yaml(file_path, selected_yaml_file):
     file_label = f"Download {selected_yaml_file}"
     with open(file_path, 'rb') as f:
         st.download_button(
@@ -914,7 +914,7 @@ The desired null value is also given. Populate the field with the null value of
                 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'] )
+                create_download_button_yaml(download_file_path, st.session_state['selected_yaml_file'] )
         
 
         # Prompt Author Information
@@ -1135,13 +1135,13 @@ The desired null value is also given. Populate the field with the null value of
         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')
+            st.session_state['new_prompt_yaml_filename'] = st.text_input("Enter filename to save your prompt as a configuration YAML:",placeholder='my_prompt_name')
             # Button to save the new YAML file
             if st.button('Save YAML', type='primary'):
-                if new_filename:
+                if st.session_state['new_prompt_yaml_filename']:
                     if check_unique_mapping_assignments():
-                        if check_prompt_yaml_filename(new_filename):
-                            save_prompt_yaml(new_filename, col_right_save)
+                        if check_prompt_yaml_filename(st.session_state['new_prompt_yaml_filename']):
+                            save_prompt_yaml(st.session_state['new_prompt_yaml_filename'], col_right_save)
                         else:
                             st.error("File name can only contain letters, numbers, underscores, and dashes. Cannot contain spaces.")
                     else:
@@ -1157,7 +1157,12 @@ The desired null value is also given. Populate the field with the null value of
             st.rerun()
 
     with col_prompt_main_right:
-        st.subheader('All Prompt Components')
+        show_prompt_name = None
+        if st.session_state['new_prompt_yaml_filename'] is not None:
+            show_prompt_name = f"Will be saved as: {st.session_state['new_prompt_yaml_filename']}"
+        else:
+            show_prompt_name = f"Editing: {st.session_state['selected_yaml_file']}"
+        st.subheader(f'Full Prompt ({show_prompt_name})')
         st.session_state['prompt_info'] = {
             'prompt_author': st.session_state['prompt_author'],
             'prompt_author_institution': st.session_state['prompt_author_institution'],
@@ -1663,6 +1668,8 @@ if 'input_list_small' not in st.session_state:
     st.session_state['input_list_small'] = []
 if 'selected_yaml_file' not in st.session_state:
     st.session_state['selected_yaml_file'] = None
+if 'new_prompt_yaml_filename' not in st.session_state:
+    st.session_state['new_prompt_yaml_filename'] = None
     
 
 # if not st.session_state.private_file: