Update app.py

app.py

@@ -1,5 +1,5 @@
-import openai
 import os
+import openai
 import matplotlib.pyplot as plt
 from docx import Document
 from docx.shared import Inches
@@ -10,8 +10,14 @@ import streamlit as st
 openai_api_key = st.secrets["OPENAI_API_KEY"]
 openai.api_key = openai_api_key  # Set the OpenAI API key
 
+# Check if the API key is loaded correctly
+if not openai_api_key:
+    st.error("OpenAI API key is missing in Streamlit secrets.")
+else:
+    st.write("OpenAI API Key loaded successfully.")
+
 def generate_text(prompt, role="user"):
-    """Generate text using the new OpenAI API format."""
+    """Generate text using the OpenAI API."""
     try:
         response = openai.ChatCompletion.acreate(
             model="gpt-4",
@@ -22,7 +28,7 @@ def generate_text(prompt, role="user"):
         )
         return response['choices'][0]['message']['content']
     except Exception as e:
-        print(f"Error generating text: {e}")
+        st.error(f"Error generating text: {e}")
         return "(*Failed to generate content. Please review manually.*)"
 
 # Create the Word document
@@ -102,11 +108,18 @@ ax.set_frame_on(False)
 # Save diagram in a designated output folder
 output_folder = "output"
 os.makedirs(output_folder, exist_ok=True)
+
+# Check if the folder is created
+if os.path.exists(output_folder):
+    st.write(f"Output folder exists: {output_folder}")
+else:
+    st.error(f"Failed to create output folder: {output_folder}")
+
 diagram_path = os.path.join(output_folder, "banking_system_architecture.png")
 plt.savefig(diagram_path)
 plt.close(fig)
 
-# Insert diagram
+# Insert diagram into the document
 doc.add_picture(diagram_path, width=Inches(5))
 
 # Process Flow Diagrams
@@ -144,4 +157,8 @@ doc.add_paragraph("(*Placeholder for references including documents, articles, a
 doc_path = os.path.join(output_folder, "Banking_FSD.docx")
 doc.save(doc_path)
 
-print(f"Functional Specification Document (FSD) generated successfully at: {doc_path}")
+# Confirm file creation
+if os.path.exists(doc_path):
+    st.write(f"Functional Specification Document (FSD) generated successfully at: {doc_path}")
+else:
+    st.error("Failed to generate the FSD document.")