Update app.py

app.py

@@ -1,28 +1,35 @@
-import openai  # OpenAI API for GenAI-generated text
+import os
+import openai
 import matplotlib.pyplot as plt
 from docx import Document
 from docx.shared import Inches
-
-# Set your OpenAI API Key
-OPENAI_API_KEY = "your-openai-api-key"
-
-def generate_text(prompt):
-    """Generate text using OpenAI API"""
-    response = openai.ChatCompletion.create(
-        model="gpt-4",
-        messages=[{"role": "system", "content": "You are an expert banking analyst."},
-                  {"role": "user", "content": prompt}],
-        api_key=OPENAI_API_KEY
-    )
-    return response["choices"][0]["message"]["content"]
-
-# Create a Word document
+from datetime import datetime
+
+# Set your OpenAI API Key securely using environment variables
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+def generate_text(prompt, role="user"):
+    """Generate text using the new OpenAI API format."""
+    try:
+        response = openai.ChatCompletion.acreate(
+            model="gpt-4",
+            messages=[
+                {"role": "system", "content": "You are an expert banking analyst."},
+                {"role": role, "content": prompt}
+            ]
+        )
+        return response['choices'][0]['message']['content']
+    except Exception as e:
+        print(f"Error generating text: {e}")
+        return "(*Failed to generate content. Please review manually.*)"
+
+# Create the Word document
 doc = Document()
 doc.add_heading('Functional Specification Document (FSD)', level=1)
 doc.add_paragraph("**Project Name:** Banking Loan & Fraud Detection System")
 doc.add_paragraph("**Version:** 1.0")
 doc.add_paragraph("**Prepared By:** Your Name")
-doc.add_paragraph("**Date:** DD/MM/YYYY")
+doc.add_paragraph(f"**Date:** {datetime.now().strftime('%d/%m/%Y')}")
 
 # Table of Contents
 doc.add_page_break()
@@ -60,6 +67,11 @@ doc.add_heading("3. Functional Requirements", level=2)
 functional_reqs = generate_text("List functional requirements for a banking fraud detection and loan system.")
 doc.add_paragraph(functional_reqs)
 
+# Non-Functional Requirements
+doc.add_heading("4. Non-Functional Requirements", level=2)
+non_functional_reqs = generate_text("List non-functional requirements for a banking loan and fraud detection system.")
+doc.add_paragraph(non_functional_reqs)
+
 # Security & Compliance
 doc.add_heading("5. Security & Compliance", level=2)
 security_text = generate_text("What are the security and compliance requirements for a banking loan system?")
@@ -68,11 +80,16 @@ doc.add_paragraph(security_text)
 # Generate System Architecture Diagram
 doc.add_heading("6. System Architecture Diagram", level=2)
 fig, ax = plt.subplots(figsize=(6, 4))
-ax.text(0.5, 0.85, "Customer", fontsize=12, ha="center", bbox=dict(facecolor="lightblue", edgecolor="black"))
-ax.text(0.5, 0.65, "Banking System", fontsize=12, ha="center", bbox=dict(facecolor="lightgreen", edgecolor="black"))
-ax.text(0.3, 0.45, "Loan Processing", fontsize=12, ha="center", bbox=dict(facecolor="lightyellow", edgecolor="black"))
-ax.text(0.7, 0.45, "Fraud Detection", fontsize=12, ha="center", bbox=dict(facecolor="lightpink", edgecolor="black"))
-ax.text(0.5, 0.25, "Database", fontsize=12, ha="center", bbox=dict(facecolor="lightgray", edgecolor="black"))
+ax.text(0.5, 0.85, "Customer", fontsize=12, ha="center", 
+        bbox=dict(facecolor="lightblue", edgecolor="black"))
+ax.text(0.5, 0.65, "Banking System", fontsize=12, ha="center", 
+        bbox=dict(facecolor="lightgreen", edgecolor="black"))
+ax.text(0.3, 0.45, "Loan Processing", fontsize=12, ha="center", 
+        bbox=dict(facecolor="lightyellow", edgecolor="black"))
+ax.text(0.7, 0.45, "Fraud Detection", fontsize=12, ha="center", 
+        bbox=dict(facecolor="lightpink", edgecolor="black"))
+ax.text(0.5, 0.25, "Database", fontsize=12, ha="center", 
+        bbox=dict(facecolor="lightgray", edgecolor="black"))
 
 ax.set_xlim(0, 1)
 ax.set_ylim(0, 1)
@@ -80,8 +97,10 @@ ax.set_xticks([])
 ax.set_yticks([])
 ax.set_frame_on(False)
 
-# Save diagram
-diagram_path = "banking_system_architecture.png"
+# Save diagram in a designated output folder
+output_folder = "output"
+os.makedirs(output_folder, exist_ok=True)
+diagram_path = os.path.join(output_folder, "banking_system_architecture.png")
 plt.savefig(diagram_path)
 plt.close(fig)
 
@@ -96,7 +115,31 @@ doc.add_paragraph("(*Placeholder for detailed banking process flow diagrams*)")
 doc.add_heading("8. UI Mockups", level=2)
 doc.add_paragraph("(*Placeholder for UI Mockups for loan applications, fraud alerts, and account dashboards*)")
 
+# Assumptions and Constraints
+doc.add_heading("9. Assumptions and Constraints", level=2)
+assumptions_text = generate_text("List assumptions and constraints for a banking loan and fraud detection system.")
+doc.add_paragraph(assumptions_text)
+
+# Dependencies
+doc.add_heading("10. Dependencies", level=2)
+dependencies_text = generate_text("List the dependencies for a banking loan and fraud detection system project.")
+doc.add_paragraph(dependencies_text)
+
+# Acceptance Criteria
+doc.add_heading("11. Acceptance Criteria", level=2)
+acceptance_criteria = generate_text("Define acceptance criteria for a banking loan and fraud detection system.")
+doc.add_paragraph(acceptance_criteria)
+
+# Glossary
+doc.add_heading("12. Glossary", level=2)
+doc.add_paragraph("(*Placeholder for glossary of terms related to banking, loans, and fraud detection*)")
+
+# References
+doc.add_heading("13. References", level=2)
+doc.add_paragraph("(*Placeholder for references including documents, articles, and regulatory guidelines*)")
+
 # Save the Word Document
-doc.save("Banking_FSD.docx")
+doc_path = os.path.join(output_folder, "Banking_FSD.docx")
+doc.save(doc_path)
 
-print("Functional Specification Document (FSD) generated successfully as 'Banking_FSD.docx'")
+print(f"Functional Specification Document (FSD) generated successfully at: {doc_path}")