Update tsadropboxretrieval.py

tsadropboxretrieval.py

@@ -22,7 +22,7 @@ import fitz
 import io
 import re
 import pyarrow
-
+from io import BytesIO
 
 """### NEW CODE - OCTOBER 26 - Marthe"""
 
@@ -101,11 +101,19 @@ def dropbox_upload_file(df, flag=0):
     return dbxTeam
 
 
-def uploadanyFile(doc,pdfname,path):
+def uploadanyFile(doc,pdfname,path,flag=0):
     try:
         dbxTeam= ADR_Access_DropboxTeam('admin')
-        path=path+pdfname
-        meta=dbxTeam.files_upload( doc.write() ,path,mode=dropbox.files.WriteMode("overwrite"))
+        
+        if flag: #tree = doc
+          pdfname=str(pdfname).split('.pdf')[0]+'.xml'
+          path=path+pdfname
+          f = BytesIO()
+          doc.write(f, encoding='utf-8', xml_declaration=True) 
+          meta=dbxTeam.files_upload(f.getvalue() ,path,mode=dropbox.files.WriteMode("overwrite"))
+        else:
+           path=path+pdfname
+           meta=dbxTeam.files_upload( doc.write() ,path,mode=dropbox.files.WriteMode("overwrite"))
         try:
           shared_link_metadata = dbxTeam.sharing_create_shared_link_with_settings(path)
         except: