Update app.py

app.py

@@ -3,6 +3,7 @@ import requests
 from fpdf import FPDF
 import os
 import time
+from datetime import datetime
 
 # Mistral API key (replace with your key or use environment variable)
 api_key = os.getenv("MISTRAL_API_KEY", "gz6lDXokxgR6cLY72oomALWcm7vhjRzQ")
@@ -82,20 +83,40 @@ def rewrite_requirement(requirement, defects):
     response = call_mistral_api(prompt)
     return response.strip()
 
-# Function to generate a PDF report
+# Function to generate a PDF report with professional formatting
 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
+    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)  # Black color for title
+    pdf.cell(200, 10, txt="AI Powered Requirement Analysis and Defect Detection using LLM Model Mistral", 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)  # Add some space
+
+    # Add requirements analysis
+    pdf.set_font("Arial", size=12)
     for i, result in enumerate(results, start=1):
+        pdf.set_font("Arial", 'B', 14)
         pdf.cell(200, 10, txt=f"Requirement R{i}: {result['Requirement']}", ln=True)
+        pdf.set_font("Arial", size=12)
         pdf.cell(200, 10, txt=f"Type: {result['Type']}", ln=True)
         pdf.cell(200, 10, txt=f"Stakeholders: {result['Stakeholders']}", ln=True)
         pdf.cell(200, 10, txt=f"Domain: {result['Domain']}", ln=True)
         pdf.cell(200, 10, txt=f"Defects: {result['Defects']}", ln=True)
         pdf.cell(200, 10, txt=f"Rewritten: {result['Rewritten']}", ln=True)
         pdf.cell(200, 10, txt="-" * 50, ln=True)
+        pdf.ln(5)  # Add some space between requirements
 
     pdf_output = "requirements_report.pdf"
     pdf.output(pdf_output)
@@ -104,11 +125,10 @@ def generate_pdf_report(results):
 # Streamlit app
 def main():
     st.title("Requirement Analysis Tool")
-    st.markdown("**Team Name:** Sadia, Areeba, Rabbia, Tesmia")
-    st.markdown("**LLM Model:** Mistral")
+    st.markdown("**Team Name:** Team Innovators")
+    st.markdown("**Model:** Mistral")
 
     # Input requirements manually
-    st.markdown("### Enter Requirements")
     input_text = st.text_area("Enter your requirements (one per line or separated by periods):")
     requirements = []
     if input_text:
@@ -148,4 +168,4 @@ def main():
 
 # Run the app
 if __name__ == "__main__":
-    main()
+    main()