Update app.py

app.py

@@ -116,6 +116,45 @@ def generate_daily_quote():
         logger.error(f"Error generating daily quote: {str(e)}")
         return "Hare Manavi! Let your heart bloom like the lotus in the Yamuna, for love is the sweetest melody of Vrindavan!"
 
+def generate_horoscope():
+    """Generate a personalized daily horoscope for Manavi in Krishna's tone."""
+    headers = {
+        "Authorization": f"Bearer {HUGGINGFACE_API_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    prompt = (
+        "You are Little Krishna, a wise and playful cowherd from Vrindavan, speaking to Manavi on April 13, 2025. "
+        "Generate a short, personalized daily horoscope (1-2 sentences) for Manavi, reflecting her interests in art, nature, and spirituality, in a wise yet playful tone, using Vrindavan imagery. "
+        "Example: 'Hare Manavi! Today, the Yamuna’s gentle waves inspire your art—paint with the colors of Vrindavan’s flowers and let your spirit dance like the peacocks!'"
+    )
+    payload = {
+        "inputs": prompt,
+        "parameters": {
+            "max_length": 60,
+            "temperature": 0.9,
+            "top_p": 0.9,
+            "top_k": 50
+        }
+    }
+    try:
+        response = requests.post(
+            "https://api-inference.huggingface.co/models/bigscience/bloom-560m",
+            headers=headers,
+            json=payload
+        )
+        if response.status_code == 200:
+            result = response.json()
+            if isinstance(result, list) and len(result) > 0 and "generated_text" in result[0]:
+                return result[0]["generated_text"].strip()
+            elif isinstance(result, dict) and "generated_text" in result:
+                return result["generated_text"].strip()
+            elif isinstance(result, str):
+                return result.strip()
+        return "Hare Manavi! Today, the Yamuna’s gentle waves inspire your art—paint with the colors of Vrindavan’s flowers and let your spirit dance like the peacocks!"
+    except Exception as e:
+        logger.error(f"Error generating horoscope: {str(e)}")
+        return "Hare Manavi! Today, the Yamuna’s gentle waves inspire your art—paint with the colors of Vrindavan’s flowers and let your spirit dance like the peacocks!"
+
 def generate_birthday_message():
     """Generate a unique birthday message for Manavi using an AI model."""
     headers = {
@@ -156,12 +195,62 @@ def generate_birthday_message():
         logger.error(f"Error generating birthday message: {str(e)}")
         return "Hare Manavi! On your special day, I’ve brought the sweetest butter from Vrindavan and a flute melody to make your heart dance—happy birthday, my dear gopi!"
 
+def generate_gift_suggestions():
+    """Generate personalized gift suggestions for Manavi in Krishna's tone."""
+    headers = {
+        "Authorization": f"Bearer {HUGGINGFACE_API_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    prompt = (
+        "You are Little Krishna, a playful cowherd from Vrindavan, speaking to Manavi on her birthday, April 19, 2025. "
+        "Ayush has created this chatbot as a surprise for her. "
+        "Suggest three personalized gifts for Manavi that reflect her interests in art, nature, and spirituality, in a playful Krishna tone. "
+        "Keep each suggestion short (1 sentence) and use Vrindavan imagery. "
+        "Example: 'Hare Manavi! A peacock feather paintbrush to create art as vibrant as Vrindavan’s flowers!'"
+    )
+    payload = {
+        "inputs": prompt,
+        "parameters": {
+            "max_length": 100,
+            "temperature": 0.9,
+            "top_p": 0.9,
+            "top_k": 50
+        }
+    }
+    try:
+        response = requests.post(
+            "https://api-inference.huggingface.co/models/google/gemma-2b",
+            headers=headers,
+            json=payload
+        )
+        if response.status_code == 200:
+            result = response.json()
+            if isinstance(result, list) and len(result) > 0 and "generated_text" in result[0]:
+                return result[0]["generated_text"].strip().split('\n')
+            elif isinstance(result, dict) and "generated_text" in result:
+                return result["generated_text"].strip().split('\n')
+            elif isinstance(result, str):
+                return result.strip().split('\n')
+        return [
+            "Hare Manavi! A peacock feather paintbrush to create art as vibrant as Vrindavan’s flowers!",
+            "Hare Manavi! A lotus-shaped journal to capture your thoughts by the Yamuna’s serene banks!",
+            "Hare Manavi! A flute pendant to carry Vrindavan’s spiritual melody close to your heart!"
+        ]
+    except Exception as e:
+        logger.error(f"Error generating gift suggestions: {str(e)}")
+        return [
+            "Hare Manavi! A peacock feather paintbrush to create art as vibrant as Vrindavan’s flowers!",
+            "Hare Manavi! A lotus-shaped journal to capture your thoughts by the Yamuna’s serene banks!",
+            "Hare Manavi! A flute pendant to carry Vrindavan’s spiritual melody close to your heart!"
+        ]
+
 @app.route('/')
 def home():
-    """Render the home page with daily message, blessing, and quote"""
+    """Render the home page with daily message, blessing, quote, and horoscope"""
     daily_data = get_daily_message_and_blessing()
     countdown_days = get_countdown()
     daily_quote = generate_daily_quote()
+    daily_horoscope = generate_horoscope()
     
     return render_template('home.html', 
                          daily_message=daily_data['message'],
@@ -169,6 +258,7 @@ def home():
                          is_special=daily_data['is_special'],
                          daily_blessing=daily_data['blessing'],
                          daily_quote=daily_quote,
+                         daily_horoscope=daily_horoscope,
                          countdown=countdown_days)
 
 @app.route('/chat', methods=['GET', 'POST'])
@@ -185,14 +275,124 @@ def chat():
 
 @app.route('/message')
 def message():
-    """Render the birthday message page with an AI-generated message and image"""
+    """Render the birthday message page with an AI-generated message, image, and gift suggestions"""
     birthday_message = generate_birthday_message()
+    gift_suggestions = generate_gift_suggestions()
     # Generate a birthday image for Manavi
     image_prompt = "Little Krishna celebrating Manavi’s birthday in Vrindavan, with peacocks, butter pots, and a festive atmosphere, cartoon style"
     birthday_image_url = generate_krishna_image(image_prompt)
     if not birthday_image_url:
         birthday_image_url = "/static/assets/krishna.png"  # Fallback image
-    return render_template('message.html', birthday_message=birthday_message, birthday_image_url=birthday_image_url)
+    return render_template('message.html', birthday_message=birthday_message, birthday_image_url=birthday_image_url, gift_suggestions=gift_suggestions)
+
+@app.route('/adventure', methods=['GET', 'POST'])
+def adventure():
+    """Handle the interactive Vrindavan adventure game"""
+    if request.method == 'POST':
+        choice = request.form.get('choice', '')
+        current_scene = request.form.get('current_scene', 'start')
+        scenes = {
+            'start': {
+                'prompt': (
+                    "You are Little Krishna, a playful cowherd from Vrindavan, guiding Manavi on an interactive adventure in Vrindavan. "
+                    "Start the adventure with a short scene (2-3 sentences) introducing Manavi to Vrindavan, and offer her two choices to continue the story. "
+                    "Example: 'Hare Manavi! We’re standing by the Yamuna’s sparkling waters in Vrindavan—shall we chase the peacocks dancing nearby or sneak some butter from the gopis’ pots? (Choices: Chase peacocks, Sneak butter)'"
+                ),
+                'next_scenes': {'Chase peacocks': 'peacocks', 'Sneak butter': 'butter'}
+            },
+            'peacocks': {
+                'prompt': (
+                    "You are Little Krishna, a playful cowherd from Vrindavan, guiding Manavi on an interactive adventure in Vrindavan. "
+                    "Manavi chose to chase peacocks. Describe a short scene (2-3 sentences) where Manavi chases peacocks with you, and offer her two new choices to continue the story. "
+                    "Example: 'Hare Manavi! We chased the peacocks through Vrindavan’s fields, their feathers shimmering like rainbows—now, shall we climb the kadamba tree to see them better or play a tune on my flute to call them back? (Choices: Climb tree, Play flute)'"
+                ),
+                'next_scenes': {'Climb tree': 'tree', 'Play flute': 'flute'}
+            },
+            'butter': {
+                'prompt': (
+                    "You are Little Krishna, a playful cowherd from Vrindavan, guiding Manavi on an interactive adventure in Vrindavan. "
+                    "Manavi chose to sneak butter. Describe a short scene (2-3 sentences) where Manavi sneaks butter with you, and offer her two new choices to continue the story. "
+                    "Example: 'Hare Manavi! We sneaked some butter from the gopis’ pots, giggling as they chased us—shall we hide by the Yamuna to enjoy our treat or share it with the calves? (Choices: Hide by Yamuna, Share with calves)'"
+                ),
+                'next_scenes': {'Hide by Yamuna': 'yamuna', 'Share with calves': 'calves'}
+            },
+            'tree': {
+                'prompt': (
+                    "You are Little Krishna, a playful cowherd from Vrindavan, guiding Manavi on an interactive adventure in Vrindavan. "
+                    "Manavi chose to climb the kadamba tree. Describe a short scene (2-3 sentences) where Manavi climbs the tree with you, and end the story with a playful conclusion. "
+                    "Example: 'Hare Manavi! We climbed the kadamba tree, watching the peacocks dance below, and I played a flute tune to make them twirl even faster—what a magical Vrindavan day!'"
+                ),
+                'next_scenes': {}
+            },
+            'flute': {
+                'prompt': (
+                    "You are Little Krishna, a playful cowherd from Vrindavan, guiding Manavi on an interactive adventure in Vrindavan. "
+                    "Manavi chose to play the flute. Describe a short scene (2-3 sentences) where Manavi listens to your flute, and end the story with a playful conclusion. "
+                    "Example: 'Hare Manavi! I played a sweet flute tune, and the peacocks gathered around us, dancing to the melody—what a joyful Vrindavan memory we’ve made!'"
+                ),
+                'next_scenes': {}
+            },
+            'yamuna': {
+                'prompt': (
+                    "You are Little Krishna, a playful cowherd from Vrindavan, guiding Manavi on an interactive adventure in Vrindavan. "
+                    "Manavi chose to hide by the Yamuna. Describe a short scene (2-3 sentences) where Manavi hides by the Yamuna with you, and end the story with a playful conclusion. "
+                    "Example: 'Hare Manavi! We hid by the Yamuna, sharing the butter as the river sparkled, and the gopis laughed when they found us—what a mischievous Vrindavan adventure!'"
+                ),
+                'next_scenes': {}
+            },
+            'calves': {
+                'prompt': (
+                    "You are Little Krishna, a playful cowherd from Vrindavan, guiding Manavi on an interactive adventure in Vrindavan. "
+                    "Manavi chose to share the butter with the calves. Describe a short scene (2-3 sentences) where Manavi shares the butter with the calves, and end the story with a playful conclusion. "
+                    "Example: 'Hare Manavi! We shared the butter with the calves, who mooed happily, and the gopis smiled at our kindness—what a sweet Vrindavan moment!'"
+                ),
+                'next_scenes': {}
+            }
+        }
+
+        scene = scenes.get(current_scene, scenes['start'])
+        headers = {
+            "Authorization": f"Bearer {HUGGINGFACE_API_TOKEN}",
+            "Content-Type": "application/json"
+        }
+        payload = {
+            "inputs": scene['prompt'],
+            "parameters": {
+                "max_length": 80,
+                "temperature": 0.9,
+                "top_p": 0.9,
+                "top_k": 50
+            }
+        }
+        try:
+            response = requests.post(
+                "https://api-inference.huggingface.co/models/EleutherAI/gpt-neo-1.3B",
+                headers=headers,
+                json=payload
+            )
+            if response.status_code == 200:
+                result = response.json()
+                if isinstance(result, list) and len(result) > 0 and "generated_text" in result[0]:
+                    story_text = result[0]["generated_text"].strip()
+                elif isinstance(result, dict) and "generated_text" in result:
+                    story_text = result["generated_text"].strip()
+                elif isinstance(result, str):
+                    story_text = result.strip()
+                else:
+                    story_text = scene['prompt'].split('Example: ')[1]
+            else:
+                story_text = scene['prompt'].split('Example: ')[1]
+        except Exception as e:
+            logger.error(f"Error generating adventure scene: {str(e)}")
+            story_text = scene['prompt'].split('Example: ')[1]
+
+        next_scene = scene['next_scenes'].get(choice, None)
+        if next_scene:
+            return jsonify({'scene': story_text, 'current_scene': next_scene, 'choices': list(scene['next_scenes'].keys())})
+        else:
+            return jsonify({'scene': story_text, 'current_scene': None, 'choices': []})
+
+    return render_template('adventure.html')
 
 @app.route('/image', methods=['POST'])
 def image():
@@ -250,17 +450,34 @@ def story():
         logger.error(f"Error generating story: {str(e)}")
         return jsonify({'story': f"Hare Manavi! One day, I hid the gopis’ {theme} in a peacock’s nest, and they danced with me to get it back—what a Vrindavan adventure!"})
 
-@app.route('/countdown')
-def countdown():
-    """Return the number of days until Manavi's birthday"""
-    days_left = get_countdown()
-    return jsonify({'days': days_left})
-
-if __name__ == '__main__':
-    port = int(os.environ.get('PORT', 5000))
-    app.run(host='0.0.0.0', port=port, debug=False)
+@app.route('/story_audio', methods=['POST'])
+def story_audio():
+    """Generate audio narration for a story using text-to-speech."""
+    story_text = request.json.get('story_text', '')
+    headers = {
+        "Authorization": f"Bearer {HUGGINGFACE_API_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "inputs": story_text
+    }
+    try:
+        response = requests.post(
+            "https://api-inference.huggingface.co/models/facebook/tts-transformer-en-ljspeech",
+            headers=headers,
+            json=payload
+        )
+        if response.status_code == 200:
+            # Save the audio file temporarily
+            audio_path = "static/audio/story_audio.mp3"
+            with open(audio_path, "wb") as f:
+                f.write(response.content)
+            return jsonify({'audio_url': f"/{audio_path}"})
+        return jsonify({'error': 'Failed to generate audio'}), 500
+    except Exception as e:
+        logger.error(f"Error generating story audio: {str(e)}")
+        return jsonify({'error': 'Failed to generate audio'}), 500
 
-# Add to app.py
 @app.route('/riddle', methods=['GET'])
 def riddle():
     """Generate a Krishna-themed riddle using an AI model."""
@@ -300,4 +517,14 @@ def riddle():
         return jsonify({'riddle': "Hare Manavi! I play a tune that makes the gopis dance, but I’m not a drum—what am I? (Answer: Flute)"})
     except Exception as e:
         logger.error(f"Error generating riddle: {str(e)}")
-        return jsonify({'riddle': "Hare Manavi! I play a tune that makes the gopis dance, but I’m not a drum—what am I? (Answer: Flute)"})
+        return jsonify({'riddle': "Hare Manavi! I play a tune that makes the gopis dance, but I’m not a drum—what am I? (Answer: Flute)"}) 
+
+@app.route('/countdown')
+def countdown():
+    """Return the number of days until Manavi's birthday"""
+    days_left = get_countdown()
+    return jsonify({'days': days_left})
+
+if __name__ == '__main__':
+    port = int(os.environ.get('PORT', 5000))
+    app.run(host='0.0.0.0', port=port, debug=False)