Update app.py

app.py

@@ -24,41 +24,113 @@ def extract_text_from_pdf(pdf_path):
     except Exception as e:
         return f"Error extracting text from PDF: {str(e)}"
 
-# Function to generate PRISMA flow chart HTML
-def generate_prisma_flow_chart(total_records_db, additional_records, duplicates_removed, screened_records, excluded_title_abstract, full_text_assessed, excluded_full_text, included_studies):
-    prisma_html = f"""
-    <div style="text-align: center; margin: 20px auto; max-width: 800px;">
+# Function to generate PRISMA flowchart HTML
+def generate_prisma_chart(num_studies):
+    return f"""
+    <div class="prisma-section" style="margin: 40px 0; text-align: center;">
         <h2>PRISMA Flow Diagram</h2>
-        <div style="border: 1px solid #000; padding: 10px; background-color: #f9f9f9;">
-            <div style="margin: 10px 0;">
-                <strong>Identification of studies via databases and registers</strong>
-                <div>Records identified from: Databases (n = {total_records_db})</div>
-                <div>Records removed before screening: Duplicates marked as ineligible by automation tools (n = {duplicates_removed})</div>
-            </div>
-            <div style="margin: 10px 0;">
-                <strong>Screening</strong>
-                <div>Records screened (n = {screened_records})</div>
-                <div>Records excluded (n = {excluded_title_abstract})</div>
+        <div class="prisma-flowchart" style="
+            display: inline-block;
+            position: relative;
+            padding: 30px 0;
+            text-align: left;
+            font-size: 14px;
+        ">
+            <!-- Identification -->
+            <div style="display: flex; align-items: center; margin-bottom: 30px;">
+                <div style="
+                    border: 2px solid #4a00e0;
+                    padding: 15px 25px;
+                    border-radius: 8px;
+                    background: #f8f5ff;
+                    width: 220px;
+                    text-align: center;
+                ">
+                    <strong>Identification</strong><br>
+                    Databases (n = 0)<br>
+                    Registers (n = 0)<br>
+                    Other sources (n = {num_studies})
+                </div>
+                <div style="margin-left: 20px;">→</div>
             </div>
-            <div style="margin: 10px 0;">
-                <strong>Eligibility</strong>
-                <div>Reports sought for retrieval (n = {full_text_assessed})</div>
-                <div>Reports not retrieved (n = 0)</div>
-                <div>Reports assessed for eligibility (n = {full_text_assessed})</div>
-                <div>Reports excluded (n = {excluded_full_text})</div>
+
+            <!-- Screening -->
+            <div style="display: flex; align-items: center; margin-bottom: 30px; margin-left: 50px;">
+                <div style="
+                    border: 2px solid #4a00e0;
+                    padding: 15px 25px;
+                    border-radius: 8px;
+                    background: #f8f5ff;
+                    width: 220px;
+                    text-align: center;
+                ">
+                    <strong>Screening</strong><br>
+                    Records screened (n = {num_studies})<br>
+                    Excluded (n = 0)
+                </div>
+                <div style="margin-left: 20px;">→</div>
             </div>
-            <div style="margin: 10px 0;">
-                <strong>Included</strong>
-                <div>New studies included in review (n = {included_studies})</div>
+
+            <!-- Eligibility -->
+            <div style="display: flex; align-items: center; margin-bottom: 30px; margin-left: 100px;">
+                <div style="
+                    border: 2px solid #4a00e0;
+                    padding: 15px 25px;
+                    border-radius: 8px;
+                    background: #f8f5ff;
+                    width: 220px;
+                    text-align: center;
+                ">
+                    <strong>Eligibility</strong><br>
+                    Full-text assessed (n = {num_studies})<br>
+                    Excluded (n = 0)
+                </div>
+                <div style="margin-left: 20px;">→</div>
             </div>
-            <div style="margin: 10px 0;">
-                <strong>Identification of studies via other methods</strong>
-                <div>Records identified from: Websites (n = 0), Organizations (n = 0), Citation searching (n = {additional_records})</div>
+
+            <!-- Included -->
+            <div style="margin-left: 150px;">
+                <div style="
+                    border: 2px solid #4a00e0;
+                    padding: 15px 25px;
+                    border-radius: 8px;
+                    background: #f8f5ff;
+                    width: 220px;
+                    text-align: center;
+                ">
+                    <strong>Included</strong><br>
+                    Studies included (n = {num_studies})
+                </div>
             </div>
+
+            <!-- Connecting lines -->
+            <div style="
+                position: absolute;
+                left: 240px;
+                top: 50px;
+                width: 20px;
+                height: 80%;
+                border-left: 2px dashed #4a00e0;
+            "></div>
+            <div style="
+                position: absolute;
+                left: 290px;
+                top: 130px;
+                width: 20px;
+                height: 60%;
+                border-left: 2px dashed #4a00e0;
+            "></div>
+            <div style="
+                position: absolute;
+                left: 340px;
+                top: 210px;
+                width: 20px;
+                height: 40%;
+                border-left: 2px dashed #4a00e0;
+            "></div>
         </div>
     </div>
     """
-    return prisma_html
 
 # Function to interact with OpenAI API for systematic review
 def generate_systematic_review(pdf_files, review_question, include_tables=True):
@@ -132,18 +204,6 @@ def generate_systematic_review(pdf_files, review_question, include_tables=True):
             pdf_texts.append(pdf_text)
             pdf_names.append(pdf_name)
         
-        # Prepare PRISMA flow chart data (example values, adjust based on actual process)
-        total_records_db = len(pdf_files)
-        additional_records = 0
-        duplicates_removed = 0
-        screened_records = total_records_db
-        excluded_title_abstract = 0
-        full_text_assessed = total_records_db
-        excluded_full_text = 0
-        included_studies = total_records_db
-        
-        prisma_chart = generate_prisma_flow_chart(total_records_db, additional_records, duplicates_removed, screened_records, excluded_title_abstract, full_text_assessed, excluded_full_text, included_studies)
-        
         # Prepare the user prompt with the review question and instructions
         table_instruction = ""
         if include_tables:
@@ -168,8 +228,19 @@ def generate_systematic_review(pdf_files, review_question, include_tables=True):
         
         # Get the AI response
         review_content = response["choices"][0]["message"]["content"]
-        
-        # Apply professional academic paper styling with PRISMA chart in the middle
+
+        # Generate PRISMA chart
+        num_studies = len(pdf_files)
+        prisma_chart = generate_prisma_chart(num_studies)
+        
+        # Insert PRISMA chart after the abstract or methodology section
+        if "Methodology</h2>" in review_content:
+            review_content = review_content.replace("Methodology</h2>", 
+                f"Methodology</h2>{prisma_chart}")
+        else:
+            review_content = prisma_chart + review_content
+        
+        # Apply professional academic paper styling
         styled_html = f"""
         <!DOCTYPE html>
         <html lang="en">
@@ -292,6 +363,29 @@ def generate_systematic_review(pdf_files, review_question, include_tables=True):
                     font-size: 12px;
                     color: #777;
                 }}
+                /* PRISMA Flowchart Styling */
+                .prisma-section {{
+                    margin: 40px 0;
+                    text-align: center;
+                }}
+                
+                .prisma-flowchart {{
+                    display: inline-block;
+                    position: relative;
+                    padding: 30px 0;
+                    text-align: left;
+                    font-size: 14px;
+                }}
+                
+                .prisma-flowchart div[style*="border: 2px solid"] {{
+                    transition: all 0.3s ease;
+                    box-shadow: 0 2px 8px rgba(0,0,0,0.1);
+                }}
+                
+                .prisma-flowchart div[style*="border: 2px solid"]:hover {{
+                    transform: translateY(-3px);
+                    box-shadow: 0 4px 12px rgba(0,0,0,0.15);
+                }}
                 @media print {{
                     body {{
                         background-color: white;
@@ -306,8 +400,6 @@ def generate_systematic_review(pdf_files, review_question, include_tables=True):
         <body>
             <div class="paper-container">
                 {review_content}
-                <!-- Insert PRISMA Flow Chart in the middle -->
-                {prisma_chart}
             </div>
         </body>
         </html>