Update app.py

app.py

@@ -61,7 +61,6 @@ def generate_functional_requirements(topic):
     if generator is None:
         return "Error: Model failed to load."
 
-    # Structured Sections for Detailed Document
     sections = [
         "Introduction: Overview, Purpose, Intended Users",
         "Scope: System Description, Key Functionalities",
@@ -75,18 +74,22 @@ def generate_functional_requirements(topic):
         "Conclusion: Summary, Future Enhancements"
     ]
 
-    document = ""  # Initialize empty document
-    
-    # Generate Detailed Content for Each Section
+    document = ""  
     for section in sections:
         prompt = f"Generate a **detailed and structured** section on '{section}' for **{topic}** in banking."
         
-        for _ in range(3):  # Generate more content for each section
+        for _ in range(3):
             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"
+
+            if output:
+                st.write("Generated Text:", output)  # Debugging: Check if text is generated
+                if isinstance(output, list) and len(output) > 0 and "generated_text" in output[0]:
+                    document += f"### {section}\n\n" + output[0]["generated_text"] + "\n\n"
+                else:
+                    st.error("Error: Model output format is incorrect.")
+                    return "Error: Model failed to generate text."
             else:
+                st.error("Error: No output from the model.")
                 return "Error: Model failed to generate text."
 
     return document
@@ -99,22 +102,31 @@ def save_to_pdf(content, filename):
     text = c.beginText(40, 750)
     text.setLeading(14)
     
+    page_count = 1  # Track pages to avoid blank PDFs
+    content_written = False  # Ensure at least some content is written
+
     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)
+        if "### " in line:
+            if content_written:
+                c.showPage()
+                text = c.beginText(40, 750)
+                text.setLeading(14)
+                c.setFont("Helvetica-Bold", 12)
 
         text.textLine(line)
+        content_written = True
         
-        if text.getY() < 50:  # Handle page overflow
+        if text.getY() < 50:
             c.drawText(text)
             c.showPage()
             text = c.beginText(40, 750)
             text.setLeading(14)
-    
-    c.drawText(text)
+
+    if content_written:
+        c.drawText(text)
+    else:
+        st.error("Error: No content was written to the PDF.")
+
     c.save()
 
 # 📌 Streamlit UI
@@ -132,16 +144,19 @@ def main():
     if st.button("Generate Functional Requirement Document"):
         with st.spinner("Generating... This may take a while."):
             content = generate_functional_requirements(topic)
-            
+
             if "Error" in content:
                 st.error(content)
             else:
+                # Show document before saving to PDF (Debugging)
+                st.text_area("Generated Document Preview", content, height=400)
+
                 filename = "functional_requirement.pdf"
                 save_to_pdf(content, filename)
                 
                 st.success("✅ Document Generated!")
                 st.download_button("📥 Download PDF", data=open(filename, "rb"), file_name=filename, mime="application/pdf")
-                os.remove(filename)  # Cleanup file after download
+                os.remove(filename)
 
 if __name__ == "__main__":
     main()