Update app.py

app.py

@@ -37,6 +37,7 @@ from transformers import pipeline
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 import os
+import textwrap
 
 # Load Hugging Face Token (Ensure it's set in Env Variables)
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -78,54 +79,43 @@ def generate_functional_requirements(topic):
     for section in sections:
         prompt = f"Generate a **detailed and structured** section on '{section}' for **{topic}** in banking."
         
-        for _ in range(3):
+        for _ in range(3):  # Ensure enough content is generated
             output = generator(prompt, max_length=2048, do_sample=True, temperature=0.7)
 
-            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."
+            if output and isinstance(output, list) and len(output) > 0 and "generated_text" in output[0]:
+                document += f"\n\n### {section}\n\n" + output[0]["generated_text"] + "\n\n"
             else:
-                st.error("Error: No output from the model.")
                 return "Error: Model failed to generate text."
 
     return document
 
 # 📌 Function to Save Generated Content as PDF
 def save_to_pdf(content, filename):
+    if not content.strip():
+        st.error("❌ Error: No content available to write to the PDF.")
+        return
+
     c = canvas.Canvas(filename, pagesize=letter)
     c.setFont("Helvetica", 10)
-    
-    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:
-            if content_written:
-                c.showPage()
-                text = c.beginText(40, 750)
-                text.setLeading(14)
-                c.setFont("Helvetica-Bold", 12)
+    # Handling text wrapping & new pages correctly
+    max_width = 80  # Approximate max characters per line
+    lines_per_page = 50  # Approximate lines per page
+    y_position = 750  # Start position for text
 
-        text.textLine(line)
-        content_written = True
-        
-        if text.getY() < 50:
-            c.drawText(text)
-            c.showPage()
-            text = c.beginText(40, 750)
-            text.setLeading(14)
+    paragraphs = content.split("\n\n")  # Preserve paragraph structure
+
+    for para in paragraphs:
+        wrapped_lines = textwrap.wrap(para, max_width)
+
+        for line in wrapped_lines:
+            if y_position < 50:  # If near bottom, create a new page
+                c.showPage()
+                c.setFont("Helvetica", 10)
+                y_position = 750  # Reset text position
 
-    if content_written:
-        c.drawText(text)
-    else:
-        st.error("Error: No content was written to the PDF.")
+            c.drawString(40, y_position, line)
+            y_position -= 14  # Move to next line
 
     c.save()
 
@@ -148,15 +138,15 @@ def main():
             if "Error" in content:
                 st.error(content)
             else:
-                # Show document before saving to PDF (Debugging)
-                st.text_area("Generated Document Preview", content, height=400)
+                # Show document preview before saving to PDF (for debugging)
+                st.text_area("Generated Document Preview", content[:5000], height=400)
 
                 filename = "functional_requirement.pdf"
                 save_to_pdf(content, filename)
                 
-                st.success("✅ Document Generated!")
+                st.success("✅ Document Generated Successfully!")
                 st.download_button("📥 Download PDF", data=open(filename, "rb"), file_name=filename, mime="application/pdf")
-                os.remove(filename)
+                os.remove(filename)  # Cleanup after download
 
 if __name__ == "__main__":
     main()