Update app.py

app.py

@@ -2,7 +2,6 @@ import streamlit as st
 from transformers import pipeline
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
-from reportlab.lib import colors
 from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, PageBreak
 import os
@@ -32,47 +31,24 @@ def generate_functional_requirements(topic):
 
     sections = {
         "Introduction": [
-            "Overview and Background",
-            "Purpose of the System",
-            "Intended Users and Audience"
+            "Overview and Purpose",
+            "Intended Users"
         ],
         "Scope": [
-            "General System Description",
-            "Key Functionalities & Modules",
-            "Out of Scope Features"
+            "System Description",
+            "Key Functionalities"
         ],
         "Functional Specifications": [
-            "User Roles & Access Levels",
-            "Main Features and Use Cases",
-            "Transaction Handling & Workflows"
+            "User Roles",
+            "Core Features"
         ],
         "Security & Compliance": [
             "Regulatory Requirements",
-            "Data Protection & Encryption",
-            "Fraud Detection Mechanisms"
-        ],
-        "User Interface": [
-            "Wireframes and UI/UX Considerations",
-            "Navigation & Accessibility Features"
-        ],
-        "System Architecture": [
-            "High-Level Architecture",
-            "Technology Stack Used",
-            "API Integrations"
-        ],
-        "Performance & Scalability": [
-            "Expected Load & System Throughput",
-            "Response Time Expectations",
-            "Scalability Strategies"
-        ],
-        "Data Management": [
-            "Data Flow and Storage",
-            "Backup & Disaster Recovery Strategies"
+            "Data Protection"
         ],
         "Future Enhancements": [
             "Potential Feature Expansions",
-            "Long-Term Roadmap",
-            "Next Steps in Development"
+            "Roadmap & Next Steps"
         ]
     }
 
@@ -84,18 +60,17 @@ def generate_functional_requirements(topic):
         document.append(Spacer(1, 10))
 
         for subsection in subsections:
-            prompt = f"Write a **detailed 500-word section** on '{subsection}' for the topic '{topic}' in banking. Provide structured paragraphs with examples."
-
-            for _ in range(2):  # Increase iterations per subsection
-                output = generator(prompt, max_length=2048, do_sample=True, temperature=0.7)
-
-                if output and isinstance(output, list) and len(output) > 0 and "generated_text" in output[0]:
-                    document.append(Paragraph(f"<b>{subsection}</b>", styles['Heading2']))
-                    document.append(Spacer(1, 6))
-                    document.append(Paragraph(output[0]["generated_text"], styles['Normal']))
-                    document.append(Spacer(1, 10))
-                else:
-                    return "Error: Model failed to generate text."
+            prompt = f"Write a **detailed 300-word section** on '{subsection}' for the topic '{topic}' in banking. Provide structured paragraphs with examples."
+
+            output = generator(prompt, max_length=1024, do_sample=True, temperature=0.7)
+
+            if output and isinstance(output, list) and len(output) > 0 and "generated_text" in output[0]:
+                document.append(Paragraph(f"<b>{subsection}</b>", styles['Heading2']))
+                document.append(Spacer(1, 6))
+                document.append(Paragraph(output[0]["generated_text"], styles['Normal']))
+                document.append(Spacer(1, 10))
+            else:
+                return "Error: Model failed to generate text."
 
         document.append(PageBreak())  # Add a page break after each major section