update chat.py tts berulang dan teks icon

app/chat.py

@@ -5,6 +5,7 @@ import base64
 import gtts
 from io import BytesIO
 
+# Inisialisasi state
 def initialize_session_state():
     if 'history' not in st.session_state:
         st.session_state['history'] = []
@@ -23,6 +24,7 @@ def initialize_session_state():
     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()
@@ -36,14 +38,17 @@ def text_to_speech(text):
     """
     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 atau Tekan tombol mic untuk berbicara!",
         key="chat_input_field"
@@ -51,15 +56,17 @@ def display_chat_history(chain):
 
     col2, col3 = st.columns([1, 1])
 
+    # Tombol toggle suara
     with col2:
         should_speak = st.session_state.get('should_speak', True)
-        icon_label = "🔊" if should_speak else "🔇"
+        icon_label = "🔊 Text-to-Speech Aktif" if should_speak else "🔇 Text-to-Speech Nonaktif"
         if st.button(icon_label, key="toggle_tts", help="Aktifkan/Nonaktifkan Text-to-Speech", use_container_width=True):
             st.session_state['should_speak'] = not should_speak
 
+    # Rekaman suara
     with col3:
         stt_text = speech_to_text(
-            start_prompt="🎤",
+            start_prompt="🎤 Input Suara",
             stop_prompt="🛑 Stop",
             language='id',
             just_once=True,
@@ -67,12 +74,12 @@ def display_chat_history(chain):
             use_container_width=True,
         )
 
-    # Jika STT aktif, simpan ke input dan rerun
+    # Jika hasil STT ada, pakai sebagai input
     if stt_text:
         st.session_state.input_text = stt_text
         st.experimental_rerun()
 
-    # Ambil input dari teks biasa atau hasil STT
+    # Ambil input dari teks atau STT
     user_input = user_input_obj or st.session_state.get("input_text", "")
 
     if user_input:
@@ -83,18 +90,25 @@ def display_chat_history(chain):
         st.session_state['generated'].append(output)
         st.session_state.input_text = ""
 
-        # Simpan output untuk TTS (dieksekusi setelah UI selesai)
+        # Simpan output untuk TTS
         if st.session_state['should_speak'] and output:
             st.session_state['tts_output'] = output
 
-    # Tampilkan riwayat obrolan
+    # 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")
 
-    # Mainkan TTS jika ada
+    # Tampilkan dan mainkan audio TTS jika ada
     if st.session_state.get('tts_output'):
         st.markdown(text_to_speech(st.session_state['tts_output']), unsafe_allow_html=True)
-        st.session_state['tts_output'] = ""
+
+        # Beri waktu render audio sebelum menghapus tts_output
+        if 'tts_played' not in st.session_state:
+            st.session_state['tts_played'] = True
+            st.experimental_rerun()
+        else:
+            st.session_state['tts_output'] = ""
+            del st.session_state['tts_played']