tambahkan putar ulang

app/chat.py

@@ -21,18 +21,34 @@ def initialize_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'] = ""
+    if 'audio_cache' not in st.session_state:
+        st.session_state['audio_cache'] = {}
+    if 'play_audio' not in st.session_state:
+        st.session_state['play_audio'] = None
 
-# Fungsi TTS
-def text_to_speech(text):
+# Fungsi TTS dengan cache
+def text_to_speech(text, message_id=None):
+    # Cek cache terlebih dahulu
+    if message_id and message_id in st.session_state['audio_cache']:
+        return st.session_state['audio_cache'][message_id]
+    
     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()
+    
+    # Simpan ke cache jika ada message_id
+    if message_id:
+        st.session_state['audio_cache'][message_id] = audio_base64
+    
+    return audio_base64
+
+# Fungsi untuk membuat audio player
+def create_audio_player(audio_base64, autoplay=False):
+    autoplay_attr = "autoplay" if autoplay else ""
     audio_player = f"""
-    <audio autoplay>
+    <audio {autoplay_attr} controls style="width: 100%; margin-top: 5px;">
         <source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3">
     </audio>
     """
@@ -92,23 +108,42 @@ def display_chat_history(chain):
         st.session_state['generated'].append(output)
         st.session_state.input_text = ""
 
-        # Simpan output untuk TTS
+        # Auto-play TTS untuk pesan baru jika diaktifkan
         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']
+            message_id = len(st.session_state['generated']) - 1
+            audio_base64 = text_to_speech(output, message_id)
+            st.session_state['play_audio'] = message_id
 
-    # Tampilkan riwayat chat
+    # Tampilkan riwayat chat dengan tombol TTS
     if st.session_state['generated']:
         with reply_container:
             for i in range(len(st.session_state['generated'])):
+                # Tampilkan pesan user
                 message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="no-avatar")
+                
+                # Tampilkan pesan bot
                 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']
+                
+                # Tombol dan audio player untuk pesan bot
+                col1, col2 = st.columns([1, 4])
+                with col1:
+                    if st.button(f"🔊", key=f"play_tts_{i}", help="Putar ulang audio"):
+                        st.session_state['play_audio'] = i
+                        st.experimental_rerun()
+                
+                # Tampilkan audio player jika pesan ini yang dipilih untuk diputar
+                if st.session_state.get('play_audio') == i:
+                    with col2:
+                        message_text = st.session_state["generated"][i]
+                        audio_base64 = text_to_speech(message_text, i)
+                        
+                        # Auto-play untuk pesan baru, manual untuk replay
+                        autoplay = (i == len(st.session_state['generated']) - 1 and 
+                                  st.session_state['should_speak'])
+                        
+                        audio_player = create_audio_player(audio_base64, autoplay)
+                        st.markdown(audio_player, unsafe_allow_html=True)
+                    
+                    # Reset play_audio setelah ditampilkan
+                    if st.session_state.get('play_audio') == i:
+                        st.session_state['play_audio'] = None