Update src/Surveyor.py

src/Surveyor.py

@@ -1471,6 +1471,8 @@ class Surveyor:
         self.survey_print_fn(cites)
 
         shutil.copytree('arxiv_data/', survey_dump_dir + '/arxiv_data/')
+        for d in [survey_pdf_dir, survey_txt_dir, survey_img_dir, survey_tab_dir]:
+            shutil.copytree(Path(d).stem, survey_dump_dir + f'/{Path(d).stem}/')
         assert (os.path.exists(survey_file))
         
         zip_name = 'arxiv_dumps_'+query.replace(' ', '_')+'.zip'
@@ -1479,4 +1481,4 @@ class Surveyor:
         self.print_fn("\n- Survey complete.. \nSurvey file path :" + str(survey_file) + 
                       "\nAll outputs zip path :" + str(zip_name))
 
-        return str(zip_name.resolve()), str(zip_name.resolve())
+        return str(zip_name.resolve()), str(survey_file.resolve())