Update app.py

app.py

@@ -2,7 +2,6 @@
 import streamlit as st
 import time
 import random
-import json
 import textwrap
 import os
 import base64
@@ -10,10 +9,9 @@ from io import BytesIO
 from PIL import Image, ImageDraw, ImageFont
 import numpy as np
 from gtts import gTTS
-from pygame import mixer
 import tempfile
-import requests
-from transformers import pipeline, set_seed
+import subprocess
+from transformers import pipeline
 
 # Set up the page
 st.set_page_config(
@@ -23,9 +21,6 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# Initialize pygame mixer
-mixer.init()
-
 # Custom CSS
 st.markdown("""
 <style>
@@ -174,6 +169,12 @@ st.markdown("""
 .stProgress > div > div > div {
     background-color: #ff5722 !important;
 }
+
+/* Audio player styling */
+.audio-player {
+    width: 100%;
+    margin-top: 10px;
+}
 </style>
 """, unsafe_allow_html=True)
 
@@ -599,12 +600,16 @@ with col2:
         
         # Play voice explanation
         if st.session_state.voice_enabled and 'speech_file' in st.session_state:
-            if st.button("🔊 Play Tavus Explanation", use_container_width=True):
-                try:
-                    mixer.music.load(st.session_state.speech_file)
-                    mixer.music.play()
-                except:
-                    st.warning("Couldn't play audio. Try enabling sound in your browser.")
+            try:
+                # Read audio file and convert to base64
+                with open(st.session_state.speech_file, "rb") as f:
+                    audio_bytes = f.read()
+                
+                # Display audio player
+                st.audio(audio_bytes, format='audio/mp3')
+                
+            except Exception as e:
+                st.warning(f"Couldn't load audio: {e}")
         
     elif st.session_state.animation_generated:
         st.warning("Couldn't generate animation. Try a different story!")
@@ -789,4 +794,4 @@ st.markdown("""
         Transforming stories into games since 2023 | No API Keys Required!
     </p>
 </center>
-""", unsafe_allow_html=True) 
+""", unsafe_allow_html=True)