file upload gallery

app.py

@@ -815,6 +815,10 @@ The desired null value is also given. Populate the field with the null value of
         st.title("Custom LLM Prompt Builder")
         st.subheader('About')
         st.write("This form allows you to craft a prompt for your specific task.")
+        st.subheader('For Hugging Face Spaces')
+        st.write("If you create a prompt with the Hugging Face Spaces implementation of VoucherVision, make sure that you download the prompt immediately after you have 'Saved' the prompt. Default storage space on HF Spaces is not persistant, so if you refresh the page your prompt will probably disappear.")
+        st.write("You can submit your prompt using this link and we will add it to our library so it will always be available.")
+
         st.subheader('How it works')
         st.write("1. Edit this page until you are happy with your instructions. We recommend looking at the basic structure, writing down your prompt inforamtion in a Word document so that it does not randomly disappear, and then copying and pasting that info into this form once your whole prompt structure is defined.")
         st.write("2. After you enter all of your prompt instructions, click 'Save' and give your file a name.")
@@ -830,23 +834,22 @@ The desired null value is also given. Populate the field with the null value of
             # 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', on_click=btn_load_prompt, args=[selected_yaml_file, dir_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])
         with col_select_yaml:
             # Dropdown for selecting a YAML file
-            selected_yaml_file = st.selectbox('Select a prompt .YAML file to load:', [''] + yaml_files)
+            st.session_state['selected_yaml_file'] = st.selectbox('Select a prompt .YAML file to load:', [''] + yaml_files)
 
         with col_download_btn:
-            if selected_yaml_file:
+            if st.session_state['selected_yaml_file']:
                 # Construct the full path to the file
-                download_file_path = os.path.join(dir_prompt, selected_yaml_file)
+                download_file_path = os.path.join(dir_prompt, st.session_state['selected_yaml_file'] )
                 # Create the download button
-                create_download_button(download_file_path, selected_yaml_file)
+                create_download_button(download_file_path, st.session_state['selected_yaml_file'] )
 
 
         # Define the options for the dropdown
@@ -1567,6 +1570,8 @@ if 'input_list' not in st.session_state:
     st.session_state['input_list'] = []
 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 not st.session_state.private_file: