update

app.py

@@ -27,11 +27,39 @@ def query(filename):
 # Function to generate a joke or uplifting text based on the mood
 def generate_text_based_on_mood(emotion):
     try:
-        # Use GPT-Neo model for text generation
+        # Predefined jokes for different emotions
+        jokes = {
+            "happy": [
+                "Why did the scarecrow win an award? Because he was outstanding in his field!",
+                "What do you call fake spaghetti? An impasta!"
+            ],
+            "sad": [
+                "Why don’t scientists trust atoms? Because they make up everything!",
+                "Why did the bicycle fall over? Because it was two-tired."
+            ],
+            "angry": [
+                "Why did the tomato turn red? Because it saw the salad dressing!",
+                "What did one angry man say to the other? Nothing, he just threw a rock!"
+            ],
+            "surprised": [
+                "Did you hear about the claustrophobic astronaut? He just needed a little space!",
+                "Why don’t skeletons fight each other? They don’t have the guts."
+            ],
+            "neutral": [
+                "What do you call cheese that isn't yours? Nacho cheese!"
+            ]
+        }
+
+        # Use predefined jokes if available
+        if emotion in jokes:
+            return jokes[emotion][0]  # Return the first joke for simplicity
+
+        # If no predefined joke, use GPT-Neo
         generator = pipeline('text-generation', model='EleutherAI/gpt-neo-125M')
-        prompt = f"Tell a joke that would cheer someone who is feeling {emotion}."
+        prompt = f"Generate a light-hearted joke for someone who is feeling {emotion}."
         response = generator(prompt, max_length=50, num_return_sequences=1)
         return response[0]['generated_text']
+        
     except Exception as e:
         st.error(f"Error generating text: {e}")
         return "Sorry, I couldn't come up with a joke at this moment."