Update app.py

app.py

@@ -83,10 +83,11 @@ def analyze_requirement(requirement):
     }
 
 # Function to generate a PDF report
-def generate_pdf(req_text, req_type, domain, stakeholders, defects, rewrite):
+def generate_pdf_report(results):
     pdf = FPDF()
     pdf.add_page()
-    
+    pdf.set_font("Arial", size=12)
+
     # Add watermark
     pdf.set_font("Arial", 'B', 50)
     pdf.set_text_color(230, 230, 230)  # Light gray color for watermark
@@ -113,15 +114,21 @@ def generate_pdf(req_text, req_type, domain, stakeholders, defects, rewrite):
             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
 
-        pdf.set_font("Arial", '', 12)
-        pdf.multi_cell(0, 10, f"📌 Type of Requirements: {req_type}")
-        pdf.multi_cell(0, 10, f"🌐 Domain of Requirements: {domain}")
-        pdf.multi_cell(0, 10, f"👥 Stakeholders: {stakeholders}")
-        pdf.multi_cell(0, 10, f"🐞 Defects Found: {defects}")
-        pdf.multi_cell(0, 10, f"✍️ Rewritten Requirements: {rewrite}")
+        # 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)
+        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')
+        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
 
-    
-    return pdf.output(dest='S').encode('latin1')  # Return PDF bytes
+    pdf_output = "requirements_report.pdf"
+    pdf.output(pdf_output)
+    return pdf_output
 
 # Custom CSS for professional styling
 st.markdown("""
@@ -248,16 +255,24 @@ def main():
                     """, unsafe_allow_html=True)
 
         # PDF Report Section
-        if st.button("📄 Generate Report"):
-            pdf_bytes = generate_pdf(
-                requirements_text,
-                requirement_type,
-                requirement_domain,
-                requirement_stakeholders,
-                requirement_defects,
-                requirement_rewrite
-            )
-            st.download_button("⬇️ Download PDF", data=pdf_bytes, file_name="Requirements_Report.pdf", mime="application/pdf")
+        st.subheader("📤 Generate Report")
+        with st.container():
+            col1, col2 = st.columns([3, 2])
+            with col1:
+                st.info("💡 Click below to generate a comprehensive PDF report with all analysis details")
+            with col2:
+                if st.button("📄 Generate PDF Report", type="secondary", use_container_width=True):
+                    with st.spinner("Generating PDF..."):
+                        pdf_report = generate_pdf_report(st.session_state.results)
+                        with open(pdf_report, "rb") as f:
+                            st.download_button(
+                                label="⬇️ Download Full Report",
+                                data=f,
+                                file_name="Requirement_Analysis_Report.pdf",
+                                mime="application/pdf",
+                                use_container_width=True,
+                                type="primary"
+                            )
 
     # Footer
     st.markdown("---")