Update app.py

app.py

@@ -317,12 +317,34 @@ def submit():
             payment_details.append(f"{name}({student_id}): ${cost}")
         payment_details_str = " | ".join(payment_details)
 
-        # Upload file to Drive
+        # Initialize Drive service
         drive_service = build('drive', 'v3', credentials=CREDENTIALS)
         
+        # Check if subfolder exists, if not create it
+        subfolder_name = sheet_name
+        subfolder_query = f"name = '{subfolder_name}' and '{FOLDER_ID}' in parents and mimeType = 'application/vnd.google-apps.folder' and trashed = false"
+        results = drive_service.files().list(q=subfolder_query).execute()
+        items = results.get('files', [])
+        
+        if not items:
+            # Create new subfolder
+            folder_metadata = {
+                'name': subfolder_name,
+                'mimeType': 'application/vnd.google-apps.folder',
+                'parents': [FOLDER_ID]
+            }
+            subfolder = drive_service.files().create(
+                body=folder_metadata,
+                fields='id'
+            ).execute()
+            subfolder_id = subfolder.get('id')
+        else:
+            subfolder_id = items[0]['id']
+
+        # Upload file to the subfolder
         file_metadata = {
             'name': f"{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}_{file.filename}",
-            'parents': [FOLDER_ID]
+            'parents': [subfolder_id]  # Use subfolder ID instead of main folder ID
         }
         
         media = MediaIoBaseUpload(