Update app.py

app.py

@@ -86,8 +86,7 @@ def analyze_requirement(requirement):
 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
@@ -114,21 +113,16 @@ def generate_pdf_report(results):
             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 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
+        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}")
 
     pdf_output = "requirements_report.pdf"
     pdf.output(pdf_output)
-    return pdf_output
+    return pdf.output(dest='S').encode('latin1')  # Return PDF bytes
 
 # Custom CSS for professional styling
 st.markdown("""
@@ -226,7 +220,6 @@ def main():
                 time.sleep(0.5)
                 st.session_state.results = results
 
-
     # Display Results
     if 'results' in st.session_state:
         st.subheader("📊 Analysis Results")
@@ -256,24 +249,25 @@ def main():
                     """, unsafe_allow_html=True)
 
         # PDF Report Section
-        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"
-                            )
+        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")
 
-    
+    # Footer
+    st.markdown("---")
+    st.markdown("""
+        <div style="text-align: center; color: #666; margin-top: 3rem;">
+            <p>AI-Powered Requirement Analysis System</p>
+            <p>🚀 Powered by Mistral AI & Groq • 🛠️ Developed by Team Four</p>
+        </div>
+    """, unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()