Update app.py

app.py

@@ -1,28 +1,42 @@
+import os
 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
+from huggingface_hub import HfApi
 
-# ✅ 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
+# ✅ 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  
+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"]
+    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
 doc = Document()