Update app.py

app.py

@@ -4,7 +4,6 @@ import fitz  # PyMuPDF for PDF processing
 import os
 import tempfile
 import base64
-import pdfkit
 from datetime import datetime
 
 # Variable to store API key
@@ -301,42 +300,6 @@ def create_html_download_link(html_content):
     
     return download_link
 
-# Function to create a downloadable PDF file
-def create_pdf_download_link(html_content):
-    if not html_content or "<div style=\"color: red; padding: 20px;" in html_content or "Please upload at least one PDF file" in html_content:
-        return None
-    
-    try:
-        # Create timestamp for the filename
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        pdf_filename = f"systematic_review_{timestamp}.pdf"
-        
-        # Create a temporary file for the PDF
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_pdf:
-            pdf_path = tmp_pdf.name
-        
-        # Convert HTML to PDF
-        pdfkit.from_string(html_content, pdf_path)
-        
-        # Read the PDF file and encode it
-        with open(pdf_path, "rb") as pdf_file:
-            pdf_data = pdf_file.read()
-        
-        # Encode the PDF content for download
-        b64_pdf = base64.b64encode(pdf_data).decode()
-        download_link = f'<a href="data:application/pdf;base64,{b64_pdf}" download="{pdf_filename}" class="download-button">Download PDF</a>'
-        
-        # Clean up the temporary file
-        try:
-            os.remove(pdf_path)
-        except:
-            pass
-        
-        return download_link
-        
-    except Exception as e:
-        return f'<div style="color: red; margin-top: 10px;">PDF generation error: {str(e)}</div>'
-
 # Add CSS styling for the Gradio interface
 custom_css = """
 <style>
@@ -428,7 +391,7 @@ custom_css = """
         margin-bottom: 20px;
     }
     
-    /* Download Buttons */
+    /* Download Button */
     .download-button {
         display: inline-block;
         background: linear-gradient(135deg, #4a00e0 0%, #8e2de2 100%);
@@ -437,7 +400,6 @@ custom_css = """
         padding: 8px 16px;
         border-radius: 4px;
         text-decoration: none;
-        margin-right: 10px;
         margin-bottom: 10px;
         transition: all 0.3s ease;
     }
@@ -478,7 +440,7 @@ with gr.Blocks(css=custom_css) as demo:
         3. Enter your review question or topic
         4. Check the "Include Tables" option if you want the review to include comparison tables
         5. Click "Generate Systematic Review" to start the process
-        6. After generation, you can download the review as HTML or PDF
+        6. After generation, you can download the review as HTML
         
         ### Tips for Best Results:
         - Upload papers that are related to the same research topic or field
@@ -509,7 +471,7 @@ with gr.Blocks(css=custom_css) as demo:
             include_tables = gr.Checkbox(label="Include Comparison Tables", value=True)
             generate_button = gr.Button("Generate Systematic Review", elem_id="generate_button", size="large")
     
-    # Download links container
+    # Download link container
     download_html_output = gr.HTML(label="Download Options")
     
     # Output with improved styling
@@ -554,17 +516,15 @@ with gr.Blocks(css=custom_css) as demo:
         # Generate review
         review = generate_systematic_review(saved_paths, question, include_tables)
         
-        # Create download links
+        # Create HTML download link
         html_link = create_html_download_link(review)
-        pdf_link = create_pdf_download_link(review)
         
-        # Create combined download links HTML
-        download_links = f"""
+        # Create download link HTML
+        download_link = f"""
         <div id="download-container">
             <div>
-                <h3>Download Options:</h3>
+                <h3>Download Option:</h3>
                 {html_link or ""}
-                {pdf_link or ""}
             </div>
         </div>
         """
@@ -576,7 +536,7 @@ with gr.Blocks(css=custom_css) as demo:
             except:
                 pass
         
-        yield review, download_links
+        yield review, download_link
     
     generate_button.click(
         process_files_and_generate_review,