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.…