Update app.py

app.py

@@ -436,17 +436,17 @@ class PaperDownloader:
                 with open(filepath, 'wb') as f:
                     f.write(pdf_content)
                 logger.info(f"Successfully downloaded: {filename}")
-                return filepath, f'<a href="https://doi.org/{doi}">{doi}</a>', ""
+                return filepath, f'<div style="display: flex; align-items: center;">✓ <a href="https://doi.org/{doi}">{doi}</a> <button onclick="copyLink(this)">Copy</button></div>', ""
             else:
                 logger.warning(f"Could not download: {doi}")
-                return None, f"Could not download {doi}", f'<a href="https://doi.org/{doi}">{doi}</a>'
+                return None, f"Could not download {doi}", f'<div style="display: flex; align-items: center;">❌ <a href="https://doi.org/{doi}">{doi}</a>  <button onclick="copyLink(this)">Copy</button></div>'
 
         except Exception as e:
             logger.error(f"Error processing {doi}: {e}")
             return None, f"Error processing {doi}: {e}", f"Error processing {doi}: {e}"
 
-    def download_multiple_dois(self, dois_text, progress_callback=None):
-        """Downloads multiple papers from a list of DOIs with progress updates and single copy button"""
+    def download_multiple_dois(self, dois_text):
+        """Downloads multiple papers from a list of DOIs"""
         if not dois_text:
             return None, "Error: No DOIs provided", "Error: No DOIs provided"
 
@@ -457,9 +457,7 @@ class PaperDownloader:
         downloaded_files = []
         failed_dois = []
         downloaded_links = []
-        total_dois = len(dois)
-
-        for i, doi in enumerate(dois):
+        for i, doi in enumerate(tqdm(dois, desc="Downloading papers")):
             filepath, success_message, fail_message = self.download_single_doi(doi)
             if filepath:
                 # Unique filename for zip
@@ -467,14 +465,10 @@ class PaperDownloader:
                 filepath_unique = os.path.join(self.output_dir, filename)
                 os.rename(filepath, filepath_unique)
                 downloaded_files.append(filepath_unique)
-                downloaded_links.append(f'<a href="https://doi.org/{doi}">{doi}</a>')
-            else:
-                failed_dois.append(f'<a href="https://doi.org/{doi}">{doi}</a>')
-            
-            if progress_callback:
-                progress = int(((i + 1) / total_dois) * 100)
-                progress_callback(progress)
+                downloaded_links.append(f'<div style="display: flex; align-items: center;">✓ <a href="https://doi.org/{doi}">{doi}</a>  <button onclick="copyLink(this)">Copy</button></div>')
 
+            else:
+                failed_dois.append(f'<div style="display: flex; align-items: center;">❌ <a href="https://doi.org/{doi}">{doi}</a>  <button onclick="copyLink(this)">Copy</button></div>')
 
         if downloaded_files:
             zip_filename = 'papers.zip'
@@ -483,17 +477,9 @@ class PaperDownloader:
                     zipf.write(file_path, arcname=os.path.basename(file_path))
             logger.info(f"ZIP file created: {zip_filename}")
 
-        
-        
-        
-        # Combine all links into a single string
-        all_links_html = "<br>".join(downloaded_links)
-        copy_button_html = f'<button onclick="copyAllLinks(\'{all_links_html}\')">Copy All Links</button>' if all_links_html else ""
+        return zip_filename if downloaded_files else None, "\n".join(downloaded_links), "\n".join(failed_dois)
 
-        
-        return zip_filename if downloaded_files else None, f"{all_links_html} {copy_button_html}", "\n".join(failed_dois)
-
-    def process_bibtex(self, bib_file, progress_callback=None):
+    def process_bibtex(self, bib_file):
         """Process BibTeX file and download papers with multiple strategies"""
         # Read BibTeX file content from the uploaded object
         try:
@@ -518,10 +504,9 @@ class PaperDownloader:
         downloaded_files = []
         failed_dois = []
         downloaded_links = []
-        total_dois = len(dois)
 
         # Download PDFs
-        for i, doi in enumerate(dois):
+        for doi in tqdm(dois, desc="Downloading papers"):
             try:
                 # Try to download with multiple methods with retries
                 pdf_content = self.download_with_retry(doi)
@@ -537,17 +522,14 @@ class PaperDownloader:
                         f.write(pdf_content)
 
                     downloaded_files.append(filepath)
-                    downloaded_links.append(f'<a href="https://doi.org/{doi}">{doi}</a>')
+                    downloaded_links.append(f'<div style="display: flex; align-items: center;">✓ <a href="https://doi.org/{doi}">{doi}</a>  <button onclick="copyLink(this)">Copy</button></div>')
                     logger.info(f"Successfully downloaded: {filename}")
                 else:
-                    failed_dois.append(f'<a href="https://doi.org/{doi}">{doi}</a>')
+                    failed_dois.append(f'<div style="display: flex; align-items: center;">❌ <a href="https://doi.org/{doi}">{doi}</a>  <button onclick="copyLink(this)">Copy</button></div>')
+
             except Exception as e:
-                failed_dois.append(f'<a href="https://doi.org/{doi}">{doi}</a>')
+                failed_dois.append(f'<div style="display: flex; align-items: center;">❌ <a href="https://doi.org/{doi}">{doi}</a>  <button onclick="copyLink(this)">Copy</button></div>')
                 logger.error(f"Error processing {doi}: {e}")
-            
-            if progress_callback:
-                progress = int(((i + 1) / total_dois) * 100)
-                progress_callback(progress)
 
         # Create ZIP of downloaded papers
         if downloaded_files:
@@ -556,16 +538,10 @@ class PaperDownloader:
                 for file_path in downloaded_files:
                     zipf.write(file_path, arcname=os.path.basename(file_path))
             logger.info(f"ZIP file created: {zip_filename}")
-        
-        
-        # Combine all links into a single string
-        all_links_html = "<br>".join(downloaded_links)
-        copy_button_html = f'<button onclick="copyAllLinks(\'{all_links_html}\')">Copy All Links</button>' if all_links_html else ""
 
+        return zip_filename, "\n".join(downloaded_links), "\n".join(failed_dois), None
 
-        return zip_filename, f"{all_links_html} {copy_button_html}", "\n".join(failed_dois), None
-
-    async def process_bibtex_async(self, bib_file, progress_callback=None):
+    async def process_bibtex_async(self, bib_file):
         """Process BibTeX file and download papers with multiple strategies"""
         # Read BibTeX file content from the uploaded object
         try:
@@ -590,10 +566,9 @@ class PaperDownloader:
         downloaded_files = []
         failed_dois = []
         downloaded_links = []
-        total_dois = len(dois)
 
         # Download PDFs
-        for i, doi in enumerate(dois):
+        for doi in tqdm(dois, desc="Downloading papers"):
             try:
                 # Try to download with multiple methods with retries
                 pdf_content = await self.download_with_retry_async(doi)
@@ -609,17 +584,14 @@ class PaperDownloader:
                         f.write(pdf_content)
 
                     downloaded_files.append(filepath)
-                    downloaded_links.append(f'<a href="https://doi.org/{doi}">{doi}</a>')
+                    downloaded_links.append(f'<div style="display: flex; align-items: center;">✓ <a href="https://doi.org/{doi}">{doi}</a>  <button onclick="copyLink(this)">Copy</button></div>')
                     logger.info(f"Successfully downloaded: {filename}")
                 else:
-                    failed_dois.append(f'<a href="https://doi.org/{doi}">{doi}</a>')
+                    failed_dois.append(f'<div style="display: flex; align-items: center;">❌ <a href="https://doi.org/{doi}">{doi}</a>  <button onclick="copyLink(this)">Copy</button></div>')
+
             except Exception as e:
-                failed_dois.append(f'<a href="https://doi.org/{doi}">{doi}</a>')
+                failed_dois.append(f'<div style="display: flex; align-items: center;">❌ <a href="https://doi.org/{doi}">{doi}</a>  <button onclick="copyLink(this)">Copy</button></div>')
                 logger.error(f"Error processing {doi}: {e}")
-            
-            if progress_callback:
-                progress = int(((i + 1) / total_dois) * 100)
-                progress_callback(progress)
 
         # Create ZIP of downloaded papers
         if downloaded_files:
@@ -628,34 +600,26 @@ class PaperDownloader:
                 for file_path in downloaded_files:
                     zipf.write(file_path, arcname=os.path.basename(file_path))
             logger.info(f"ZIP file created: {zip_filename}")
-        
-        
-        # Combine all links into a single string
-        all_links_html = "<br>".join(downloaded_links)
-        copy_button_html = f'<button onclick="copyAllLinks(\'{all_links_html}\')">Copy All Links</button>' if all_links_html else ""
 
-
-        return zip_filename, f"{all_links_html} {copy_button_html}", "\n".join(failed_dois), None
+        return zip_filename, "\n".join(downloaded_links), "\n".join(failed_dois), None
 
 def create_gradio_interface():
     """Create Gradio interface for Paper Downloader"""
     downloader = PaperDownloader()
 
-    async def download_papers(bib_file, doi_input, dois_input, progress=gr.Progress()):
-        progress_callback = lambda p: progress(p, desc="Downloading Papers")
-    
+    async def download_papers(bib_file, doi_input, dois_input):
         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
 
-            zip_path, downloaded_dois, failed_dois, _ = await downloader.process_bibtex_async(bib_file, progress_callback)
+            zip_path, downloaded_dois, failed_dois, _ = await downloader.process_bibtex_async(bib_file)
             return zip_path, downloaded_dois, failed_dois, None
         elif doi_input:
             filepath, message, failed_doi = downloader.download_single_doi(doi_input)
             return None, message, failed_doi, filepath
         elif dois_input:
-            zip_path, downloaded_dois, failed_dois = downloader.download_multiple_dois(dois_input, progress_callback)
+            zip_path, downloaded_dois, failed_dois = downloader.download_multiple_dois(dois_input)
             return zip_path, downloaded_dois, failed_dois, None
         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
@@ -686,7 +650,7 @@ def create_gradio_interface():
                     <div id="failed-dois"></div>
                 </div>
             """),
-             gr.File(label="Downloaded Single PDF")
+            gr.File(label="Downloaded Single PDF")
         ],
         title="🔬 Academic Paper Batch Downloader",
         description="Upload a BibTeX file or enter DOIs to download PDFs. We'll attempt to fetch PDFs from multiple sources like Sci-Hub, Libgen, Google Scholar and Crossref. You can use any of the three inputs at any moment.",
@@ -736,17 +700,20 @@ def create_gradio_interface():
     
     interface.head = """
     <script>
-       function copyAllLinks(linksHTML) {
-            const tempElement = document.createElement('div');
-            tempElement.innerHTML = linksHTML;
-            const links = Array.from(tempElement.querySelectorAll('a')).map(a => a.href).join('\\n');
-
-            navigator.clipboard.writeText(links)
+        function copyLink(button) {
+            const linkElement = button.previousElementSibling;
+            const link = linkElement.href;
+            navigator.clipboard.writeText(link)
             .then(() => {
-                alert('All links copied to clipboard!');
+                button.innerText = '✓ Copied';
+                button.style.color = 'green';
+                setTimeout(() => {
+                    button.innerText = 'Copy';
+                    button.style.color = '';
+                }, 2000);
             })
             .catch(err => {
-                console.error('Failed to copy links: ', err);
+                console.error('Failed to copy link: ', err);
             });
         }
     </script>