Update app.py

app.py

@@ -3,6 +3,9 @@ import openai
 import fitz  # PyMuPDF for PDF processing
 import os
 import tempfile
+import base64
+import pdfkit
+from datetime import datetime
 
 # Variable to store API key
 api_key = ""
@@ -283,6 +286,57 @@ def save_uploaded_files(files):
     
     return saved_paths
 
+# Function to create a downloadable HTML file
+def create_html_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
+    
+    # Create timestamp for the filename
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    filename = f"systematic_review_{timestamp}.html"
+    
+    # Encode the HTML content for download
+    b64_html = base64.b64encode(html_content.encode()).decode()
+    download_link = f'<a href="data:text/html;base64,{b64_html}" download="{filename}" class="download-button">Download HTML</a>'
+    
+    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>
@@ -374,6 +428,35 @@ custom_css = """
         margin-bottom: 20px;
     }
     
+    /* Download Buttons */
+    .download-button {
+        display: inline-block;
+        background: linear-gradient(135deg, #4a00e0 0%, #8e2de2 100%);
+        color: white;
+        font-weight: bold;
+        padding: 8px 16px;
+        border-radius: 4px;
+        text-decoration: none;
+        margin-right: 10px;
+        margin-bottom: 10px;
+        transition: all 0.3s ease;
+    }
+    .download-button:hover {
+        background: linear-gradient(135deg, #5b10f1 0%, #9f3ef3 100%);
+        transform: translateY(-2px);
+        box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+    }
+    
+    /* Download Container */
+    #download-container {
+        display: flex;
+        justify-content: center;
+        margin: 20px 0;
+        padding: 15px;
+        background-color: #f5f5f5;
+        border-radius: 8px;
+    }
+    
     /* Responsive adjustments */
     @media screen and (max-width: 768px) {
         .gradio-container {
@@ -395,6 +478,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
         
         ### Tips for Best Results:
         - Upload papers that are related to the same research topic or field
@@ -425,6 +509,9 @@ 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_html_output = gr.HTML(label="Download Options")
+    
     # Output with improved styling
     with gr.Row(elem_classes="output-container"):
         review_output = gr.HTML(label="Systematic Review")
@@ -435,18 +522,18 @@ with gr.Blocks(css=custom_css) as demo:
     # Generate systematic review
     def process_files_and_generate_review(files, question, include_tables):
         if not files:
-            return """
+            return ("""
             <div style="padding: 20px; border: 1px solid #e0e0e0; border-radius: 5px; background-color: #f9f9f9;">
                 <h3 style="color: #666;">Please upload at least one PDF file.</h3>
                 <p>To generate a systematic review, upload one or more research papers in PDF format.</p>
             </div>
-            """
+            """, "")
         
         # Save uploaded files
         saved_paths = save_uploaded_files(files)
         
         # Show loading message
-        yield """
+        loading_message = """
         <div style="padding: 20px; text-align: center;">
             <h3>Generating Systematic Review...</h3>
             <p>This may take a few minutes depending on the number and size of papers.</p>
@@ -462,9 +549,26 @@ with gr.Blocks(css=custom_css) as demo:
         </div>
         """
         
+        yield loading_message, ""
+        
         # Generate review
         review = generate_systematic_review(saved_paths, question, include_tables)
         
+        # Create download links
+        html_link = create_html_download_link(review)
+        pdf_link = create_pdf_download_link(review)
+        
+        # Create combined download links HTML
+        download_links = f"""
+        <div id="download-container">
+            <div>
+                <h3>Download Options:</h3>
+                {html_link or ""}
+                {pdf_link or ""}
+            </div>
+        </div>
+        """
+        
         # Clean up temporary files
         for path in saved_paths:
             try:
@@ -472,12 +576,12 @@ with gr.Blocks(css=custom_css) as demo:
             except:
                 pass
         
-        yield review
+        yield review, download_links
     
     generate_button.click(
         process_files_and_generate_review, 
         inputs=[pdf_files, review_question, include_tables], 
-        outputs=[review_output]
+        outputs=[review_output, download_html_output]
     )
 
 # Launch the app