Update app.py

app.py

@@ -1,44 +1,22 @@
-import os
-import openai
+import openai  # OpenAI API for GenAI-generated text
 import matplotlib.pyplot as plt
 from docx import Document
 from docx.shared import Inches
-from huggingface_hub import HfApi
-
-# ✅ Retrieve API key securely from Hugging Face secret store (using environment variables)
-def retrieve_openai_api_key():
-    """Retrieve OpenAI API Key securely from Hugging Face Secrets storage."""
-    # Fetch the API Key using the Hugging Face API
-    api = HfApi()
-    try:
-        openai_api_key = api.get_secret("OPENAI_API_KEY")  # Fetch API key from Hugging Face Secrets
-        if openai_api_key is None:
-            raise ValueError("OPENAI API Key not found in Hugging Face Secrets.")
-        return openai_api_key
-    except Exception as e:
-        raise RuntimeError(f"Error retrieving OpenAI API Key: {str(e)}") from e
-
-# ✅ Retrieve API key from Hugging Face secret store
-OPENAI_API_KEY = retrieve_openai_api_key()
-
-# ✅ Configure OpenAI API Key
-openai.api_key = OPENAI_API_KEY
+
+# Set your OpenAI API Key
+OPENAI_API_KEY = "your-openai-api-key"
 
 def generate_text(prompt):
-    """Generate text using OpenAI GPT securely"""
-    try:
-        response = openai.ChatCompletion.create(
-            model="gpt-4",
-            messages=[
-                {"role": "system", "content": "You are a banking expert creating an FSD."},
-                {"role": "user", "content": prompt}
-            ]
-        )
-        return response["choices"][0]["message"]["content"]
-    except Exception as e:
-        raise RuntimeError(f"Error generating content with OpenAI API: {str(e)}") from e
-
-# ✅ Create Word Document
+    """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
 doc = Document()
 doc.add_heading('Functional Specification Document (FSD)', level=1)
 doc.add_paragraph("**Project Name:** Banking Loan & Fraud Detection System")
@@ -46,13 +24,48 @@ doc.add_paragraph("**Version:** 1.0")
 doc.add_paragraph("**Prepared By:** Your Name")
 doc.add_paragraph("**Date:** DD/MM/YYYY")
 
-# ✅ Generate FSD Content
+# Table of Contents
+doc.add_page_break()
+doc.add_heading("Table of Contents", level=2)
+toc = [
+    "1. Introduction",
+    "2. System Overview",
+    "3. Functional Requirements",
+    "4. Non-Functional Requirements",
+    "5. Security & Compliance",
+    "6. System Architecture Diagram",
+    "7. Process Flow Diagrams",
+    "8. UI Mockups",
+    "9. Assumptions and Constraints",
+    "10. Dependencies",
+    "11. Acceptance Criteria",
+    "12. Glossary",
+    "13. References"
+]
+for item in toc:
+    doc.add_paragraph(item)
+
+# Generate Content Using AI
 doc.add_page_break()
 doc.add_heading("1. Introduction", level=2)
-intro_text = generate_text("Write a detailed introduction for a banking loan and fraud detection system FSD.")
+intro_text = generate_text("Write an introduction for a banking loan and fraud detection system FSD.")
 doc.add_paragraph(intro_text)
 
-# ✅ Add System Architecture Diagram
+doc.add_heading("2. System Overview", level=2)
+overview_text = generate_text("Write a detailed system overview of a banking fraud detection and loan system.")
+doc.add_paragraph(overview_text)
+
+# Functional Requirements
+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)
+
+# 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?")
+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"))
@@ -67,14 +80,23 @@ ax.set_xticks([])
 ax.set_yticks([])
 ax.set_frame_on(False)
 
-diagram_path = "/mnt/data/banking_system_architecture.png"
+# Save diagram
+diagram_path = "banking_system_architecture.png"
 plt.savefig(diagram_path)
 plt.close(fig)
 
+# Insert diagram
 doc.add_picture(diagram_path, width=Inches(5))
 
-# ✅ Save the Word Document
-fsd_path = "/mnt/data/Banking_FSD.docx"
-doc.save(fsd_path)
+# Process Flow Diagrams
+doc.add_heading("7. Process Flow Diagrams", level=2)
+doc.add_paragraph("(*Placeholder for detailed banking process flow diagrams*)")
+
+# UI Mockups
+doc.add_heading("8. UI Mockups", level=2)
+doc.add_paragraph("(*Placeholder for UI Mockups for loan applications, fraud alerts, and account dashboards*)")
+
+# Save the Word Document
+doc.save("Banking_FSD.docx")
 
-print(f"✅ Functional Specification Document (FSD) generated successfully: {fsd_path}")
+print("Functional Specification Document (FSD) generated successfully as 'Banking_FSD.docx'")