Update app.py

app.py

@@ -9,6 +9,7 @@ from messages import daily_blessings
 import logging
 import requests
 import random
+import time
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -55,7 +56,6 @@ def get_daily_message_and_blessing():
     # weekday() returns 0 for Monday, 6 for Sunday, so we shift it: (weekday() + 1) % 7
     day_of_week = (today.weekday() + 1) % 7  # Maps Monday (0) to 1, Tuesday (1) to 2, ..., Sunday (6) to 0
     month_day = (today.month, today.day)
-    day_of_month = today.day
 
     # Check for festival messages first
     if month_day in FESTIVAL_MESSAGES:
@@ -67,8 +67,8 @@ def get_daily_message_and_blessing():
                          "verse": "Bhagavad Gita 6.30"})
         daily_message['special'] = False
 
-    # Get the daily blessing based on the day of the month
-    daily_blessing_index = (day_of_month - 1) % len(daily_blessings)
+    # Precompute daily blessing index
+    daily_blessing_index = (today.day - 1) % len(daily_blessings)
     daily_blessing = daily_blessings[daily_blessing_index]
 
     return {
@@ -78,6 +78,27 @@ def get_daily_message_and_blessing():
         "blessing": daily_blessing
     }
 
+def make_api_request(url, headers, payload, retries=3, delay=5):
+    """Helper function to make API requests with retry logic."""
+    for attempt in range(retries):
+        try:
+            response = requests.post(url, headers=headers, json=payload)
+            if response.status_code == 200:
+                return response
+            elif response.status_code == 429:  # Rate limit
+                logger.warning(f"Rate limit hit on attempt {attempt + 1}. Retrying after {delay} seconds...")
+                time.sleep(delay)
+                continue
+            else:
+                logger.error(f"API error: {response.text}")
+                return None
+        except Exception as e:
+            logger.error(f"API request failed: {str(e)}")
+            if attempt < retries - 1:
+                time.sleep(delay)
+            continue
+    return None
+
 def generate_daily_quote():
     """Generate a daily inspirational quote in Krishna's tone using an AI model."""
     headers = {
@@ -98,34 +119,33 @@ def generate_daily_quote():
             "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]:
-                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! Let your heart bloom like the lotus in the Yamuna, for love is the sweetest melody of Vrindavan!"
-    except Exception as e:
-        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!"
+    response = make_api_request(
+        "https://api-inference.huggingface.co/models/EleutherAI/gpt-neo-1.3B",
+        headers=headers,
+        json=payload
+    )
+    if response:
+        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()
+    logger.error("Failed to generate daily quote after retries.")
+    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."""
+    today = datetime.now().strftime("%B %d, %Y")  # Dynamically get the current date
     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. "
+        f"You are Little Krishna, a wise and playful cowherd from Vrindavan, speaking to Manavi on {today}. "
         "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. "
+        "Do not include any self-referential text about being Little Krishna or Manavi's identity. "
         "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 = {
@@ -137,24 +157,21 @@ def generate_horoscope():
             "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!"
+    response = make_api_request(
+        "https://api-inference.huggingface.co/models/bigscience/bloom-560m",
+        headers=headers,
+        json=payload
+    )
+    if response:
+        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()
+    logger.error("Failed to generate horoscope after retries.")
+    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."""
@@ -177,24 +194,21 @@ def generate_birthday_message():
             "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! 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!"
-    except Exception as e:
-        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!"
+    response = make_api_request(
+        "https://api-inference.huggingface.co/models/bigscience/bloom-560m",
+        headers=headers,
+        json=payload
+    )
+    if response:
+        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()
+    logger.error("Failed to generate birthday message after retries.")
+    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."""
@@ -218,32 +232,25 @@ def generate_gift_suggestions():
             "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!"
-        ]
+    response = make_api_request(
+        "https://api-inference.huggingface.co/models/google/gemma-2b",
+        headers=headers,
+        json=payload
+    )
+    if response:
+        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')
+    logger.error("Failed to generate gift suggestions after retries.")
+    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():
@@ -282,6 +289,10 @@ def message():
     # 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:
+        # Retry with a simpler prompt
+        image_prompt = "Little Krishna with a peacock feather, playing a flute in Vrindavan, 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, gift_suggestions=gift_suggestions)
@@ -365,26 +376,23 @@ def adventure():
                 "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]
+        response = make_api_request(
+            "https://api-inference.huggingface.co/models/EleutherAI/gpt-neo-1.3B",
+            headers=headers,
+            json=payload
+        )
+        if response:
+            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]
-        except Exception as e:
-            logger.error(f"Error generating adventure scene: {str(e)}")
+        else:
+            logger.error("Failed to generate adventure scene after retries.")
             story_text = scene['prompt'].split('Example: ')[1]
 
         next_scene = scene['next_scenes'].get(choice, None)
@@ -458,12 +466,8 @@ def story():
                 "inputs": prompt,
                 "parameters": model["parameters"]
             }
-            response = requests.post(
-                model["endpoint"],
-                headers=headers,
-                json=payload
-            )
-            if response.status_code == 200:
+            response = make_api_request(model["endpoint"], headers=headers, json=payload)
+            if response and response.status_code == 200:
                 result = response.json()
                 if isinstance(result, list) and len(result) > 0 and "generated_text" in result[0]:
                     return jsonify({'story': result[0]["generated_text"].strip()})
@@ -471,7 +475,7 @@ def story():
                     return jsonify({'story': result["generated_text"].strip()})
                 elif isinstance(result, str):
                     return jsonify({'story': result.strip()})
-            print(f"Error with {model['name']}: {response.text}")
+            print(f"Error with {model['name']}: {response.text if response else 'No response'}")
             continue
         except Exception as e:
             print(f"Error connecting to {model['name']}: {str(e)}")
@@ -491,22 +495,19 @@ def story_audio():
     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
+    response = make_api_request(
+        "https://api-inference.huggingface.co/models/facebook/tts-transformer-en-ljspeech",
+        headers=headers,
+        json=payload
+    )
+    if response and 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}"})
+    logger.error("Failed to generate story audio after retries.")
+    return jsonify({'error': 'Failed to generate audio'}), 500
 
 @app.route('/riddle', methods=['GET'])
 def riddle():
@@ -530,24 +531,21 @@ def riddle():
             "top_k": 50
         }
     }
-    try:
-        response = requests.post(
-            "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1",
-            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 jsonify({'riddle': result[0]["generated_text"].strip()})
-            elif isinstance(result, dict) and "generated_text" in result:
-                return jsonify({'riddle': result["generated_text"].strip()})
-            elif isinstance(result, str):
-                return jsonify({'riddle': result.strip()})
-        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)"}) 
+    response = make_api_request(
+        "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1",
+        headers=headers,
+        json=payload
+    )
+    if response and response.status_code == 200:
+        result = response.json()
+        if isinstance(result, list) and len(result) > 0 and "generated_text" in result[0]:
+            return jsonify({'riddle': result[0]["generated_text"].strip()})
+        elif isinstance(result, dict) and "generated_text" in result:
+            return jsonify({'riddle': result["generated_text"].strip()})
+        elif isinstance(result, str):
+            return jsonify({'riddle': result.strip()})
+    logger.error("Failed to generate riddle after retries.")
+    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():