file upload gallery

app.py

@@ -929,6 +929,13 @@ The desired null value is also given. Populate the field with the null value of
         st.session_state['prompt_description'] = st.text_input("Enter description of prompt", value=st.session_state['prompt_info'].get('prompt_description', st.session_state['default_prompt_description']))
 
         
+        st.write('---')
+        # Input for new file name
+        st.header("Prompt Name")
+        st.write('Provide a name for your custom prompt. It can only conatin letters, numbers, and underscores. No spaces, dashes, or special characters.')
+        st.session_state['new_prompt_yaml_filename'] = st.text_input("Enter filename to save your prompt as a configuration YAML:", value=None, placeholder='my_prompt_name')
+
+
         st.write('---')
         st.header("Set LLM Model Type")
         # Define the options for the dropdown
@@ -1130,12 +1137,9 @@ The desired null value is also given. Populate the field with the null value of
 
         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
-            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 st.session_state['new_prompt_yaml_filename']:
@@ -1157,12 +1161,16 @@ The desired null value is also given. Populate the field with the null value of
             st.rerun()
 
     with col_prompt_main_right:
-        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']}"
+        if st.session_state['new_prompt_yaml_filename']:
+            st.session_state['show_prompt_name_e'] = f"Editing --> {st.session_state['selected_yaml_file']}"
+            st.session_state['show_prompt_name_w'] = f"Will be saved as --> {st.session_state['new_prompt_yaml_filename']}.yaml"
         else:
-            show_prompt_name = f"Editing: {st.session_state['selected_yaml_file']}"
-        st.subheader(f'Full Prompt ({show_prompt_name})')
+            st.session_state['show_prompt_name_e'] = f"Editing --> {st.session_state['selected_yaml_file']}"
+            st.session_state['show_prompt_name_w'] = f"Please set name"
+
+        st.subheader(f'Full Prompt')
+        st.write(st.session_state['show_prompt_name_e'])
+        st.write(st.session_state['show_prompt_name_w'])
         st.session_state['prompt_info'] = {
             'prompt_author': st.session_state['prompt_author'],
             'prompt_author_institution': st.session_state['prompt_author_institution'],
@@ -1670,6 +1678,10 @@ 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 'show_prompt_name_e' not in st.session_state:
+    st.session_state['show_prompt_name_e'] = None
+if 'show_prompt_name_w' not in st.session_state:
+    st.session_state['show_prompt_name_w'] = None
     
 
 # if not st.session_state.private_file: