Update app.py

app.py

@@ -1,90 +1,15 @@
 import streamlit as st
 from fpdf import FPDF
-import os
 
-# Function to create various types of documents
-def generate_document(doc_type, text, style="formal"):
-    if doc_type == "Proposal":
-        if style == "formal":
-            return f"Dear Sir/Madam,\n\nI am writing to propose the following idea:\n\n{text}\n\nSincerely,\nYour Name"
-        elif style == "casual":
-            return f"Hey there!\n\nI’ve been thinking about this idea:\n\n{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{text}\n\nLooking forward to your feedback.\nBest regards,\n[Your Name]"
-        else:
-            return f"Proposal Idea:\n\n{text}\n\nBest regards,\n[Your Name]"
-
-    elif doc_type == "Letter":
-        return f"Dear [Recipient's Name],\n\n{text}\n\nSincerely,\n[Your Name]"
-
-    elif doc_type == "Report":
-        return f"Report: {text}"
-
-    elif doc_type == "Message":
-        return f"Message:\n\n{text}"
-
-    elif doc_type == "Notice":
-        return f"NOTICE:\n\n{text}"
-
-    else:
-        return f"Document:\n\n{text}"
-
-def chatbot_response(user_input):
-    user_input = user_input.lower()  # Normalize the input to lowercase
-
-    # Friendly and conversational responses
-    if "hello" in user_input or "hi" in user_input:
-        return "Hey there! 😊 How’s everything going? How can I assist you today?"
-    
-    elif "how are you" in user_input:
-        return "I’m doing great, thanks for asking! How about you? 😊"
-
-    elif "proposal" in user_input:
-        return ("I’d be happy to help with your proposal! 🎯 Would you like help with the structure, "
-                "content, or brainstorming ideas for your proposal? Let me know what you need, and I’ll guide you through it!")
-
-    elif "letter" in user_input:
-        return ("I can help you with writing a letter! 💌 What type of letter are you writing? "
-                "Is it formal, informal, or something else? Let me know, and I'll assist you in crafting the perfect letter!")
-
-    elif "report" in user_input:
-        return ("Reports can be tricky! 📊 Let me know what type of report you're working on (business, research, etc.), "
-                "and I can help with structure, content, or formatting.")
-
-    elif "help" in user_input:
-        return ("Of course! 😊 I’m here for you. What do you need help with today? Writing, editing, or something else?")
-
-    elif "thank you" in user_input:
-        return "You're very welcome! 😊 Always happy to help!"
-
-    elif "goodbye" in user_input or "bye" in user_input:
-        return "Goodbye for now! 👋 Don’t hesitate to reach out again whenever you need assistance!"
-
-    elif "how do i" in user_input or "what is" in user_input:
-        return "Great question! Let me guide you through it. What do you need help with? I'll make it easy for you to understand!"
-
-    else:
-        return ("Hmm, I didn’t quite catch that, but I’m all ears! 😊 Tell me a bit more, "
-                "and I’ll do my best to help. Whether it’s a proposal, letter, report, or something else, "
-                "I’m here to assist with anything you need!")
-
-
-# Function to generate and return the PDF of the document
+# Function to create a PDF document
 def generate_pdf(document_text):
     pdf = FPDF()
     pdf.add_page()
-
-    # Check if DejaVu font exists, if not fallback to default
-    font_path = 'DejaVuSans.ttf'
-    if os.path.exists(font_path):
-        pdf.add_font('DejaVu', '', font_path, uni=True)
-        pdf.set_font('DejaVu', '', 12)
-    else:
-        pdf.set_font('Arial', '', 12)  # Fallback font
-
+    pdf.set_font("Arial", size=12)
+    
     # Title
-    pdf.cell(200, 10, txt="Generated Document", ln=True, align="C")
-
+    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
@@ -93,43 +18,34 @@ def generate_pdf(document_text):
 
 # Streamlit app
 def main():
-    st.title("GM: Document Generator with Chatbot")
-
-    # Chatbot interaction
-    st.subheader("Chat with me! I'm your friendly assistant. 😊")
-    user_input = st.text_input("Ask me anything:", "")
-    
-    if user_input:
-        response = chatbot_response(user_input)
-        st.write(f"**Bot**: {response}")
-
-    # Document input and generation
-    st.subheader("Generate Your Document")
-
-    doc_type = st.selectbox("What kind of document do you want to create?", ["Proposal", "Letter", "Report", "Message", "Notice"])
+    st.title("Proposal Generator App")
 
-    document_text = st.text_area(f"Write your {doc_type} here...", height=150)
+    # Instructions for the user
+    st.subheader("Write your proposal below")
 
-    style = st.selectbox("Choose a writing style:", ["formal", "casual", "business"])
+    # Input text for the proposal
+    proposal_text = st.text_area("Enter the content of your proposal:", height=300)
 
-    if st.button("Generate Document"):
-        if document_text:
-            document = generate_document(doc_type, document_text, style)
-            st.subheader(f"Generated {doc_type}:")
-            st.markdown(f"```\n{document}\n```")
+    # 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```")
 
-            # PDF Download Button
-            pdf = generate_pdf(document)
-            pdf_output = pdf.output(dest='S')  # No encoding issues
+            # 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 PDF",
+                label="Download Proposal as PDF",
                 data=pdf_output,
-                file_name=f"{doc_type.lower()}.pdf",
+                file_name="proposal.pdf",
                 mime="application/pdf"
             )
         else:
-            st.warning("Please write your document text.")
+            st.warning("Please enter the content of your proposal.")
 
 if __name__ == "__main__":
     main()