import datetime
import streamlit as st
from config.print_data import ConversationProcessor
from arabic_support import support_arabic_text
from pathlib import Path
from PIL import Image
from datetime import date


current_file_path = Path(__file__).resolve()
project_root = current_file_path.parents[1]
assets_dir = project_root / "asset"
database_dir = project_root / "data"
styles_dir = project_root / "styles"

st.set_page_config(
    page_title="Data Handling",
    page_icon="📈",
    layout="wide",
    initial_sidebar_state="expanded",
)
support_arabic_text(all=True)
with open(styles_dir / "main.css", "r") as file:
    css_code = file.read()

st.markdown(f"<style>{css_code}</style>", unsafe_allow_html=True)
side = st.sidebar

state = st.session_state
if "messages" not in state:
    state["messages"] = []

support_arabic_text(all=True)

# Sidebar menu
with side:
    # Display image
    img = Image.open(assets_dir / "us.png")
    st.image(img, width=250, caption="بررسی های صورت گرفته را میتوانید در این بخش دریافت نمایید")


def main():

    st.title('بررسی نتایج و گزارش وضعیت')

    processor = ConversationProcessor(state)

    if st.button('ذخیره گزارش'):
        output_file = f"{date.today()}_report.txt"
        processor.save_conversation_to_text(output_file)

        with open(output_file, 'r', encoding='utf-8') as file:
            report_text = file.read()

        st.text_area("گزارش", report_text, height=300)

        st.download_button(
            label="دانلود فایل متنی",
            data=report_text,
            file_name=output_file,
            mime="text/plain"
        )


if __name__ == "__main__":
    main()