Update app.py

app.py

@@ -61,27 +61,34 @@ def generate_functional_requirements(topic):
     if generator is None:
         return "Error: Model failed to load."
 
-    prompt = f"""
-    Generate a **detailed functional requirements document** for **{topic}** in the banking sector.
-    Ensure it includes the following sections:
-    
-    1️⃣ **Introduction**: Overview, Purpose, Intended Users
-    2️⃣ **Scope**: System Description, Key Functionalities
-    3️⃣ **Functional Specifications**: Features, User Roles, Transactions
-    4️⃣ **Security & Compliance**: Regulations, Data Protection
-    5️⃣ **Conclusion**: Summary, Future Enhancements
-    """
+    # Structured Sections for Detailed Document
+    sections = [
+        "Introduction: Overview, Purpose, Intended Users",
+        "Scope: System Description, Key Functionalities",
+        "Functional Specifications: Features, User Roles, Transactions",
+        "Security & Compliance: Regulations, Data Protection",
+        "User Interface Requirements: Wireframes, UI/UX Considerations",
+        "System Architecture: High-Level Architecture, Technology Stack",
+        "Performance & Scalability: Expected Load, Response Time, Uptime",
+        "Regulatory & Legal Compliance: Banking Regulations, Audits",
+        "Data Management: Data Flow, Storage, Backup Strategies",
+        "Conclusion: Summary, Future Enhancements"
+    ]
 
     document = ""  # Initialize empty document
     
-    for _ in range(3):  # Generate in chunks (Avoids memory crash)
-        output = generator(prompt, max_length=1024, do_sample=True, temperature=0.7)
+    # Generate Detailed Content for Each Section
+    for section in sections:
+        prompt = f"Generate a **detailed and structured** section on '{section}' for **{topic}** in banking."
         
-        if output and "generated_text" in output[0]:
-            document += output[0]['generated_text'] + "\n\n"
-        else:
-            return "Error: Model failed to generate text."
-    
+        for _ in range(3):  # Generate more content for each section
+            output = generator(prompt, max_length=2048, do_sample=True, temperature=0.7)
+            
+            if output and "generated_text" in output[0]:
+                document += f"### {section}\n\n" + output[0]['generated_text'] + "\n\n"
+            else:
+                return "Error: Model failed to generate text."
+
     return document
 
 # 📌 Function to Save Generated Content as PDF
@@ -93,7 +100,14 @@ def save_to_pdf(content, filename):
     text.setLeading(14)
     
     for line in content.split("\n"):
+        if "### " in line:  # Section Header - Start New Page
+            c.showPage()
+            text = c.beginText(40, 750)
+            text.setLeading(14)
+            c.setFont("Helvetica-Bold", 12)
+
         text.textLine(line)
+        
         if text.getY() < 50:  # Handle page overflow
             c.drawText(text)
             c.showPage()
@@ -130,4 +144,4 @@ def main():
                 os.remove(filename)  # Cleanup file after download
 
 if __name__ == "__main__":
-    main()
+    main()