Update app.py

app.py

@@ -82,6 +82,7 @@ def analyze_requirement(requirement):
         "Rewritten": rewritten
     }
 
+# Function to generate a PDF report
 def generate_pdf_report(results):
     pdf = FPDF()
     pdf.add_page()
@@ -113,13 +114,12 @@ 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 - MODIFIED TO INCLUDE STAKEHOLDERS
+        # 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')  # 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')
@@ -129,7 +129,6 @@ def generate_pdf_report(results):
     pdf.output(pdf_output)
     return pdf_output
 
-
 # Custom CSS for professional styling
 st.markdown("""
     <style>
@@ -226,7 +225,6 @@ def main():
                 time.sleep(0.5)
                 st.session_state.results = results
 
-    
     # Display Results
     if 'results' in st.session_state:
         st.subheader("📊 Analysis Results")
@@ -255,27 +253,27 @@ def main():
                         </div>
                     """, unsafe_allow_html=True)
 
-    # Display result
-    # Generate and download PDF report only if results are present
-    if results:  # Check if results are not empty
-        pdf_report = generate_pdf_report(results)
-        with open(pdf_report, "rb") as f:
-            st.download_button(
-                label="Download PDF Report",
-                data=f,
-                file_name="requirements_report.pdf",
-                mime="application/pdf"
-            )
-    else:
-        st.warning("⚠️ No results to generate a PDF report.")
+        # 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",
+                                use_container_width=True,
+                                type="primary"
+                            )
 
-    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()