Update app.py

app.py

@@ -1,20 +1,15 @@
 import streamlit as st
-from fpdf import FPDF
 
-# Function to create a PDF document
-def generate_pdf(document_text):
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
-    
-    # Title
-    pdf.cell(200, 10, txt="Generated Proposal", ln=True, align="C")
-    
-    # Adding the document text
-    pdf.ln(10)  # Line break
-    pdf.multi_cell(0, 10, document_text)  # Allow wrapping text
-
-    return pdf
+# Function to create various types of proposals with different patterns
+def generate_proposal(document_text, style="formal"):
+    if style == "formal":
+        return f"Dear Sir/Madam,\n\nI am writing to propose the following idea:\n\n{document_text}\n\nSincerely,\nYour Name"
+    elif style == "casual":
+        return f"Hey there!\n\nI’ve been thinking about this idea:\n\n{document_text}\n\nLet me know what you think!\nCheers!"
+    elif style == "business":
+        return f"Subject: Proposal for the Idea\n\nDear [Recipient's Name],\n\nI would like to formally propose the following idea:\n\n{document_text}\n\nLooking forward to your feedback.\nBest regards,\n[Your Name]"
+    else:
+        return f"Proposal:\n\n{document_text}\n\nBest regards,\n[Your Name]"
 
 # Streamlit app
 def main():
@@ -26,26 +21,21 @@ def main():
     # Input text for the proposal
     proposal_text = st.text_area("Enter the content of your proposal:", height=300)
 
+    # Format (pattern) selection
+    st.subheader("Choose the format for your proposal")
+    format_choice = st.selectbox("Select the proposal format", ["Formal", "Casual", "Business"])
+
     # Check if the text is provided and if the button is clicked
     if st.button("Generate Proposal"):
         if proposal_text:
-            # Display the written proposal
-            st.subheader("Your Proposal:")
-            st.markdown(f"```\n{proposal_text}\n```")
-
-            # Generate the PDF
-            pdf = generate_pdf(proposal_text)
-            pdf_output = pdf.output(dest='S').encode('latin1')  # Convert PDF to bytes for download
-
-            # Download button for the generated PDF
-            st.download_button(
-                label="Download Proposal as PDF",
-                data=pdf_output,
-                file_name="proposal.pdf",
-                mime="application/pdf"
-            )
+            # Generate the proposal with the selected format
+            formatted_proposal = generate_proposal(proposal_text, style=format_choice.lower())
+            
+            # Display the generated proposal
+            st.subheader("Your Generated Proposal:")
+            st.markdown(f"```\n{formatted_proposal}\n```")
         else:
             st.warning("Please enter the content of your proposal.")
 
 if __name__ == "__main__":
-    main()
+    main()