Update app.py

app.py

@@ -10,6 +10,10 @@ from urllib.parse import quote, urlencode
 import gradio as gr
 from bs4 import BeautifulSoup
 import io
+from docx import Document
+from docx.shared import Inches
+from docx.enum.text import WD_ALIGN_PARAGRAPH
+
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, 
@@ -280,14 +284,14 @@ class PaperDownloader:
                  bib_content = f.read()
         except Exception as e:
              logger.error(f"Error reading uploaded file {bib_file.name}: {e}")
-             return None, f"Error reading uploaded file {bib_file.name}: {e}", f"Error reading uploaded file {bib_file.name}: {e}"
+             return None, f"Error reading uploaded file {bib_file.name}: {e}", f"Error reading uploaded file {bib_file.name}: {e}", None
 
         # Parse BibTeX data
         try:
             bib_database = bibtexparser.loads(bib_content)
         except Exception as e:
              logger.error(f"Error parsing BibTeX data: {e}")
-             return None, f"Error parsing BibTeX data: {e}", f"Error parsing BibTeX data: {e}"
+             return None, f"Error parsing BibTeX data: {e}", f"Error parsing BibTeX data: {e}", None
 
         # Extract DOIs
         dois = [entry.get('doi') for entry in bib_database.entries if entry.get('doi')]
@@ -334,6 +338,39 @@ class PaperDownloader:
 
         return zip_filename, "\n".join(downloaded_links), "\n".join(failed_dois), None
 
+    def create_report_docx(self, downloaded_dois, failed_dois):
+        """Creates a Word document report of downloaded and failed DOIs."""
+        document = Document()
+
+        # Add a title
+        title_paragraph = document.add_paragraph()
+        title_run = title_paragraph.add_run("DOI Download Report")
+        title_run.bold = True
+        title_paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
+
+        # Add downloaded DOIs
+        document.add_heading('Downloaded DOIs', level=2)
+        if downloaded_dois:
+            for doi in downloaded_dois.split('\n'):
+                 if doi:
+                     document.add_paragraph(doi)
+        else:
+            document.add_paragraph("No DOIs were successfully downloaded.")
+
+
+        # Add failed DOIs
+        document.add_heading('Failed DOIs', level=2)
+        if failed_dois:
+            for doi in failed_dois.split('\n'):
+                if doi:
+                     document.add_paragraph(doi)
+        else:
+            document.add_paragraph("No DOIs failed to download.")
+
+
+        report_path = "doi_report.docx"
+        document.save(report_path)
+        return report_path
 
 def create_gradio_interface():
     """Create Gradio interface for Paper Downloader"""
@@ -355,7 +392,13 @@ def create_gradio_interface():
            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
-
+    
+    def create_report(downloaded_dois, failed_dois):
+        """Creates a report and returns the report path."""
+        if downloaded_dois or failed_dois:
+            report_path = downloader.create_report_docx(downloaded_dois, failed_dois)
+            return report_path
+        return None
 
     # Gradio Interface
     interface = gr.Interface(
@@ -367,9 +410,26 @@ def create_gradio_interface():
         ],
         outputs=[
             gr.File(label="Download Papers (ZIP) or Single PDF"),
-            gr.HTML(label="Downloaded DOIs/Message"),
-            gr.HTML(label="Failed DOIs"),
-            gr.File(label="Downloaded Single PDF")
+             gr.HTML(label="""
+               <div style='padding-bottom: 5px; font-weight: bold;'>
+                        Enter Single DOI
+               </div>
+               <div style='border: 1px solid #ddd; padding: 5px; border-radius: 5px;'>
+                    <div style='padding-bottom: 5px; font-weight: bold;'>
+                        Downloaded DOIs
+                    </div>
+                    <div id="downloaded-dois"></div>
+                 </div>
+            """),
+           gr.HTML(label="""
+                <div style='border: 1px solid #ddd; padding: 5px; border-radius: 5px;'>
+                    <div style='padding-bottom: 5px; font-weight: bold;'>
+                        Failed DOIs
+                    </div>
+                    <div id="failed-dois"></div>
+                 </div>
+            """),
+            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.",
@@ -381,20 +441,50 @@ def create_gradio_interface():
          ],
          css="""
         .gradio-container {
-            background-color: #222222;
+            background-color: black;
         }
         .gr-interface {
             max-width: 800px;
             margin: 0 auto;
         }
         .gr-box {
-            background-color: white;
+            background-color: black;
             border-radius: 10px;
             box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
         }
+           .output-text a {
+               color: #007bff; /* Blue color for hyperlinks */
+            }
         """,
-        cache_examples = False
+        cache_examples = False,
     )
+    
+    # Add Javascript to update HTML
+    interface.load = """
+        function(downloaded_dois, failed_dois){
+             let downloaded_html =  '<ul>';
+             downloaded_dois.split('\\n').filter(Boolean).forEach(doi => {
+                downloaded_html +=  '<li>' + doi + '</li>';
+            });
+            downloaded_html +=  '</ul>';
+            document.querySelector("#downloaded-dois").innerHTML = downloaded_html;
+
+             let failed_html = '<ul>';
+             failed_dois.split('\\n').filter(Boolean).forEach(doi => {
+                failed_html += '<li>' + doi + '</li>';
+            });
+             failed_html += '</ul>';
+            document.querySelector("#failed-dois").innerHTML = failed_html;
+            return [downloaded_html, failed_html];
+
+        }
+    """
+    
+    # Add the report button
+    with gr.Row():
+        report_button = gr.Button("Create Report")
+        report_output = gr.File(label="Download Report")
+        report_button.click(create_report, inputs = [interface.outputs[1],interface.outputs[2]], outputs=report_output)
 
     return interface