Update app.py

app.py

@@ -85,58 +85,40 @@ def generate_pdf_report(results):
 
     # Add watermark
     pdf.set_font("Arial", 'B', 50)
-    pdf.set_text_color(230, 230, 230)
-    pdf.rotate(45)
+    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)
+    pdf.rotate(0)  # Reset rotation
 
     # Add title and date/time
     pdf.set_font("Arial", 'B', 16)
-    pdf.set_text_color(0, 0, 0)
+    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(200, 10, txt=f"Report Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", ln=True, align='C')
-    pdf.ln(10)
+    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:
-            pdf.add_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(200, 10, txt="AI Powered Requirement Analysis and Defect Detection", ln=True, align='C')
             pdf.set_font("Arial", size=12)
             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)
+            pdf.ln(10)  # Add some space
 
+        # Add requirement details
         pdf.set_font("Arial", 'B', 14)
         pdf.multi_cell(200, 10, txt=f"Requirement R{i}: {result['Requirement']}", align='L')
         pdf.set_font("Arial", size=12)
-        
-        # Add Type with colored background
-        pdf.set_fill_color(232, 245, 233)  # Light green for functional
-        if "non-functional" in result['Type'].lower():
-            pdf.set_fill_color(255, 243, 224)  # Light orange for non-functional
-        pdf.cell(40, 10, txt="Type:", ln=0, fill=True)
-        pdf.cell(0, 10, txt=f" {result['Type']}", ln=True)
-        
-        # Add Domain
-        pdf.set_fill_color(227, 242, 253)  # Light blue
-        pdf.cell(40, 10, txt="Domain:", ln=0, fill=True)
-        pdf.cell(0, 10, txt=f" {result['Domain']}", ln=True)
-        
-        # Add Defects
-        pdf.set_fill_color(255, 235, 238)  # Light red
-        pdf.cell(40, 10, txt="Defects:", ln=0, fill=True)
-        pdf.multi_cell(0, 10, txt=f" {result['Defects']}")
-        
-        # Add Rewritten version
-        pdf.set_fill_color(220, 237, 200)  # Light green
-        pdf.cell(40, 10, txt="Improved:", ln=0, fill=True)
-        pdf.multi_cell(0, 10, txt=f" {result['Rewritten']}")
-        
+        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"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)
+        pdf.ln(5)  # Add some space between requirements
 
     pdf_output = "requirements_report.pdf"
     pdf.output(pdf_output)