update

app.py

@@ -29,12 +29,17 @@ def query_emotion(filename):
 # 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}."
+        # Define a clear and directive prompt
+        prompt = f"Please tell a funny joke that would cheer someone up who is feeling {emotion}. Make it light-hearted and fun."
         
         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']
+                generated_text = response.json()[0]['generated_text']
+                if generated_text.strip():  # Ensure the response is not empty
+                    return generated_text
+                else:
+                    st.warning("Received an empty joke, retrying...")
             elif response.status_code == 503:  # Service Unavailable
                 st.warning("Model is loading, retrying...")
                 time.sleep(5)  # Wait before retrying