Update app.py

app.py

@@ -7,6 +7,7 @@ from image_api import generate_krishna_image, generate_comic_strip
 from countdown import get_countdown
 from messages import daily_blessings
 import logging
+import requests
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -23,7 +24,7 @@ except ImportError as e:
 
 # Load environment variables
 load_dotenv()
-
+HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
 app = Flask(__name__)
 app.config['JSONIFY_PRETTYPRINT_REGULAR'] = True
 
@@ -42,7 +43,7 @@ DAILY_MESSAGES = {
 FESTIVAL_MESSAGES = {
     (1, 26): {"message": "Happy Republic Day! Let Dharma guide our nation.", "verse": "Bhagavad Gita 4.7", "special": True},
     (8, 19): {"message": "Happy Krishna Janmashtami! Celebrate the divine birth.", "verse": "Bhagavad Gita 4.9", "special": True},
-    (4, 19): {"message": "Happy Birthday, Manavi! I’ve brought Vrindavan’s magic to celebrate with you!", "verse": "Bhagavad Gita 2.22", "special": True}  # Manavi's birthday
+    (4, 19): {"message": "Happy Birthday, Manavi! I’ve brought Vrindavan’s magic to celebrate with you!", "verse": "Bhagavad Gita 2.22", "special": True}
 }
 
 def get_daily_message_and_blessing():
@@ -72,6 +73,46 @@ def get_daily_message_and_blessing():
         "blessing": daily_blessing
     }
 
+def generate_birthday_message():
+    """Generate a unique birthday message for Manavi using an AI model."""
+    headers = {
+        "Authorization": f"Bearer {HUGGINGFACE_API_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    prompt = (
+        "You are Little Krishna, a playful and loving cowherd from Vrindavan, speaking to Manavi on her birthday, April 19, 2025. "
+        "Ayush has created this chatbot as a surprise for her, and you are his wingman. "
+        "Write a short, heartfelt birthday message (1-2 sentences) for Manavi, filled with Vrindavan imagery and a playful tone. "
+        "Example: '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!'"
+    )
+    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! 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!"
+
 @app.route('/')
 def home():
     """Render the home page with daily message and blessing"""
@@ -98,7 +139,13 @@ def chat():
 
 @app.route('/message')
 def message():
-    return render_template('message.html')
+    birthday_message = generate_birthday_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:
+        birthday_image_url = "/static/assets/krishna.png"  # Fallback image
+    return render_template('message.html', birthday_message=birthday_message, birthday_image_url=birthday_image_url)
 
 @app.route('/image', methods=['POST'])
 def image():
@@ -119,5 +166,5 @@ def countdown():
     return jsonify({'days': days_left})
 
 if __name__ == '__main__':
-    port = int(os.environ.get('PORT', 5000))  # Use port 5000 for python SDK
+    port = int(os.environ.get('PORT', 5000))
     app.run(host='0.0.0.0', port=port, debug=False)