Update app.py

app.py

@@ -336,36 +336,25 @@ class PaperDownloader:
 
 def create_gradio_interface():
     """Create Gradio interface for Paper Downloader"""
+    downloader = PaperDownloader()
+
     def download_papers(bib_file, doi_input, dois_input):
-        # Your existing download_papers function implementation
-        
-        # Update the HTML labels
-        downloaded_dois_html = ""
-        failed_dois_html = ""
-        
-        # Build the HTML for Found DOIs
-        if downloaded_dois:
-            downloaded_dois_html = """
-                <div class="found-dois-container">
-                    <div class="found-dois-header">📥 Found DOIs</div>
-                    <div class="found-dois-list">
-                        {}
-                    </div>
-                </div>
-            """.format("\n".join([f'<a href="https://doi.org/{doi}" class="doi-link found-doi-link">{doi}</a>' for doi in downloaded_dois.split("\n") if doi]))
-        
-        # Build the HTML for Missed DOIs
-        if failed_dois:
-            failed_dois_html = """
-                <div class="missed-dois-container">
-                    <div class="missed-dois-header">❌ Missed DOIs</div>
-                    <div class="missed-dois-list">
-                        {}
-                    </div>
-                </div>
-            """.format("\n".join([f'<a href="https://doi.org/{doi}" class="doi-link missed-doi-link">{doi}</a>' for doi in failed_dois.split("\n") if doi]))
-        
-        return zip_path, downloaded_dois_html, failed_dois_html, filepath
+        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, _ = downloader.process_bibtex(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)
+           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
+
 
     # Gradio Interface
     interface = gr.Interface(
@@ -377,40 +366,73 @@ def create_gradio_interface():
         ],
         outputs=[
             gr.File(label="Download Papers (ZIP) or Single PDF"),
-            gr.HTML(label=downloaded_dois_html),
-            gr.HTML(label=failed_dois_html),
+             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.",
         theme="Hev832/Applio",
         examples=[
-            ["example.bib", None, None],
-            [None, "10.1038/nature12373", None],
-            [None, None, "10.1109/5.771073\n10.3390/horticulturae8080677"]
-        ],
-        cache_examples=False
+            ["example.bib", None, None], # Bibtex File
+            [None, "10.1038/nature12373", None], # Single DOI
+            [None, None, "10.1109/5.771073\n10.3390/horticulturae8080677"], # Multiple DOIs
+         ],
+         css="""
+        .gradio-container {
+            background-color: #222222;
+        }
+        .gr-interface {
+            max-width: 800px;
+            margin: 0 auto;
+        }
+        .gr-box {
+            background-color: white;
+            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,
     )
     
-    return interface
-    
     # Add Javascript to update HTML
     interface.load = """
         function(downloaded_dois, failed_dois){
-            let downloaded_html = '<ul style="list-style-type: none; padding: 0; margin: 0;">';
-            downloaded_dois.split('\\n').filter(Boolean).forEach(doi => {
-                downloaded_html += '<li style="margin-bottom: 5px; padding: 5px; background-color: #e6f3e6; border-radius: 3px;">' + doi + '</li>';
+             let downloaded_html =  '<ul>';
+             downloaded_dois.split('\\n').filter(Boolean).forEach(doi => {
+                downloaded_html +=  '<li>' + doi + '</li>';
             });
-            downloaded_html += '</ul>';
+            downloaded_html +=  '</ul>';
             document.querySelector("#downloaded-dois").innerHTML = downloaded_html;
-    
-            let failed_html = '<ul style="list-style-type: none; padding: 0; margin: 0;">';
-            failed_dois.split('\\n').filter(Boolean).forEach(doi => {
-                failed_html += '<li style="margin-bottom: 5px; padding: 5px; background-color: #f2dede; border-radius: 3px;">' + doi + '</li>';
+
+             let failed_html = '<ul>';
+             failed_dois.split('\\n').filter(Boolean).forEach(doi => {
+                failed_html += '<li>' + doi + '</li>';
             });
-            failed_html += '</ul>';
+             failed_html += '</ul>';
             document.querySelector("#failed-dois").innerHTML = failed_html;
             return [downloaded_html, failed_html];
+
         }
     """
     return interface