Update app.py

app.py

@@ -27,6 +27,50 @@ technologies_df = pd.DataFrame()
 technology_embeddings = None # Will store pre-computed embeddings for descriptions
 model = None
 
+
+###- GOOGLE DRIVE API
+
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
+from googleapiclient.http import MediaIoBaseDownload, MediaIoBaseUpload
+
+# Environment variables
+FOLDER_ID = os.getenv("FOLDER_ID")
+GOOGLE_CREDENTIALS = os.environ.get("GOOGLE_CREDENTIALS")
+
+def create_new_file_in_drive(username, dataframe_to_upload, credentials_json, folder_id):
+    """Crée un nouveau fichier CSV dans Google Drive à partir d'un DataFrame Pandas."""
+
+    creds_dict = json.loads(credentials_json)
+
+    # Charger les informations d'identification du compte de service
+    creds = service_account.Credentials.from_service_account_info(creds_dict)
+
+    # Construire le service API Drive
+    service = build('drive', 'v3', credentials=creds)
+
+    # Convertir le DataFrame en fichier CSV en mémoire
+    csv_buffer = io.BytesIO()
+    dataframe_to_upload.to_csv(csv_buffer, index=False, sep=';', encoding='utf-8')
+    csv_buffer.seek(0)
+
+    # Créer les métadonnées du fichier
+    filename = f"rating-results-{username}.csv"
+    file_metadata = {
+        'name': filename,
+        'parents': [folder_id]
+    }
+
+    # Télécharger le fichier CSV sur Google Drive
+    media = MediaIoBaseUpload(csv_buffer, mimetype='text/csv', resumable=True)
+    file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
+
+    print(f"File '{filename}' created successfully.")
+
+
+###-
+ 
+
 # --- Load Data and Model (Load once at startup) ---
 def load_data_and_model():
     global categories_data, category_names, category_embeddings