update

app.py

@@ -4,6 +4,7 @@ import streamlit as st
 from transformers import pipeline
 from PIL import Image
 from dotenv import load_dotenv
+import time
 
 # Load environment variables from .env file
 load_dotenv()
@@ -25,17 +26,24 @@ def query_emotion(filename):
         st.error("Error detecting facial expression: " + response.text)
         return None
 
-# Function to generate a joke or uplifting text based on the detected emotion
+# Function to generate a joke or uplifting text based on the mood
 def generate_text_based_on_mood(emotion):
     try:
         prompt = f"Generate a light-hearted joke or uplifting message for someone who is feeling {emotion}."
-        response = requests.post(text_model_url, headers=headers, json={"inputs": prompt})
         
-        if response.status_code == 200:
-            return response.json()[0]['generated_text']
-        else:
-            st.error("Error generating text: " + response.text)
-            return "Sorry, I couldn't come up with a joke at this moment."
+        for attempt in range(5):  # Retry up to 5 times
+            response = requests.post(text_model_url, headers=headers, json={"inputs": prompt})
+            if response.status_code == 200:
+                return response.json()[0]['generated_text']
+            elif response.status_code == 503:  # Service Unavailable
+                st.warning("Model is loading, retrying...")
+                time.sleep(5)  # Wait before retrying
+            else:
+                st.error("Error generating text: " + response.text)
+                return "Sorry, I couldn't come up with a joke at this moment."
+        
+        st.error("Failed to generate text after multiple attempts.")
+        return "Sorry, I couldn't come up with a joke at this moment."
         
     except Exception as e:
         st.error(f"Error generating text: {e}")
@@ -44,10 +52,18 @@ def generate_text_based_on_mood(emotion):
 # Function to convert text to speech using gTTS
 def text_to_speech(text):
     from gtts import gTTS
-    tts = gTTS(text, lang='en')
-    audio_file = "output.mp3"
-    tts.save(audio_file)  # Save the audio file
-    return audio_file
+    try:
+        tts = gTTS(text, lang='en')
+        audio_file = "output.mp3"
+        tts.save(audio_file)  # Save the audio file
+        return audio_file
+    except Exception as e:
+        if "429" in str(e):
+            st.error("Too many requests to the TTS service. Please try again later.")
+            return None
+        else:
+            st.error(f"Error with TTS: {e}")
+            return None
 
 # Streamlit UI
 st.title("Facial Expression Mood Detector")
@@ -80,5 +96,6 @@ if uploaded_file is not None:
         # Convert the generated joke to audio
         audio_file = text_to_speech(joke)
 
-        # Provide an audio player in the Streamlit app
-        st.audio(audio_file)  # Streamlit will handle playback
+        # Provide an audio player in the Streamlit app if audio file exists
+        if audio_file:
+            st.audio(audio_file)  # Streamlit will handle playback