app/chat.py

import streamlit as st
from streamlit_chat import message
from streamlit_mic_recorder import speech_to_text
import base64
import gtts
import os
from dotenv import load_dotenv
from io import BytesIO
import smtplib
from email.message import EmailMessage

load_dotenv()
def send_feedback_email(feedback):
    email_sender = os.getenv("EMAIL_SENDER")
    email_password = os.getenv("EMAIL_PASSWORD")
    email_receiver = os.getenv("EMAIL_RECEIVER")

    msg = EmailMessage()
    msg.set_content(f"Feedback Chatbot:\n\n{feedback}")
    msg["Subject"] = "Feedback Chatbot PNP"
    msg["From"] = email_sender
    msg["To"] = email_receiver

    try:
        with smtplib.SMTP_SSL("smtp.gmail.com", 465) as smtp:
            smtp.login(email_sender, email_password)
            smtp.send_message(msg)
        return True
    except Exception as e:
        st.error(f"Gagal mengirim feedback: {e}")
        return False

# Inisialisasi state
def initialize_session_state():
    if 'history' not in st.session_state:
        st.session_state['history'] = []
    if 'generated' not in st.session_state:
        st.session_state['generated'] = ["Halo! Saya bisa membantu anda menjawab pertanyaan seputar Politeknik Negeri Padang!"]
    if 'past' not in st.session_state:
        st.session_state['past'] = ["Hai! 👋"]
    if 'data_len' not in st.session_state:
        st.session_state['data_len'] = 0
    if 'vector_store' not in st.session_state:
        st.session_state['vector_store'] = None
    if 'should_speak' not in st.session_state:
        st.session_state['should_speak'] = True
    if 'input_text' not in st.session_state:
        st.session_state['input_text'] = ""
    if 'tts_output' not in st.session_state:
        st.session_state['tts_output'] = ""

# Fungsi TTS
def text_to_speech(text):
    tts = gtts.gTTS(text, lang="id")
    audio_bytes = BytesIO()
    tts.write_to_fp(audio_bytes)
    audio_bytes.seek(0)
    audio_base64 = base64.b64encode(audio_bytes.read()).decode()
    audio_player = f"""
    <audio autoplay>
        <source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3">
    </audio>
    """
    return audio_player

# Fungsi untuk memanggil chatbot
def conversation_chat(query, chain, history):
    result = chain({"question": query, "chat_history": history})
    history.append((query, result["answer"]))
    return result["answer"]

# Tampilan chat dan logika STT + TTS
def display_chat_history(chain):
    reply_container = st.container()

    # Input teks biasa
    user_input_obj = st.chat_input(
        "Masukkan pertanyaan",
        key="chat_input_field"
    )

    col2, col3 = st.columns([1, 1])

    # Tombol toggle suara
    with col2:
        if st.button("🔊 Text-to-Speech Aktif" if st.session_state['should_speak'] else "🔇 Text-to-Speech Nonaktif", 
                    key="toggle_tts", 
                    help="Aktifkan/Nonaktifkan Text-to-Speech", 
                    use_container_width=True):
            st.session_state['should_speak'] = not st.session_state['should_speak']
            st.experimental_rerun()

    # Rekaman suara
    with col3:
        stt_text = speech_to_text(
            start_prompt="🎤 Input Suara",
            stop_prompt="🛑 Stop",
            language='id',
            just_once=True,
            key='stt_input',
            use_container_width=True,
        )

    # Jika hasil STT ada, pakai sebagai input
    if stt_text:
        st.session_state.input_text = stt_text
        st.experimental_rerun()

    # Ambil input dari teks atau STT
    user_input = user_input_obj or st.session_state.get("input_text", "")

    if user_input:
        with st.spinner('Sedang membuat jawaban...'):
            output = conversation_chat(user_input, chain, st.session_state['history'])

        st.session_state['past'].append(user_input)
        st.session_state['generated'].append(output)
        st.session_state.input_text = ""

        # Simpan output untuk TTS
        if st.session_state['should_speak'] and output:
            st.session_state['tts_output'] = output
            if 'tts_played' in st.session_state:
                del st.session_state['tts_played']

    # Tampilkan riwayat chat
    if st.session_state['generated']:
        with reply_container:
            for i in range(len(st.session_state['generated'])):
                message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="no-avatar")
                message(st.session_state["generated"][i], key=str(i), avatar_style="no-avatar")

    # Tampilkan dan mainkan audio TTS jika ada
    if st.session_state.get('tts_output') and not st.session_state.get('tts_played'):
        st.markdown(text_to_speech(st.session_state['tts_output']), unsafe_allow_html=True)
        st.session_state['tts_played'] = True
    elif st.session_state.get('tts_played'):
        st.session_state['tts_output'] = ""
        del st.session_state['tts_played']


    with st.expander("💬 Kirim Feedback"):
            with st.form("feedback_form"):
                feedback_text = st.text_area("Masukkan feedback Anda untuk chatbot ini:")
                submitted = st.form_submit_button("Kirim")

                if submitted and feedback_text.strip():
                    success = send_feedback_email(feedback_text)
                    if success:
                        st.success("✅ Terima kasih! Feedback Anda telah dikirim.")