file upload gallery

app.py

@@ -11,7 +11,9 @@ import pandas as pd
 from typing import Union
 from google.oauth2 import service_account
 from streamlit_extras.let_it_rain import rain
-from streamlit_image_select import image_select
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
+from googleapiclient.http import MediaFileUpload
 from vouchervision.LeafMachine2_Config_Builder import write_config_file
 from vouchervision.VoucherVision_Config_Builder import build_VV_config, run_demo_tests_GPT, run_demo_tests_Palm , TestOptionsGPT, TestOptionsPalm, check_if_usable, run_api_tests
 from vouchervision.vouchervision_main import voucher_vision, voucher_vision_OCR_test
@@ -704,10 +706,37 @@ def save_prompt_yaml(filename, col_right_save):
         yaml.safe_dump(yaml_content, file)
 
     st.success(f"Prompt saved as '{filename}.yaml'.")
+    
+    upload_to_drive(filepath, filename)
 
     with col_right_save:
         create_download_button(filepath, filename)
 
+# Function to upload files to Google Drive
+def upload_to_drive(filepath, filename):
+    # Parse the service account info from the environment variable
+    creds_info = json.loads(os.environ.get('GDRIVE_API'))
+    creds = service_account.Credentials.from_service_account_info(
+        creds_info, scopes=["https://www.googleapis.com/auth/drive"]
+    )
+    service = build('drive', 'v3', credentials=creds)
+
+    # Get the folder ID from the environment variable
+    folder_id = os.environ.get('GDRIVE_FOLDER_ID')
+
+    file_metadata = {
+        'name': filename,
+        'parents': [folder_id]
+    }
+
+    media = MediaFileUpload(filepath, mimetype='application/x-yaml')
+
+    service.files().create(
+        body=file_metadata,
+        media_body=media,
+        fields='id'
+    ).execute()
+
 def check_unique_mapping_assignments():
     if len(st.session_state['assigned_columns']) != len(set(st.session_state['assigned_columns'])):
         st.error("Each column name must be assigned to only one category.")

requirements.txt

@@ -21,6 +21,7 @@ openpyxl
 google-generativeai
 google-cloud-storage
 google-cloud-vision
+google-api-python-client
 opencv-python
 chromadb
 chroma-migrate
@@ -33,4 +34,4 @@ psutil
 py-cpuinfo
 azureml-sdk
 azure-identity
-streamlit-image-select
+