file upload gallery

app.py

@@ -772,6 +772,16 @@ 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):
+    file_label = f"Download {selected_yaml_file}"
+    with open(file_path, 'rb') as f:
+        st.download_button(
+            label=file_label,
+            data=f,
+            file_name=os.path.basename(file_path),
+            mime='application/x-yaml'
+        )
+    
 def build_LLM_prompt_config():
     st.session_state['assigned_columns'] = []
     st.session_state['default_instructions'] = """1. Refactor the unstructured OCR text into a dictionary based on the JSON structure outlined below.
@@ -815,18 +825,28 @@ The desired null value is also given. Populate the field with the null value of
 
         dir_prompt = os.path.join(st.session_state.dir_home, 'custom_prompts')
         yaml_files = [f for f in os.listdir(dir_prompt) if f.endswith('.yaml')]
-        col_load_text, col_load_btn = st.columns([8,2])
-        with col_load_text:
+        col_upload_yaml, col_upload_btn = st.columns([8,2])
+        with col_upload_yaml:
             # Upload a prompt from your computer
             upload_local_prompt_to_server(dir_prompt)
 
-            # Dropdown for selecting a YAML file
-            selected_yaml_file = st.selectbox('Select a prompt YAML file to load:', [''] + yaml_files)
-        with col_load_btn:
+            
+        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])
 
+        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)
+
+        with col_download_btn:
+            if selected_yaml_file:
+                # Construct the full path to the file
+                download_file_path = os.path.join(dir_prompt, selected_yaml_file)
+                # Create the download button
+                create_download_button(download_file_path, selected_yaml_file)
 
 
         # Define the options for the dropdown

vouchervision/prompt_catalog.py

@@ -645,7 +645,7 @@ class PromptCatalog:
             }
 
             # Convert the structure to a JSON string without indentation
-            structure_json_str = json.dumps(dictionary_structure, default_flow_style=False, sort_keys=False)
+            structure_json_str = json.dumps(dictionary_structure, sort_keys=False)
             return structure_json_str
         
         else:
@@ -665,7 +665,7 @@ class PromptCatalog:
             }
 
             # Convert the structure to a JSON string without indentation
-            structure_json_str = json.dumps(full_structure, default_flow_style=False, sort_keys=False)
+            structure_json_str = json.dumps(full_structure, sort_keys=False)
             return structure_json_str
     
     def create_structure(self, is_palm=False):
@@ -674,7 +674,7 @@ class PromptCatalog:
             dictionary_structure = {key: "" for key in self.rules_list['Dictionary'].keys()}
 
             # Convert the structure to a JSON string with indentation for readability
-            structure_json_str = json.dumps(dictionary_structure, default_flow_style=False, sort_keys=False)
+            structure_json_str = json.dumps(dictionary_structure, sort_keys=False)
             return structure_json_str
         else:
             # Start with an empty structure for the "Dictionary" section
@@ -690,7 +690,7 @@ class PromptCatalog:
             }
 
             # Convert the structure to a JSON string with indentation for readability
-            structure_json_str = json.dumps(full_structure, default_flow_style=False, sort_keys=False)
+            structure_json_str = json.dumps(full_structure, sort_keys=False)
             return structure_json_str
 
     def generate_xlsx_headers(self, is_palm):