file upload gallery

app.py

@@ -679,6 +679,7 @@ def save_changes_to_API_keys(cfg_private,openai_api_key,azure_openai_api_version
 
 # Function to load a YAML file and update session_state
 def load_prompt_yaml(filename):
+    st.session_state['user_clicked_load_prompt_yaml'] = filename
     with open(filename, 'r') as file:
         st.session_state['prompt_info'] = yaml.safe_load(file)
         st.session_state['prompt_author'] = st.session_state['prompt_info'].get('prompt_author', st.session_state['default_prompt_author']) 
@@ -1161,12 +1162,18 @@ The desired null value is also given. Populate the field with the null value of
             st.rerun()
 
     with col_prompt_main_right:
-        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"
+        if st.session_state['user_clicked_load_prompt_yaml'] is None: # see if user has loaded a yaml to edit
+            st.session_state['show_prompt_name_e'] = f"Prompt: building prompt from scratch"
+            if st.session_state['new_prompt_yaml_filename']:
+                st.session_state['show_prompt_name_w'] = f"Prompt Saved As: {st.session_state['new_prompt_yaml_filename']}.yaml"
+            else:
+                st.session_state['show_prompt_name_w'] = f"Prompt Saved As:  PLEASE SET NAME"
         else:
-            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.session_state['show_prompt_name_e'] = f"Prompt: editing --> {st.session_state['selected_yaml_file']}"
+            if st.session_state['new_prompt_yaml_filename']:
+                st.session_state['show_prompt_name_w'] = f"Prompt Saved As: {st.session_state['new_prompt_yaml_filename']}.yaml"
+            else:
+                st.session_state['show_prompt_name_w'] = f"Prompt Saved As: PLEASE SET NAME"
 
         st.subheader(f'Full Prompt')
         st.write(st.session_state['show_prompt_name_e'])
@@ -1682,6 +1689,8 @@ 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 'user_clicked_load_prompt_yaml' not in st.session_state:
+    st.session_state['user_clicked_load_prompt_yaml'] = None
     
 
 # if not st.session_state.private_file: