Update app.py

app.py

@@ -86,50 +86,44 @@ def generate_pdf_report(results):
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
-    
-    # Page dimensions (A4: 210mm x 297mm)
-    page_width = 210
-    margin = 20
-    effective_width = page_width - 2*margin
-    
-    # Watermark
+
+    # Add watermark
     pdf.set_font("Arial", 'B', 50)
-    pdf.set_text_color(230, 230, 230)
-    pdf.rotate(45)
-    pdf.text(60, 150, "AI Powered Requirement Defects Analysis")
-    pdf.rotate(0)
-    
-    # Reset font for content
+    pdf.set_text_color(230, 230, 230)  # Light gray color for watermark
+    pdf.rotate(45)  # Rotate the text for watermark effect
+    pdf.text(60, 150, "AI Powered Requirement Analysis")
+    pdf.rotate(0)  # Reset rotation
+
+    # Add title and date/time
     pdf.set_font("Arial", 'B', 16)
-    pdf.set_text_color(0, 0, 0)
-    pdf.cell(effective_width, 10, txt="AI Powered Requirement Analysis", ln=True, align='C')
+    pdf.set_text_color(0, 0, 0)  # Black color for title
+    pdf.cell(200, 10, txt="AI Powered Requirement Analysis and Defect Detection", ln=True, align='C')
     pdf.set_font("Arial", size=12)
-    pdf.cell(effective_width, 10, txt=f"Report Generated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", ln=True, align='C')
-    pdf.ln(10)
+    pdf.cell(200, 10, txt=f"Report Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", ln=True, align='C')
+    pdf.ln(10)  # Add some space
 
+    # Add requirements analysis
+    pdf.set_font("Arial", size=12)
     for i, result in enumerate(results, start=1):
-        if pdf.get_y() > 250:  # Check if near bottom
-            pdf.add_page()
-            # Reset formatting for new page
+        if pdf.get_y() > 250:  # If the content is near the bottom of the page
+            pdf.add_page()  # Add a new page
             pdf.set_font("Arial", 'B', 16)
-            pdf.cell(effective_width, 10, txt="AI Powered Requirement Analysis", ln=True, align='C')
+            pdf.cell(200, 10, txt="AI Powered Requirement Analysis and Defect Detection", ln=True, align='C')
             pdf.set_font("Arial", size=12)
-            pdf.cell(effective_width, 10, txt=f"Report Generated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", ln=True, align='C')
-            pdf.ln(10)
+            pdf.cell(200, 10, txt=f"Report Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", ln=True, align='C')
+            pdf.ln(10)  # Add some space
 
-        # Requirement content with proper widths
+        # Add requirement details - MODIFIED TO INCLUDE STAKEHOLDERS
         pdf.set_font("Arial", 'B', 14)
-        pdf.multi_cell(effective_width, 8, txt=f"Requirement {i}: {result['Requirement']}")
+        pdf.multi_cell(200, 10, txt=f"Requirement R{i}: {result['Requirement']}", align='L')
         pdf.set_font("Arial", size=12)
-        pdf.multi_cell(effective_width, 8, txt=f"Type: {result['Type']}")
-        pdf.multi_cell(effective_width, 8, txt=f"Domain: {result['Domain']}")
-        pdf.multi_cell(effective_width, 8, txt=f"Stakeholders: {result['Stakeholders']}")
-        pdf.multi_cell(effective_width, 8, txt=f"Defects: {result['Defects']}")
-        pdf.multi_cell(effective_width, 8, txt=f"Rewritten: {result['Rewritten']}")
-        pdf.ln(5)
-        pdf.set_draw_color(200)
-        pdf.cell(effective_width, 0, border='T')
-        pdf.ln(8)
+        pdf.multi_cell(200, 10, txt=f"Type: {result['Type']}", align='L')
+        pdf.multi_cell(200, 10, txt=f"Domain: {result['Domain']}", align='L')
+        pdf.multi_cell(200, 10, txt=f"Stakeholders: {result['Stakeholders']}", align='L')  # Added this line
+        pdf.multi_cell(200, 10, txt=f"Defects: {result['Defects']}", align='L')
+        pdf.multi_cell(200, 10, txt=f"Rewritten: {result['Rewritten']}", align='L')
+        pdf.multi_cell(200, 10, txt="-" * 50, align='L')
+        pdf.ln(5)  # Add some space between requirements
 
     pdf_output = "requirements_report.pdf"
     pdf.output(pdf_output)