Update app.py

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}")

app.py

@@ -1,65 +1 @@
-import openai
-from huggingface_hub import HfApi
-import matplotlib.pyplot as plt
-from docx import Document
-from docx.shared import Inches
-
-# Initialize Hugging Face API
-api = HfApi()
-
-# Retrieve the API key from Hugging Face Secrets
-OPENAI_API_KEY = api.get_secret("OPENAI_API_KEY")
-
-# Configure OpenAI API Key
-openai.api_key = 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}
-        ]
-    )
-    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")
-doc.add_paragraph("**Version:** 1.0")
-doc.add_paragraph("**Prepared By:** Your Name")
-doc.add_paragraph("**Date:** DD/MM/YYYY")
-
-# Add System Overview
-doc.add_heading("1. Introduction", level=2)
-intro_text = generate_text("Write an introduction for a banking loan and fraud detection system FSD.")
-doc.add_paragraph(intro_text)
-
-# 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 Word Document
-fsd_path = "/mnt/data/Banking_FSD.docx"
-doc.save(fsd_path)
-
-print(f"Functional Specification Document (FSD) generated successfully: {fsd_path}")
+ Document (FSD) generated successfully: {fsd_path}")