file upload gallery

app.py

@@ -24,6 +24,8 @@ COLORS_EXPENSE_REPORT = {
         'GPT_3_5': '#006400',  # Dark Green
         'PALM2': '#66a8ff'     # blue
     }
+MAX_GALLERY_IMAGES = 50
+GALLERY_IMAGE_SIZE = 128
 
 class ProgressReport:
     def __init__(self, overall_bar, batch_bar, text_overall, text_batch):
@@ -755,6 +757,21 @@ def btn_load_prompt(selected_yaml_file, dir_prompt):
             'LLM': st.session_state['LLM']
         }
 
+def upload_local_prompt_to_server(dir_prompt):
+    uploaded_file = st.file_uploader("Upload a custom prompt file", type=['yaml'])
+    if uploaded_file is not None:
+        # Check the file extension
+        file_name = uploaded_file.name
+        if file_name.endswith('.yaml'):
+            file_path = os.path.join(dir_prompt, file_name)
+            
+            # Save the file
+            with open(file_path, 'wb') as f:
+                f.write(uploaded_file.getbuffer())
+            st.success(f"Saved file {file_name} in {dir_prompt}")
+        else:
+            st.error("Please upload a .yaml file that you previously created using this Prompt Builder tool.")
+
 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.
@@ -800,13 +817,16 @@ The desired null value is also given. Populate the field with the null value of
         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:
-        # Dropdown for selecting a YAML file
+            # 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:
             st.write('##')
             # Button to load the selected prompt
             st.button('Load Prompt', on_click=btn_load_prompt, args=[selected_yaml_file, dir_prompt])
-                
+
 
 
         # Define the options for the dropdown
@@ -1211,7 +1231,7 @@ def content_tab_settings():
                 st.session_state['input_list'].append(file_path)
 
                 img = Image.open(file_path)
-                img.thumbnail((64, 64), Image.Resampling.LANCZOS)  
+                img.thumbnail((GALLERY_IMAGE_SIZE, GALLERY_IMAGE_SIZE), Image.Resampling.LANCZOS)  
                 file_path_small = save_uploaded_file(st.session_state['dir_uploaded_images_small'], uploaded_file, img)
                 st.session_state['input_list_small'].append(file_path_small)
                 print(uploaded_file.name)
@@ -1219,9 +1239,9 @@ def content_tab_settings():
     with col_local_2:
         if st.session_state['input_list_small']:
             st.subheader('Image Gallery')
-            if len(st.session_state['input_list_small']) > 100:
+            if len(st.session_state['input_list_small']) > MAX_GALLERY_IMAGES:
                 # Only take the first 100 images from the list
-                images_to_display = st.session_state['input_list_small'][:100]
+                images_to_display = st.session_state['input_list_small'][:MAX_GALLERY_IMAGES]
             else:
                 # If there are less than 100 images, take them all
                 images_to_display = st.session_state['input_list_small']