import streamlit as st
#from fpdf import FPDF
import base64
import openai
from streamlit_quill import st_quill
import os
from io import BytesIO
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()


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",
   messages=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):
    global messages
    processed_text = ""
    for key, value in inputs.items():
        processed_text += f"{key}: {value}\n\n"

    info= str(processed_text)

    messages=[
      {"role": "system", "content": "You are a obituary writer given information in the from of json"},
      {"role": "user", "content": "Please write custom obituary based on this information. Make sure you write little long obituary using the information provided. Here is information: \n{}".format(info)}
  ]

    model_response = get_chatgpt_response(messages)
    return model_response


from html2docx import html2docx
from io import BytesIO



def save_as_doc(text, html_content):
    doc_title = "Custom Obituary"
    buf = html2docx(html_content, title=doc_title)
    filename = f"{doc_title}.docx"

    with open(filename, "wb") as f:
        f.write(buf.getvalue())

    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("### 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": "",
        "Age at the time of passing": "",
        "Date and place of birth": "",
        "Date and place of passing": "",
        "Family members who have preceded them in death": "",
        "Surviving family members": "",
        "Education, career, hobbies and interests": "",
        "Accomplishments or contributions they made to their community or society": "",
        "Personal traits that made them special": "",
        "Service or memorial arrangements": "",
        "Any personal message or special requests you would like to include": "",
        "Additional information, like childhood or special bond": "",
    }

    for key in inputs:
        inputs[key] = st.text_input(key)

    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:
        
        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)


        if "additional_info" not in st.session_state:
            st.session_state.additional_info = ""

        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)

        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()

        # 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

def send_email(email, subject, content):
    sender_email = "myobituary.writer@gmail.com"  # Replace with your email
    sender_password = "acolructdnephudf"  # Replace with your email password

    yag = yagmail.SMTP(sender_email, sender_password)
    yag.send(to=email, subject=subject, contents=content)




def validate_email(email):
    email_regex = r"[^@]+@[^@]+\.[^@]+"
    return re.match(email_regex, email) is not None


def editor_page():
    #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, html=True)

    st.session_state.edited_text = edited_text
    #st.session_state.edited_html = edited_text['html']

    if st.button("Save as DOCX"):
        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.")

    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 {
                background-color: #f3f3f3;
                border-radius: 5px;
                padding: 5px;
            }
            .ql-container {
                border-radius: 5px;
                border: 1px solid #ccc;
                height: 700px;
            }
            .ql-editor {
                height: 100%;
            }
            {edited_text.get('css')}
        </style>
    """, unsafe_allow_html=True)






PAGES = {
    "Form Page": form_page,
    "Editor Page": editor_page,
}

def app():
    
    #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:

        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()