Update app.py

app.py

@@ -1 +1,66 @@
- Document (FSD) generated successfully: {fsd_path}")
+import openai
+import matplotlib.pyplot as plt
+from docx import Document
+from docx.shared import Inches
+from huggingface_hub import get_secret_value  # Import secret retrieval function
+
+# ✅ Retrieve API key securely from Hugging Face secret store
+try:
+    OPENAI_API_KEY = get_secret_value("OPENAI_API_KEY")  # Fetch stored API key
+except Exception as e:
+    raise RuntimeError("Failed to retrieve OpenAI API Key from Hugging Face Secrets.") from e
+
+# ✅ Configure OpenAI API Key
+openai.api_key = OPENAI_API_KEY  
+
+def generate_text(prompt):
+    """Generate text using OpenAI GPT securely"""
+    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"]
+
+# ✅ Create 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")
+
+# ✅ Generate FSD Content
+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.")
+doc.add_paragraph(intro_text)
+
+# ✅ Add 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.set_xlim(0, 1)
+ax.set_ylim(0, 1)
+ax.set_xticks([])
+ax.set_yticks([])
+ax.set_frame_on(False)
+
+diagram_path = "/mnt/data/banking_system_architecture.png"
+plt.savefig(diagram_path)
+plt.close(fig)
+
+doc.add_picture(diagram_path, width=Inches(5))
+
+# ✅ Save the Word Document
+fsd_path = "/mnt/data/Banking_FSD.docx"
+doc.save(fsd_path)
+
+print(f"✅ Functional Specification Document (FSD) generated successfully: {fsd_path}")