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 weasyprint import HTML
import tempfile
import weasyprint


os.environ["OPENAI_API_KEY"] = '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'
openai.api_key = os.getenv("OPENAI_API_KEY")

def get_chatgpt_response(messages):
   response = openai.ChatCompletion.create(
   model="gpt-3.5-turbo",
   messages=messages
 )
   return  response['choices'][0]['message']['content']

def process_text(inputs):
    # Perform some processing on the input dictionary here
    # For example, you could concatenate all the inputs into a single string
    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

# 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

def save_as_pdf(text):
    # create a PDF from the HTML content
    pdf_file = BytesIO()
    options = {
        'quiet': '',
        'page-size': 'Letter',
        'margin-top': '0.75in',
        'margin-right': '0.75in',
        'margin-bottom': '0.75in',
        'margin-left': '0.75in',
    }
    pdfkit.from_string(text, pdf_file, options=options)

    # encode the PDF to base64
    b64 = base64.b64encode(pdf_file.getvalue()).decode('utf-8')

    # generate the filename
    filename = 'output.pdf'

    return b64, filename



def form_page():

    st.markdown("### Your custom obituary writer :pencil:")
    st.markdown("### Please fill the details to get custom obituary")

    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("Process"):
        output_text = process_text(inputs)
        st.session_state.output_text = output_text
        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()



# def editor_page():
#     #st.markdown("### Editor :smile:")
#     st.markdown("<h1 style='text-align: center; color: Red;'>Editor</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)

#     st.session_state.edited_text= edited_text

#     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.")

#     # 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 editor_page():
    st.markdown("<h1 style='text-align: center; color: Red;'>Editor</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.session_state.edited_text = edited_text

    if st.button("Save as PDF"):
        # parse the HTML content and extract the CSS styles
        soup = BeautifulSoup(edited_text, 'html.parser')
        css = ""
        for style in soup.find_all('style'):
            css += style.string
            style.decompose()

        # create a new HTML document with the CSS styles inline
        new_html = f"""<html><head><style>{css}</style></head><body>{soup}</body></html>"""

        # Save the output text as a PDF
        pdf_data, filename = save_as_pdf(new_html)

        # display a download button for the PDF file
        st.download_button(
            label='Download PDF',
            data=pdf_data,
            file_name=filename,
            mime='application/pdf'
        )

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


    


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

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()
    else:
        st.sidebar.radio("",["Editor"])
        editor_page()

if __name__ == "__main__":
    app()