Create app.py

app.py

@@ -0,0 +1,61 @@
+import streamlit as st
+from transformers import pipeline
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+import random
+
+def generate_functional_requirements(topic):
+    generator = pipeline("text-generation", model="meta-llama/Llama-2-7b-chat-hf")
+    prompt = f"Generate a comprehensive functional requirement document for {topic} in the banking sector."
+    
+    # Generate content
+    output = generator(prompt, max_length=4096, do_sample=True, temperature=0.8)
+    return output[0]['generated_text']
+
+def save_to_pdf(content, filename):
+    c = canvas.Canvas(filename, pagesize=letter)
+    c.setFont("Helvetica", 12)
+    
+    text = c.beginText(40, 750)
+    text.setFont("Helvetica", 10)
+    text.setLeading(14)
+    
+    for line in content.split("\n"):
+        text.textLine(line)
+        if text.getY() < 50:  # Start new page when space runs out
+            c.drawText(text)
+            c.showPage()
+            text = c.beginText(40, 750)
+            text.setFont("Helvetica", 10)
+            text.setLeading(14)
+    
+    c.drawText(text)
+    c.save()
+
+def main():
+    st.title("AI-Powered Functional Requirement Generator for Banking")
+    
+    banking_topics = [
+        "Core Banking System",
+        "Loan Management System",
+        "Payment Processing Gateway",
+        "Risk and Fraud Detection",
+        "Regulatory Compliance Management",
+        "Digital Banking APIs",
+        "Customer Onboarding & KYC",
+        "Treasury Management",
+        "Wealth & Portfolio Management"
+    ]
+    
+    topic = st.selectbox("Select a Banking Functional Requirement Topic", banking_topics)
+    
+    if st.button("Generate Functional Requirement Document"):
+        with st.spinner("Generating... This may take a while."):
+            content = generate_functional_requirements(topic)
+            filename = "functional_requirement.pdf"
+            save_to_pdf(content, filename)
+            st.success("Functional requirement document generated successfully!")
+            st.download_button(label="Download PDF", data=open(filename, "rb"), file_name=filename, mime="application/pdf")
+
+if __name__ == "__main__":
+    main()