Update app.py

app.py

@@ -1,19 +1,193 @@
-
 import streamlit as st
-from fpdf import FPDF
+#from fpdf import FPDF
 import base64
 import openai
 from streamlit_quill import st_quill
 import os
 from io import BytesIO
-#from weasyprint import HTML
-import tempfile
-import weasyprint
+from docx import Document
+import yagmail
+import re
+#import re
+import html2text
+from docx import Document
+from docx.shared import Pt
+from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
+from docx.oxml.ns import qn
+from docx.oxml import OxmlElement
+from bs4 import BeautifulSoup
+
+
+from html2docx import HTML2Docx
+
+import streamlit as st
+import streamlit.components.v1 as components
 
+import streamlit as st
+from captcha.image import ImageCaptcha
+import random, string
+
+# define the constant
+length_captcha = 4
+width = 380
+height = 200
+
+
+def read_index_html():
+    with open("index.html") as f:
+        return f.read()
 
-os.environ["OPENAI_API_KEY"] = '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'
+
+custom_css = """
+<style>
+    body {
+        background-color: #F8F9F9;
+    }
+    h1 {
+        color: #2980B9;
+    }
+    h2 {
+        color: #21618C;
+    }
+    h3 {
+        color: #1C2833;
+    }
+    input, textarea {
+        background-color: #F2F3F4;
+        border-radius: 5px;
+    }
+    button {
+        background-color: #2980B9;
+        color: white;
+        font-weight: bold;
+        border-radius: 5px;
+    }
+    a {
+        color: #2980B9;
+    }
+    .streamlit-button {
+        display: flex;
+        justify-content: center;
+        align-items: center;
+    }
+    .streamlit-button:hover {
+        background-color: #21618C;
+    }
+</style>
+"""
+
+# custom_css = """
+# <style>
+#     body {
+#         background-color: #F8F9F9;
+#     }
+#     h1 {
+#         color: #2980B9;
+#     }
+#     h2 {
+#         color: #21618C;
+#     }
+#     h3 {
+#         color: #1C2833;
+#     }
+#     input, textarea {
+#         background-color: #F2F3F4;
+#         border-radius: 5px;
+#     }
+#     button {
+#         background-color: #2980B9;
+#         color: white;
+#         font-weight: bold;
+#         border-radius: 5px;
+#     }
+#     a {
+#         color: #2980B9;
+#     }
+#     .streamlit-button {
+#         display: flex;
+#         justify-content: center;
+#         align-items: center;
+#         color: white;
+#     }
+#     .streamlit-button:hover {
+#         background-color: #21618C;
+#     }
+#     .stTextInput > div > div > input {
+#         border: 1px solid #C0C0C0;
+#         border-radius: 5px;
+#     }
+# </style>
+# """
+
+
+os.environ["OPENAI_API_KEY"] = 'sk-E'
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
+# define the function for the captcha control
+def captcha_control():
+    # control if the captcha is correct
+    if 'controllo' not in st.session_state or st.session_state['controllo'] == False:
+        #st.title("Please verify the Captcha to proceed")
+        st.markdown("<h1 style='text-align: center;'>Please verify the Captcha to proceed</h1>", unsafe_allow_html=True)
+
+        # define the session state for control if the captcha is correct
+        st.session_state['controllo'] = False
+
+        # define the session state for the captcha text because it doesn't change during refreshes
+        if 'Captcha' not in st.session_state:
+            st.session_state['Captcha'] = ''.join(random.choices(string.ascii_uppercase + string.digits, k=length_captcha))
+        print("the captcha is: ", st.session_state['Captcha'])
+
+        # # setup the captcha widget
+        # image = ImageCaptcha(width=width, height=height)
+        # data = image.generate(st.session_state['Captcha'])
+        # st.image(data)
+        # capta2_text = st.text_input('Enter captcha text',  max_chars=4)
+
+        # setup the captcha widget
+        # image = ImageCaptcha(width=width, height=height)
+        # data = image.generate(st.session_state['Captcha'])
+        # st.image(data)
+
+        # # Create a centered column with the same width as the captcha image
+        # left_col, center_col, right_col = st.columns([150, width, 150])
+
+        # capta2_text = center_col.text_input('Enter captcha text', max_chars=4)
+        # Create a centered column with the same width as the captcha image for the captcha image
+        image = ImageCaptcha(width=width, height=height)
+        data = image.generate(st.session_state['Captcha'])
+        left_col_img, center_col_img, right_col_img = st.columns([150, width, 150])
+        center_col_img.image(data)
+
+        # Create a centered column with the same width as the captcha image for the text input
+        left_col_input, center_col_input, right_col_input = st.columns([150, width, 150])
+        capta2_text = center_col_input.text_input('Enter captcha text', max_chars=4)
+
+        # Create a centered column with the same width as the captcha image for the button
+        left_col_btn, center_col_btn, right_col_btn = st.columns([150, width, 150])
+
+        if center_col_btn.button("Verify the code"):
+
+
+        #if st.button("Verify the code"):
+            print(capta2_text, st.session_state['Captcha'])
+            capta2_text = capta2_text.replace(" ", "")
+            # if the captcha is correct, the controllo session state is set to True
+            if st.session_state['Captcha'].lower() == capta2_text.lower().strip():
+                del st.session_state['Captcha']
+                st.session_state['controllo'] = True
+                st.experimental_rerun()
+            else:
+                # if the captcha is wrong, the controllo session state is set to False and the captcha is regenerated
+                st.error("🚨 The captcha code is incorrect, please try again.")
+                #st.write("🚨 The captcha code is incorrect, please try again.")
+                del st.session_state['Captcha']
+                del st.session_state['controllo']
+                st.experimental_rerun()
+        else:
+            # wait for the button click
+            st.stop()
+
 def get_chatgpt_response(messages):
    response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
@@ -21,9 +195,12 @@ def get_chatgpt_response(messages):
  )
    return  response['choices'][0]['message']['content']
 
+def update_chat(messages, role, content):
+  messages.append({"role": role, "content": content})
+  return messages
+
 def process_text(inputs):
-    # Perform some processing on the input dictionary here
-    # For example, you could concatenate all the inputs into a single string
+    global messages
     processed_text = ""
     for key, value in inputs.items():
         processed_text += f"{key}: {value}\n\n"
@@ -36,84 +213,34 @@ def process_text(inputs):
   ]
 
     model_response = get_chatgpt_response(messages)
-
     return model_response
 
-# def save_as_pdf(text):
-#     pdf = FPDF()
-#     pdf.add_page()
-#     pdf.set_font("Arial", size=12)
-#     encoded_text = text.encode('latin-1', 'replace').decode('latin-1')
-#     pdf.multi_cell(0, 10, txt=encoded_text)
-#     file_name = "output.pdf"
-#     pdf.output(file_name)
-#     with open(file_name, "rb") as pdf_file:
-#         b64 = base64.b64encode(pdf_file.read()).decode('utf-8')
-#     href = f'<a href="data:application/octet-stream;base64,{b64}" download="{file_name}">Download PDF</a>'
-#     st.markdown(href, unsafe_allow_html=True)
-
-
-
-# import base64
-# from io import BytesIO
-# from weasyprint import HTML
-# #from streamlit_quill import Delta, Renderer
-# from quill.delta import Delta
-
-# def save_as_pdf(text):
-#     # convert quill editor HTML to PDF
-#     pdf_file = BytesIO()
-#     HTML(string=text).write_pdf(pdf_file, stylesheets=[CSS(string="""
-#         .ql-editor {
-#             font-size: 16px;
-#             line-height: 1.6;
-#         }
-#         """)], style_tags=True)
-
-#     # encode the PDF to base64
-#     b64 = base64.b64encode(pdf_file.getvalue()).decode('utf-8')
-
-#     # generate a download link for the PDF
-#     href = f'<a href="data:application/octet-stream;base64,{b64}" download="output.pdf">Download PDF</a>'
-#     st.markdown(href, unsafe_allow_html=True)
-
-
-import tempfile
-import weasyprint
-from bs4 import BeautifulSoup
-
-import pdfkit
 
-from docx import Document
+from html2docx import html2docx
 from io import BytesIO
-import base64
 
-def save_as_doc(text):
-    # create a Word document
-    doc = Document()
 
-    # add the edited text to the document
-    doc.add_paragraph(text)
 
-    # save the document to a BytesIO object
-    doc_file = BytesIO()
-    doc.save(doc_file)
-    doc_file.seek(0)
+def save_as_doc(text, html_content):
+    doc_title = "Custom Obituary"
+    buf = html2docx(html_content, title=doc_title)
+    filename = f"{doc_title}.docx"
 
-    # encode the document to base64
-    b64 = base64.b64encode(doc_file.getvalue()).decode('utf-8')
+    with open(filename, "wb") as f:
+        f.write(buf.getvalue())
 
-    # generate a download link for the document
-    href = f'<a href="data:application/octet-stream;base64,{b64}" download="output.docx">Download DOCX</a>'
-    st.markdown(href, unsafe_allow_html=True)
+    with open(filename, "rb") as f:
+        base64_content = base64.b64encode(f.read()).decode("utf-8")
+        href = f'<a href="data:application/octet-stream;base64,{base64_content}" download="{filename}" id="download-docx-link">Download</a>'
+        st.markdown(href, unsafe_allow_html=True)
 
+    os.remove(filename)
 
 
 
 def form_page():
-
-    st.markdown("### Your custom obituary writer :pencil:")
-    st.markdown("### Please fill the details to get custom obituary")
+    st.markdown("### Free Obituary Writing Assistant :pencil:")
+    st.write("Complete any applicable fields below. If you don't know the answers, that's OK you can add them later")
 
     inputs = {
         "Full name, including any nicknames": "",
@@ -133,93 +260,118 @@ def form_page():
     for key in inputs:
         inputs[key] = st.text_input(key)
 
-    if st.button("Process"):
-        output_text = process_text(inputs)
-        st.session_state.output_text = output_text
-        st.write(output_text)
-        st.session_state.form = True
+    if st.button("Create Obituary"):
+        
+        global messages
+        with st.spinner('Generating obituary...'):
+            output_text = process_text(inputs)
+            messages = update_chat(messages, "assistant", output_text)
+            st.session_state.output_text = output_text
+            st.session_state.messages = messages
+            #st.write(output_text)
+            st.session_state.form = True
 
-    if st.session_state.form:
-        if st.button("I want to add more information"):
-            additional_info = st.text_input("What else would you like to add?")
-            if additional_info:
-                message = f"{additional_info}"
-                output_text = get_chatgpt_response(message)
-                st.session_state.output_text += "\n" + output_text
-                st.write(st.session_state.output_text)
 
-        if st.button("I want to export and edit manually"):
-            st.session_state.export_manually = True
-            st.session_state.form = False
-            st.experimental_rerun()
 
+    if st.session_state.form:
+        
+        output_placeholder = st.empty()
+        #output_placeholder.write(st.session_state.output_text)
+        #output_placeholder.markdown(f'<div style="background-color: #6495ED; color: white; padding: 10px; border-radius: 5px;">{st.session_state.output_text}</div>', unsafe_allow_html=True)
+        output_placeholder.markdown(f'<div style="background-color: #AED6F1; color: black; padding: 10px; border-radius: 5px;">{st.session_state.output_text}</div>', unsafe_allow_html=True)
 
 
-# def editor_page():
-#     #st.markdown("### Editor :smile:")
-#     st.markdown("<h1 style='text-align: center; color: Red;'>Editor</h1>", unsafe_allow_html=True)
+        if "additional_info" not in st.session_state:
+            st.session_state.additional_info = ""
 
-#     st.write("Use the editor below to edit the obituary:")
+        if "additional_info_key" not in st.session_state:
+            st.session_state.additional_info_key = "additional_info_input_1"
 
+        additional_info_placeholder = st.empty()
+        additional_info = additional_info_placeholder.text_input("What else would you like to add?", value=st.session_state.additional_info, key=st.session_state.additional_info_key)
 
-#     quill_text = st.session_state.output_text
-#     edited_text = st_quill(quill_text)
-    
+        if st.button("I want to add more information"):
+            
+            messages = st.session_state.messages
+            with st.spinner('Generating obituary...'):
+                if additional_info:
+                    message = "can you please re-write this obituary by using earlier information and new information. Please make sure you write obituary little long. new information is here: " + additional_info
+                    messages = update_chat(messages, "user", message)
+                    output_text = get_chatgpt_response(messages)
+                    messages = update_chat(messages, "assistant", output_text)
+                    st.session_state.output_text = output_text
+                    #output_placeholder.write(output_text)
+                    #output_placeholder.markdown(f'<div style="background-color: #6495ED; color: white; padding: 10px; border-radius: 5px;">{output_text}</div>', unsafe_allow_html=True)
+                    output_placeholder.markdown(f'<div style="background-color: #AED6F1; color: black; padding: 10px; border-radius: 5px;">{output_text}</div>', unsafe_allow_html=True)
+
+
+                    st.session_state.additional_info = ""
+                    st.session_state.additional_info_key = f"additional_info_input_{int(st.session_state.additional_info_key.split('_')[-1]) + 1}"
+                    additional_info_placeholder.text_input("What else would you like to add?", value=st.session_state.additional_info, key=st.session_state.additional_info_key)  
+      
+
+        if st.button("I want to export and edit manually"):
+            
+            st.session_state.export_manually = True
+            st.session_state.form = False
+            st.experimental_rerun()
 
-#     st.write("Here is the edited obituary:")
-#     #st.write(edited_text)
+        # if not st.session_state.form:
+        #     if st.button("I want to export and edit manually"):
+        #         st.session_state.form = True
+        # else:
+        #     if st.button("Go Back to Form"):
+        #         st.session_state.form = False
 
-#     st.session_state.edited_text= edited_text
+def send_email(email, subject, content):
+    sender_email = "[email protected]"  # Replace with your email
+    sender_password = "acolructdnephudf"  # Replace with your email password
 
-#     if st.button("Save as PDF"):
-#             # Save the output text as a PDF
-#         #save_as_pdf(st.session_state.output_text)
-#         save_as_pdf(st.session_state.edited_text)
-#         st.write("The custom obituary has been saved as a PDF.")
+    yag = yagmail.SMTP(sender_email, sender_password)
+    yag.send(to=email, subject=subject, contents=content)
 
-#     # Add some custom CSS to style the editor
-#     st.markdown("""
-#         <style>
-#             #toolbar {
-#                 background-color: #f3f3f3;
-#                 border-radius: 5px;
-#                 padding: 5px;
-#             }
-#             .ql-container {
-#                 border-radius: 5px;
-#                 border: 1px solid #ccc;
-#                 height: 400px;
-#             }
-#             .ql-editor {
-#                 height: 100%;
-#             }
-#         </style>
-#     """, unsafe_allow_html=True)
 
 
-#from quill.delta import Delta, Renderer
 
+def validate_email(email):
+    email_regex = r"[^@]+@[^@]+\.[^@]+"
+    return re.match(email_regex, email) is not None
 
 
 def editor_page():
-    st.markdown("<h1 style='text-align: center; color: Red;'>Editor</h1>", unsafe_allow_html=True)
-
+    #form_page()
+    
+    st.markdown("<h1 style='text-align: center; color: Red;'>Obituary Draft</h1>", unsafe_allow_html=True)
     st.write("Use the editor below to edit the obituary:")
 
     quill_text = st.session_state.output_text
-    edited_text = st_quill(quill_text)
-
-    st.write("Here is the edited obituary:")
-    st.write(edited_text)
+    edited_text = st_quill(quill_text, html=True)
 
     st.session_state.edited_text = edited_text
+    #st.session_state.edited_html = edited_text['html']
 
     if st.button("Save as DOCX"):
-        # Save the output text as a Word document
-        save_as_doc(st.session_state.edited_text)
+        html_content = st.session_state.edited_text
+        save_as_doc(st.session_state.edited_text, html_content)
         st.write("The custom obituary has been saved as a Word document.")
 
-    # Add some custom CSS to style the editor
+    if "email_obituary" not in st.session_state:
+        st.session_state.email_obituary = False
+
+    if st.button("Email Obituary"):
+        st.session_state.email_obituary = not st.session_state.email_obituary
+
+    if st.session_state.email_obituary:
+        recipient_email = st.text_input("Enter your email address:", key="recipient_email")
+
+        if st.button("Send", key="send_email_button"):
+            if recipient_email and validate_email(recipient_email):
+                with st.spinner('Sending...'):
+                    send_email(recipient_email, "Your Custom Obituary", st.session_state.edited_text)
+                    st.success("Obituary sent to your email!")
+            else:
+                st.error("Please enter a valid email address.")
+
     st.markdown("""
         <style>
             #toolbar {
@@ -240,7 +392,8 @@ def editor_page():
     """, unsafe_allow_html=True)
 
 
-    
+
+
 
 
 PAGES = {
@@ -249,17 +402,115 @@ PAGES = {
 }
 
 def app():
-    if "form" not in st.session_state:
-        st.session_state.form = False
-    if "export_manually" not in st.session_state:
-        st.session_state.export_manually = False
-
-    if not st.session_state.export_manually:
-        st.sidebar.radio("",["form"])
-        form_page()
+    
+    #st.markdown(custom_css, unsafe_allow_html=True)
+
+#      custom_css = """
+#      <style>
+
+#          /* Input box boundary */
+#          .stTextInput > div > div > input {
+#              border: 1px solid #C0C0C0;
+#              border-radius: 5px;
+#          }
+#      </style>
+#  """
+
+    st.markdown(custom_css, unsafe_allow_html=True)
+
+#     button_text_style = """
+#     <style>
+#   .stButton > button {
+#     color: white !important;
+#   }
+#   /* Button hover color */
+#   .stButton > button:hover {
+#             background-color: #45a049;
+#   }
+
+#   .stTextInput > div > div > input {
+#         border: 1px solid #C0C0C0;
+#         border-radius: 5px;
+#     }
+# </style>
+# """
+#     st.markdown(button_text_style, unsafe_allow_html=True)
+
+    button_text_style = """
+<style>
+    .stButton > button {
+        color: white !important;
+    }
+    
+    .stButton > button:hover {
+        background-color: #45a049;
+        color: red !important;
+    }
+
+    .stTextInput > div > div > input {
+        border: 1px solid #C0C0C0;
+        border-radius: 5px;
+    }
+    /* Background color for buttons */
+    .stButton > button {
+        background-color: #ff0290;
+        color: white;
+        }
+
+</style>
+"""
+
+    st.markdown(button_text_style, unsafe_allow_html=True)
+
+
+    hide_streamlit_style = """
+            <style>
+            #MainMenu {visibility: hidden;}
+            footer {visibility: hidden;}
+            </style>
+            """
+    st.markdown(hide_streamlit_style, unsafe_allow_html=True)
+
+    hide_decoration_bar_style = '''
+    <style>
+        header {visibility: hidden;}
+    </style>
+'''
+    st.markdown(hide_decoration_bar_style, unsafe_allow_html=True) 
+
+    #reduce top margin
+    reduce_header_height_style = """      
+    <style>
+        div.block-container {padding-top:0.01rem;}
+    </style>
+"""
+    st.markdown(reduce_header_height_style, unsafe_allow_html=True)
+
+    if 'controllo' not in st.session_state or st.session_state['controllo'] == False:
+        captcha_control()
     else:
-        st.sidebar.radio("",["Editor"])
-        editor_page()
+
+        if "form" not in st.session_state:
+            st.session_state.form = False
+        if "export_manually" not in st.session_state:
+            st.session_state.export_manually = False
+
+        if not st.session_state.export_manually:
+            form_page()
+            components.html(
+        read_index_html(),
+        height=0,
+        width=0,
+    )
+        else:
+            editor_page()
+
+            components.html(
+        read_index_html(),
+        height=0,
+        width=0,
+    )
 
 if __name__ == "__main__":
-    app()
+    
+    app()