Update app.py

app.py

@@ -3,9 +3,19 @@ import streamlit as st
 from transformers import pipeline
 from PIL import Image
 import torch
-from gtts import gTTS
 import os
 import tempfile
+import sys
+import subprocess
+
+# Try to import gTTS, install if missing
+try:
+    from gtts import gTTS
+except ImportError:
+    st.warning("Installing required package: gTTS...")
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "gTTS"])
+    from gtts import gTTS
+    st.success("gTTS installed successfully!")
 
 # Simple image-to-text function
 def img2text(image):
@@ -56,7 +66,7 @@ def text2story(text):
     # If no good ending is found, return as is
     return story_text
 
-# Updated text-to-audio function using gTTS instead of transformers
+# Updated text-to-audio function using gTTS
 def text2audio(story_text):
     # Create a temporary file
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp3')
@@ -106,4 +116,30 @@ if uploaded_file is not None:
             st.audio(audio_bytes, format='audio/mp3')
         except Exception as e:
             st.error(f"Error generating or playing audio: {e}")
-            st.write("Make sure gTTS is installed with: pip install gTTS")
+            st.info("If you're having issues with gTTS, you might need to manually install it with: pip install gTTS")
+            
+            # Fallback to a simple TTS if gTTS fails
+            try:
+                st.write("Attempting fallback to pyttsx3...")
+                import pyttsx3
+                engine = pyttsx3.init()
+                
+                # Create a temporary file for the fallback audio
+                temp_wav = tempfile.NamedTemporaryFile(delete=False, suffix='.wav')
+                temp_wav_filename = temp_wav.name
+                temp_wav.close()
+                
+                # Generate and save speech
+                engine.save_to_file(story, temp_wav_filename)
+                engine.runAndWait()
+                
+                # Read the audio file
+                with open(temp_wav_filename, 'rb') as audio_file:
+                    fallback_audio = audio_file.read()
+                
+                # Clean up
+                os.unlink(temp_wav_filename)
+                
+                st.audio(fallback_audio, format='audio/wav')
+            except:
+                st.error("Both TTS methods failed. Please install gTTS manually.")