Update app.py

app.py

@@ -5,34 +5,33 @@ from docx import Document
 from docx.shared import Inches
 from datetime import datetime
 import streamlit as st
+from openai import OpenAI  # New import style
 
-# Retrieve the OpenAI API Key from Streamlit secrets
-openai_api_key = st.secrets["OPENAI_API_KEY"]
-openai.api_key = openai_api_key  # Set the OpenAI API key
+# Initialize OpenAI client
+client = OpenAI(api_key=st.secrets.get("OPENAI_API_KEY"))
 
 # Check if the API key is loaded correctly
-if not openai_api_key:
+if not client.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 OpenAI API."""
+    """Generate text using the OpenAI API (v1.0+ compatible)"""
     try:
-        # Make a request to OpenAI's new API using `openai.Completion.create()`
-        response = openai.Completion.create(
-            model="gpt-4",  # Use the model you want
+        response = client.chat.completions.create(
+            model="gpt-4-1106-preview",  # Updated model name
             messages=[
                 {"role": "system", "content": "You are an expert banking analyst."},
                 {"role": role, "content": prompt}
             ]
         )
-        return response['choices'][0]['message']['content']
+        return response.choices[0].message.content
     except Exception as e:
         st.error(f"Error generating text: {e}")
         return "(*Failed to generate content. Please review manually.*)"
 
-# Create the Word document
+# Document creation remains the same
 doc = Document()
 doc.add_heading('Functional Specification Document (FSD)', level=1)
 doc.add_paragraph("**Project Name:** Banking Loan & Fraud Detection System")
@@ -106,11 +105,10 @@ ax.set_xticks([])
 ax.set_yticks([])
 ax.set_frame_on(False)
 
-# Save diagram in a designated output folder
+# Save diagram
 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:
@@ -120,7 +118,7 @@ diagram_path = os.path.join(output_folder, "banking_system_architecture.png")
 plt.savefig(diagram_path)
 plt.close(fig)
 
-# Insert diagram into the document
+# Insert diagram into document
 doc.add_picture(diagram_path, width=Inches(5))
 
 # Process Flow Diagrams
@@ -154,12 +152,11 @@ doc.add_paragraph("(*Placeholder for glossary of terms related to banking, loans
 doc.add_heading("13. References", level=2)
 doc.add_paragraph("(*Placeholder for references including documents, articles, and regulatory guidelines*)")
 
-# Save the Word Document
+# Save Document
 doc_path = os.path.join(output_folder, "Banking_FSD.docx")
 doc.save(doc_path)
 
-# Confirm file creation
 if os.path.exists(doc_path):
-    st.write(f"Functional Specification Document (FSD) generated successfully at: {doc_path}")
+    st.write(f"FSD generated successfully at: {doc_path}")
 else:
     st.error("Failed to generate the FSD document.")