print fix, title model switch

app.py

@@ -42,6 +42,9 @@ def survey_space(surveyor):
 
 if __name__ == '__main__':
     st.container().title('Auto-Research V0.1 - Automated Survey generation from research keywords')
+    std_col, survey_col = st.columns(2)
+    std_col.header('execution log:')
+    survey_col.header('Generated_survey:')
     global surveyor
-    surveyor_obj = Surveyor(print_fn=st.write)
+    surveyor_obj = Surveyor(print_fn=std_col.write, survey_print_fn=survey_col.write)
     survey_space(surveyor_obj)

src/Surveyor.py

@@ -45,7 +45,8 @@ class Surveyor:
             high_gpu=False,
             refresh_models=False,
             no_save_models=False,
-            print_fn=None
+            print_fn=None,
+            survey_print_fn=None
     ):
         '''
         Initializes models and directory structure for the surveyor
@@ -75,6 +76,9 @@ class Surveyor:
         self.print_fn = print
         if print_fn is not None:
             self.print_fn = print_fn
+        self.survey_print_fn = self.print_fn
+        if survey_print_fn is not None:
+            self.survey_print_fn = survey_print_fn
 
         self.torch_device = 'cpu'
         self.print_fn("\nTorch_device: " + self.torch_device)
@@ -342,82 +346,82 @@ class Surveyor:
             query = 'Internal(existing) research'
         file.write("----------------------------------------------------------------------")
         file.write("Title: A survey on " + query)
-        self.print_fn("")
-        self.print_fn("----------------------------------------------------------------------")
-        self.print_fn("Title: A survey on " + query)
+        self.survey_print_fn("")
+        self.survey_print_fn("----------------------------------------------------------------------")
+        self.survey_print_fn("Title: A survey on " + query)
         file.write("Author: Auto-Research (github.com/sidphbot/Auto-Research)")
-        self.print_fn("Author: Auto-Research (github.com/sidphbot/Auto-Research)")
+        self.survey_print_fn("Author: Auto-Research (github.com/sidphbot/Auto-Research)")
         file.write("Dev: Auto-Research (github.com/sidphbot/Auto-Research)")
-        self.print_fn("Dev: Auto-Research (github.com/sidphbot/Auto-Research)")
+        self.survey_print_fn("Dev: Auto-Research (github.com/sidphbot/Auto-Research)")
         file.write("Disclaimer: This survey is intended to be a research starter. This Survey is Machine-Summarized, "+
                    "\nhence some sentences might be wrangled or grammatically incorrect. However all sentences are "+
                    "\nmined with proper citations. As All of the text is practically quoted texted, hence to "+
                    "\nimprove visibility, all the papers are duly cited in the Bibiliography section. as bibtex "+
                    "\nentries(only to avoid LaTex overhead). ")
-        self.print_fn("Disclaimer: This survey is intended to be a research starter. This Survey is Machine-Summarized, "+
+        self.survey_print_fn("Disclaimer: This survey is intended to be a research starter. This Survey is Machine-Summarized, "+
                 "\nhence some sentences might be wrangled or grammatically incorrect. However all sentences are "+
                 "\nmined with proper citations. As All of the text is practically quoted texted, hence to "+
                 "\nimprove visibility, all the papers are duly cited in the Bibiliography section. as bibtex "+
                 "\nentries(only to avoid LaTex overhead). ")
         file.write("----------------------------------------------------------------------")
-        self.print_fn("----------------------------------------------------------------------")
+        self.survey_print_fn("----------------------------------------------------------------------")
         file.write("")
-        self.print_fn("")
+        self.survey_print_fn("")
         file.write('ABSTRACT')
-        self.print_fn('ABSTRACT')
-        self.print_fn("=================================================")
+        self.survey_print_fn('ABSTRACT')
+        self.survey_print_fn("=================================================")
         file.write("=================================================")
         file.write("")
-        self.print_fn("")
+        self.survey_print_fn("")
         file.write(research_sections['abstract'])
-        self.print_fn(research_sections['abstract'])
+        self.survey_print_fn(research_sections['abstract'])
         file.write("")
-        self.print_fn("")
+        self.survey_print_fn("")
         file.write('INTRODUCTION')
-        self.print_fn('INTRODUCTION')
-        self.print_fn("=================================================")
+        self.survey_print_fn('INTRODUCTION')
+        self.survey_print_fn("=================================================")
         file.write("=================================================")
         file.write("")
-        self.print_fn("")
+        self.survey_print_fn("")
         file.write(research_sections['introduction'])
-        self.print_fn(research_sections['introduction'])
+        self.survey_print_fn(research_sections['introduction'])
         file.write("")
-        self.print_fn("")
+        self.survey_print_fn("")
         for k, v in research_sections.items():
             if k not in ['abstract', 'introduction', 'conclusion']:
                 file.write(k.upper())
-                self.print_fn(k.upper())
-                self.print_fn("=================================================")
+                self.survey_print_fn(k.upper())
+                self.survey_print_fn("=================================================")
                 file.write("=================================================")
                 file.write("")
-                self.print_fn("")
+                self.survey_print_fn("")
                 file.write(v)
-                self.print_fn(v)
+                self.survey_print_fn(v)
                 file.write("")
-                self.print_fn("")
+                self.survey_print_fn("")
         file.write('CONCLUSION')
-        self.print_fn('CONCLUSION')
-        self.print_fn("=================================================")
+        self.survey_print_fn('CONCLUSION')
+        self.survey_print_fn("=================================================")
         file.write("=================================================")
         file.write("")
-        self.print_fn("")
+        self.survey_print_fn("")
         file.write(research_sections['conclusion'])
-        self.print_fn(research_sections['conclusion'])
+        self.survey_print_fn(research_sections['conclusion'])
         file.write("")
-        self.print_fn("")
+        self.survey_print_fn("")
 
         file.write('REFERENCES')
-        self.print_fn('REFERENCES')
-        self.print_fn("=================================================")
+        self.survey_print_fn('REFERENCES')
+        self.survey_print_fn("=================================================")
         file.write("=================================================")
         file.write("")
-        self.print_fn("")
+        self.survey_print_fn("")
         for entry in bibentries:
             file.write(entry)
-            self.print_fn(entry)
+            self.survey_print_fn(entry)
             file.write("")
-            self.print_fn("")
-        self.print_fn("========================XXX=========================")
+            self.survey_print_fn("")
+        self.survey_print_fn("========================XXX=========================")
         file.write("========================XXX=========================")
         file.close()
 

src/defaults.py

@@ -28,9 +28,9 @@ DEFAULTS_CPU_COMPAT = {
     "tab_dir": "arxiv_data/tables/",
     "dump_dir": "arxiv_dumps/",
     "models_dir": "saved_models/",
-    "title_model_name": "ccdv/lsg-bart-base-4096-arxiv",
+    "title_model_name": "Callidior/bert2bert-base-arxiv-titlegen",
     "ex_summ_model_name": "allenai/scibert_scivocab_uncased",
-    "ledmodel_name": "bhuvaneswari/t5-small-text_summarization",
+    "ledmodel_name": "ccdv/lsg-bart-base-4096-arxiv",
     "embedder_name": "paraphrase-MiniLM-L6-v2",
     "nlp_name": "en_core_sci_scibert",
     "similarity_nlp_name": "en_core_sci_lg",