Update app.py

app.py

@@ -393,24 +393,30 @@ def create_gradio_interface():
         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):
+    def create_report(downloaded_dois_html, failed_dois_html):
         """Creates a report and returns the report path."""
+        downloaded_dois = ""
+        failed_dois = ""
+
+         # Extract text content from HTML elements
+        if downloaded_dois_html:
+             soup = BeautifulSoup(downloaded_dois_html, 'html.parser')
+             downloaded_dois_items = [li.text for li in soup.find_all('li')]
+             downloaded_dois = "\n".join(downloaded_dois_items)
+
+        if failed_dois_html:
+           soup = BeautifulSoup(failed_dois_html, 'html.parser')
+           failed_dois_items = [li.text for li in soup.find_all('li')]
+           failed_dois = "\n".join(failed_dois_items)
+        
         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(
-        fn=download_papers,
-        inputs=[
-            gr.File(file_types=['.bib'], label="Upload BibTeX File"),
-            gr.Textbox(label="Enter Single DOI", placeholder="10.xxxx/xxxx"),
-            gr.Textbox(label="Enter Multiple DOIs (one per line)", placeholder="10.xxxx/xxxx\n10.yyyy/yyyy\n...")
-        ],
-        outputs=[
-            gr.File(label="Download Papers (ZIP) or Single PDF"),
-             gr.HTML(label="""
+    downloaded_dois_html = gr.HTML(label="""
                <div style='padding-bottom: 5px; font-weight: bold;'>
                         Enter Single DOI
                </div>
@@ -420,15 +426,28 @@ def create_gradio_interface():
                     </div>
                     <div id="downloaded-dois"></div>
                  </div>
-            """),
-           gr.HTML(label="""
+            """)
+    failed_dois_html = 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>
-            """),
+            """)
+
+
+    interface = gr.Interface(
+        fn=download_papers,
+        inputs=[
+            gr.File(file_types=['.bib'], label="Upload BibTeX File"),
+            gr.Textbox(label="Enter Single DOI", placeholder="10.xxxx/xxxx"),
+            gr.Textbox(label="Enter Multiple DOIs (one per line)", placeholder="10.xxxx/xxxx\n10.yyyy/yyyy\n...")
+        ],
+        outputs=[
+            gr.File(label="Download Papers (ZIP) or Single PDF"),
+             downloaded_dois_html,
+             failed_dois_html,
             gr.File(label="Downloaded Single PDF"),
         ],
         title="🔬 Academic Paper Batch Downloader",
@@ -484,7 +503,7 @@ def create_gradio_interface():
     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)
+        report_button.click(create_report, inputs = [downloaded_dois_html,failed_dois_html], outputs=report_output)
 
     return interface