Update app.py

app.py

@@ -2,25 +2,45 @@ import os
 import openai
 import matplotlib.pyplot as plt
 from docx import Document
-from docx.shared import Inches
+from docx.shared import Inches, Pt
+from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
 from datetime import datetime
 import streamlit as st
-from openai import OpenAI  # New import style
+from openai import OpenAI
 
 # Initialize OpenAI client
 client = OpenAI(api_key=st.secrets.get("OPENAI_API_KEY"))
 
-# Check if the API key is loaded correctly
-if not client.api_key:
-    st.error("OpenAI API key is missing in Streamlit secrets.")
-else:
-    st.write("OpenAI API Key loaded successfully.")
+# Custom CSS for styling
+st.markdown("""
+<style>
+    .stTextInput>div>div>input, .stTextArea>div>div>textarea {
+        background-color: #f0f2f6;
+        border-radius: 5px;
+    }
+    .stButton>button {
+        background-color: #4CAF50;
+        color: white;
+        padding: 10px 24px;
+        border-radius: 5px;
+        border: none;
+    }
+    .stButton>button:hover {
+        background-color: #45a049;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+def init_session_state():
+    if 'current_step' not in st.session_state:
+        st.session_state.current_step = 1
+    if 'user_inputs' not in st.session_state:
+        st.session_state.user_inputs = {}
 
 def generate_text(prompt, role="user"):
-    """Generate text using the OpenAI API (v1.0+ compatible)"""
     try:
         response = client.chat.completions.create(
-            model="gpt-4-1106-preview",  # Updated model name
+            model="gpt-4-1106-preview",
             messages=[
                 {"role": "system", "content": "You are an expert banking analyst."},
                 {"role": role, "content": prompt}
@@ -29,134 +49,157 @@ def generate_text(prompt, role="user"):
         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.*)"
-
-# 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")
-doc.add_paragraph("**Version:** 1.0")
-doc.add_paragraph("**Prepared By:** Your Name")
-doc.add_paragraph(f"**Date:** {datetime.now().strftime('%d/%m/%Y')}")
-
-# Table of Contents
-doc.add_page_break()
-doc.add_heading("Table of Contents", level=2)
-toc = [
-    "1. Introduction",
-    "2. System Overview",
-    "3. Functional Requirements",
-    "4. Non-Functional Requirements",
-    "5. Security & Compliance",
-    "6. System Architecture Diagram",
-    "7. Process Flow Diagrams",
-    "8. UI Mockups",
-    "9. Assumptions and Constraints",
-    "10. Dependencies",
-    "11. Acceptance Criteria",
-    "12. Glossary",
-    "13. References"
-]
-for item in toc:
-    doc.add_paragraph(item)
-
-# Generate Content Using AI
-doc.add_page_break()
-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)
-
-doc.add_heading("2. System Overview", level=2)
-overview_text = generate_text("Write a detailed system overview of a banking fraud detection and loan system.")
-doc.add_paragraph(overview_text)
-
-# Functional Requirements
-doc.add_heading("3. Functional Requirements", level=2)
-functional_reqs = generate_text("List functional requirements for a banking fraud detection and loan system.")
-doc.add_paragraph(functional_reqs)
-
-# Non-Functional Requirements
-doc.add_heading("4. Non-Functional Requirements", level=2)
-non_functional_reqs = generate_text("List non-functional requirements for a banking loan and fraud detection system.")
-doc.add_paragraph(non_functional_reqs)
-
-# Security & Compliance
-doc.add_heading("5. Security & Compliance", level=2)
-security_text = generate_text("What are the security and compliance requirements for a banking loan system?")
-doc.add_paragraph(security_text)
-
-# Generate 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)
-
-# Save diagram
-output_folder = "output"
-os.makedirs(output_folder, exist_ok=True)
-
-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 into document
-doc.add_picture(diagram_path, width=Inches(5))
-
-# Process Flow Diagrams
-doc.add_heading("7. Process Flow Diagrams", level=2)
-doc.add_paragraph("(*Placeholder for detailed banking process flow diagrams*)")
-
-# UI Mockups
-doc.add_heading("8. UI Mockups", level=2)
-doc.add_paragraph("(*Placeholder for UI Mockups for loan applications, fraud alerts, and account dashboards*)")
-
-# Assumptions and Constraints
-doc.add_heading("9. Assumptions and Constraints", level=2)
-assumptions_text = generate_text("List assumptions and constraints for a banking loan and fraud detection system.")
-doc.add_paragraph(assumptions_text)
-
-# Dependencies
-doc.add_heading("10. Dependencies", level=2)
-dependencies_text = generate_text("List the dependencies for a banking loan and fraud detection system project.")
-doc.add_paragraph(dependencies_text)
-
-# Acceptance Criteria
-doc.add_heading("11. Acceptance Criteria", level=2)
-acceptance_criteria = generate_text("Define acceptance criteria for a banking loan and fraud detection system.")
-doc.add_paragraph(acceptance_criteria)
-
-# Glossary
-doc.add_heading("12. Glossary", level=2)
-doc.add_paragraph("(*Placeholder for glossary of terms related to banking, loans, and fraud detection*)")
-
-# References
-doc.add_heading("13. References", level=2)
-doc.add_paragraph("(*Placeholder for references including documents, articles, and regulatory guidelines*)")
-
-# Save Document
-doc_path = os.path.join(output_folder, "Banking_FSD.docx")
-doc.save(doc_path)
-
-if os.path.exists(doc_path):
-    st.write(f"FSD generated successfully at: {doc_path}")
-else:
-    st.error("Failed to generate the FSD document.")
+        return "(*Content generation failed. Please fill manually.*)"
+
+def create_fsd_document(inputs):
+    doc = Document()
+    
+    # Set default font
+    style = doc.styles['Normal']
+    font = style.font
+    font.name = 'Calibri'
+    font.size = Pt(11)
+
+    # Title Page
+    doc.add_heading('Functional Specification Document (FSD)', 0)
+    doc.add_paragraph("Project Name: " + inputs.get('project_name', 'Banking System'))
+    doc.add_paragraph(f"Version: {inputs.get('version', '1.0')}")
+    doc.add_paragraph(f"Prepared By: {inputs.get('author', 'Your Name')}")
+    doc.add_paragraph(f"Date: {datetime.now().strftime('%d/%m/%Y')}")
+    doc.add_page_break()
+
+    # Table of Contents
+    doc.add_heading("Table of Contents", 1)
+    toc = [
+        "1. Introduction",
+        "2. System Overview",
+        "3. Functional Requirements",
+        "4. Non-Functional Requirements",
+        "5. Security & Compliance",
+        "6. System Architecture",
+        "7. Process Flows",
+        "8. Data Requirements",
+        "9. User Interface",
+        "10. Reporting",
+        "11. Error Handling",
+        "12. Assumptions",
+        "13. Dependencies",
+        "14. Glossary"
+    ]
+    for item in toc:
+        doc.add_paragraph(item, style='ListBullet')
+    doc.add_page_break()
+
+    # Detailed Sections
+    sections = {
+        "1. Introduction": f"Write a comprehensive introduction for {inputs['project_name']} covering: {inputs['brief_description']}",
+        "2. System Overview": f"Create detailed system overview for {inputs['project_name']} including: {inputs['system_components']}",
+        "3. Functional Requirements": f"Expand these functional requirements into detailed specifications: {inputs['functional_reqs']}",
+        "4. Non-Functional Requirements": f"Create non-functional requirements for: {inputs['non_functional_reqs']}",
+        "5. Security & Compliance": f"Detail security measures and compliance requirements for: {inputs['security_needs']}",
+    }
+
+    for header, prompt in sections.items():
+        doc.add_heading(header, level=1)
+        content = generate_text(prompt)
+        for paragraph in content.split('\n'):
+            if paragraph.strip():
+                p = doc.add_paragraph(paragraph)
+                p.paragraph_format.space_after = Pt(6)
+
+        # Add detailed subsections
+        if header == "3. Functional Requirements":
+            doc.add_heading("3.1 Detailed Use Cases", level=2)
+            use_cases = generate_text(f"Create detailed use cases for: {inputs['functional_reqs']}")
+            doc.add_paragraph(use_cases)
+
+        doc.add_page_break()
+
+    # Add diagrams
+    doc.add_heading("6. System Architecture", 1)
+    fig = create_architecture_diagram()
+    diagram_path = "system_architecture.png"
+    fig.savefig(diagram_path)
+    doc.add_picture(diagram_path, width=Inches(6))
+    
+    return doc
+
+def create_architecture_diagram():
+    fig, ax = plt.subplots(figsize=(10, 6))
+    components = [
+        ("User Interface", 0.5, 0.9),
+        ("Application Layer", 0.3, 0.7),
+        ("Business Logic", 0.7, 0.7),
+        ("Database", 0.5, 0.5),
+        ("External APIs", 0.5, 0.3)
+    ]
+    
+    for text, x, y in components:
+        ax.text(x, y, text, 
+                ha='center', va='center',
+                bbox=dict(facecolor='lightblue', edgecolor='black', boxstyle='round,pad=0.5'),
+                fontsize=10)
+    
+    ax.set_xlim(0, 1)
+    ax.set_ylim(0, 1)
+    ax.axis('off')
+    return fig
+
+def main():
+    init_session_state()
+    st.title("FSD Document Generator")
+    st.subheader("Create Custom Functional Specification Documents")
+
+    if st.session_state.current_step == 1:
+        with st.form("basic_info"):
+            st.header("Step 1: Basic Information")
+            st.session_state.user_inputs['project_name'] = st.text_input("Project Name")
+            st.session_state.user_inputs['brief_description'] = st.text_area("Brief Description", height=150)
+            if st.form_submit_button("Next →"):
+                st.session_state.current_step = 2
+                st.rerun()
+
+    elif st.session_state.current_step == 2:
+        with st.form("detailed_info"):
+            st.header("Step 2: Detailed Requirements")
+            st.session_state.user_inputs['functional_reqs'] = st.text_area(
+                "Functional Requirements (comma-separated)",
+                height=100,
+                value="User authentication, Transaction monitoring, Report generation"
+            )
+            st.session_state.user_inputs['non_functional_reqs'] = st.text_input(
+                "Non-Functional Requirements",
+                value="High availability, GDPR compliance"
+            )
+            st.session_state.user_inputs['security_needs'] = st.text_input(
+                "Security Requirements",
+                value="Data encryption, Role-based access control"
+            )
+            if st.form_submit_button("Generate FSD"):
+                st.session_state.current_step = 3
+                st.rerun()
+
+    elif st.session_state.current_step == 3:
+        st.header("Step 3: Generated Document")
+        with st.spinner("Generating 10+ page FSD document..."):
+            doc = create_fsd_document(st.session_state.user_inputs)
+            output_folder = "generated_docs"
+            os.makedirs(output_folder, exist_ok=True)
+            doc_path = os.path.join(output_folder, "FSD_Document.docx")
+            doc.save(doc_path)
+            
+            with open(doc_path, "rb") as file:
+                st.download_button(
+                    label="Download FSD Document",
+                    data=file,
+                    file_name="FSD_Document.docx",
+                    mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+                )
+            
+            if st.button("Start New Document"):
+                st.session_state.current_step = 1
+                st.session_state.user_inputs = {}
+                st.rerun()
+
+if __name__ == "__main__":
+    main()