Update app.py

app.py

@@ -436,10 +436,10 @@ 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}")
@@ -465,10 +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>')
+                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'<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>')
 
         if downloaded_files:
             zip_filename = 'papers.zip'
@@ -522,13 +522,13 @@ 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}")
 
         # Create ZIP of downloaded papers
@@ -584,13 +584,13 @@ 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}")
 
         # Create ZIP of downloaded papers
@@ -682,20 +682,41 @@ def create_gradio_interface():
 
     # Add Javascript to update HTML
     interface.load = """
-        function(downloaded_dois, failed_dois){
-             let downloaded_html = '';
-             downloaded_dois.split('\\n').filter(Boolean).forEach(doi => {
-                downloaded_html +=  '[' + doi + ']<br>';
-            });
-            document.querySelector("#downloaded-dois").innerHTML = downloaded_html;
-
-             let failed_html = '';
-             failed_dois.split('\\n').filter(Boolean).forEach(doi => {
-                failed_html += '[' + doi + ']<br>';
+       function(downloaded_dois, failed_dois) {
+          let downloaded_html = '';
+          downloaded_dois.split('\\n').filter(Boolean).forEach(doi => {
+              downloaded_html +=  doi + '<br>';
+          });
+          document.querySelector("#downloaded-dois").innerHTML = downloaded_html;
+
+          let failed_html = '';
+            failed_dois.split('\\n').filter(Boolean).forEach(doi => {
+            failed_html += doi + '<br>';
+          });
+          document.querySelector("#failed-dois").innerHTML = failed_html;
+          return [downloaded_html, failed_html];
+       }
+    """
+    
+    interface.head = """
+    <script>
+        function copyLink(button) {
+            const linkElement = button.previousElementSibling;
+            const link = linkElement.href;
+            navigator.clipboard.writeText(link)
+            .then(() => {
+                button.innerText = '✓ Copied';
+                button.style.color = 'green';
+                setTimeout(() => {
+                    button.innerText = 'Copy';
+                    button.style.color = '';
+                }, 2000);
+            })
+            .catch(err => {
+                console.error('Failed to copy link: ', err);
             });
-            document.querySelector("#failed-dois").innerHTML = failed_html;
-            return [downloaded_html, failed_html];
         }
+    </script>
     """
     return interface