file upload gallery

app.py

@@ -684,11 +684,22 @@ def load_prompt_yaml(filename):
         st.session_state['prompt_author'] = st.session_state['prompt_info'].get('prompt_author', st.session_state['default_prompt_author']) 
         st.session_state['prompt_author_institution'] = st.session_state['prompt_info'].get('prompt_author_institution', st.session_state['default_prompt_author_institution']) 
         st.session_state['prompt_description'] = st.session_state['prompt_info'].get('prompt_description', st.session_state['default_prompt_description']) 
+        st.session_state['LLM'] = st.session_state['prompt_info'].get('LLM', 'gpt')
         st.session_state['instructions'] = st.session_state['prompt_info'].get('instructions', st.session_state['default_instructions']) 
         st.session_state['json_formatting_instructions'] = st.session_state['prompt_info'].get('json_formatting_instructions', st.session_state['default_json_formatting_instructions'] )
         st.session_state['rules'] = st.session_state['prompt_info'].get('rules', {})
         st.session_state['mapping'] = st.session_state['prompt_info'].get('mapping', {})
-        st.session_state['LLM'] = st.session_state['prompt_info'].get('LLM', 'gpt')
+
+        st.session_state['prompt_info'] = {
+            'prompt_author': st.session_state['prompt_author'],
+            'prompt_author_institution': st.session_state['prompt_author_institution'],
+            'prompt_description': st.session_state['prompt_description'],
+            'LLM': st.session_state['LLM'],
+            'instructions': st.session_state['instructions'],
+            'json_formatting_instructions': st.session_state['json_formatting_instructions'],
+            'rules': st.session_state['rules'],
+            'mapping': st.session_state['mapping'],
+        }
 
         # Placeholder:
         st.session_state['assigned_columns'] = list(chain.from_iterable(st.session_state['mapping'].values())) 
@@ -798,11 +809,11 @@ def btn_load_prompt(selected_yaml_file, dir_prompt):
             'prompt_author': st.session_state['prompt_author'],
             'prompt_author_institution': st.session_state['prompt_author_institution'],
             'prompt_description': st.session_state['prompt_description'],
+            'LLM': st.session_state['LLM'],
             'instructions': st.session_state['instructions'],
             'json_formatting_instructions': st.session_state['json_formatting_instructions'],
             'rules': st.session_state['rules'],
             'mapping': st.session_state['mapping'],
-            'LLM': st.session_state['LLM']
         }