Update app.py

app.py

@@ -397,10 +397,10 @@ class PaperDownloader:
           
 
             for doi in dois:
-                    self._download_single_doi(doi, lambda a,b,c: progress_callback(a,b,c, doi) , cancel_event )
+                    self._download_single_doi(doi, lambda a,b,c,d: progress_callback(a,b,c,d) , cancel_event ) # changed lambda
                     if cancel_event.is_set():
                          logger.info("Downloads cancelled on multiple dois download")
-                         progress_callback(None,  "Downloads cancelled","Downloads cancelled") # early return on cancelled
+                         progress_callback(None,  "Downloads cancelled","Downloads cancelled", ) # early return on cancelled
                          return None #break here when is cancelled
 
                     result = self.results_dict.get(doi, (None,None,"")) # obtain from self.results
@@ -456,7 +456,7 @@ class PaperDownloader:
         
 
             # Devolver resultados
-            progress_callback( zip_filename if downloaded_files else None, "\n".join(downloaded_links),"\n".join(failed_dois))
+            progress_callback( zip_filename if downloaded_files else None, "\n".join(downloaded_links),"\n".join(failed_dois), "")
             return
             
     def process_bibtex(self, bib_file, progress_callback, cancel_event):# removed async here
@@ -489,10 +489,10 @@ class PaperDownloader:
 
             for doi in dois:
 
-                 self._download_single_doi(doi,  lambda a,b,c: progress_callback(a,b,c, doi), cancel_event )
+                 self._download_single_doi(doi,  lambda a,b,c, d: progress_callback(a,b,c, d), cancel_event ) # changed lambda
                  if cancel_event.is_set():
                        logger.info("Download Cancelled in bibtex mode")
-                       progress_callback(None, "Download Cancelled", "Download Cancelled")
+                       progress_callback(None, "Download Cancelled", "Download Cancelled", )
                        return None #cancel if requested
                        
                  result = self.results_dict.get(doi, (None,None,"")) # obtain from self.results
@@ -527,7 +527,7 @@ class PaperDownloader:
                 loop.run_in_executor(self.executor, lambda:  self.create_zip(zip_filename,downloaded_files))
                 logger.info(f"ZIP file created: {zip_filename}")
     
-            progress_callback(zip_filename, "\n".join(downloaded_links), "\n".join(failed_dois)) #after process finishes
+            progress_callback(zip_filename, "\n".join(downloaded_links), "\n".join(failed_dois), "") # send 4 vars
             return
 
     def create_zip(self, zip_filename, files):
@@ -564,47 +564,46 @@ def create_gradio_interface():
                     if doi:
                           downloader.results_dict[doi] = (filepath, message,fail_message)
 
-                    updates = update_progress(message,fail_message) #use update function here
+                    updates = update_progress(message, fail_message)
                     return updates
 
             if bib_file:
                  # Check file type
                  if not bib_file.name.lower().endswith('.bib'):
-                      return None, "Error: Please upload a .bib file", "Error: Please upload a .bib file", None
+                      return None, "Error: Please upload a .bib file", "Error: Please upload a .bib file", "",""
 
                  downloader.download_task = downloader.executor.submit(
                        downloader.process_bibtex, 
                         bib_file,
-                        lambda a,b,c:  update_progress(a,f"{b}<br>{c}"), #convert for ui output, the return data will contain the HTML
-                        cancel_event # Added cancelllation event.
-                    )
-                    
-                 return None,"","",None
+                       lambda a,b,c,d :  update_progress(a,f"{b}<br>{c}"), #convert for ui output, the return data will contain the HTML
+                         cancel_event # Added cancelllation event.
+                 )
+                 
+                 return None,"","", "", None # output here correct
            
             elif doi_input:
                
-                  downloader.download_task = downloader.executor.submit( #changed async execution method
+                 downloader.download_task = downloader.executor.submit( #changed async execution method
                        downloader._download_single_doi, 
-                         doi_input, 
-                           lambda a,b,c: update_progress(a,f"{b}<br>{c}")  , #callback function, format output and send html info, removed lambda from executor calls
+                        doi_input, 
+                          lambda a,b,c,d: update_progress(a,f"{b}<br>{c}") , #callback function, format output and send html info, removed lambda from executor calls
                             cancel_event  # Add cancellation event.
-                      )
+                     )
                  
-                  return  None, "","", None
-              
+                 return  None, "","", "", None  # output here correct
+             
             elif dois_input:
-                 downloader.download_task = downloader.executor.submit( #changed async execution method
-                      downloader.download_multiple_dois,
-                        dois_input,
-                       lambda a,b,c: update_progress(a,f"{b}<br>{c}") ,#callback function
-                         cancel_event #Add cancellation event.
+                downloader.download_task = downloader.executor.submit( #changed async execution method
+                       downloader.download_multiple_dois,
+                         dois_input,
+                        lambda a,b,c,d: update_progress(a,f"{b}<br>{c}") ,#callback function
+                        cancel_event #Add cancellation event.
                    )
-                 
-                 return  None, "","", None
+                
+                return  None, "","", "", None #output here correct
 
             else:
-                 return None, "Please provide a .bib file, a single DOI, or a list of DOIs", "Please provide a .bib file, a single DOI, or a list of DOIs","", None
-        
+                 return None, "Please provide a .bib file, a single DOI, or a list of DOIs", "Please provide a .bib file, a single DOI, or a list of DOIs","",  None #output here correct
 
     
     with gr.Blocks(theme="Hev832/Applio", css="""