Update app.py

app.py

@@ -5,12 +5,13 @@ from dotenv import load_dotenv
 from chatbot import get_krishna_response
 from image_api import generate_krishna_image, generate_comic_strip
 from countdown import get_countdown
-from messages import daily_blessings, birthday_messages, auto_generate_birthday_message
+from messages import daily_blessings, auto_generate_birthday_message
+from ayush_messages import ayush_surprises
 import logging
 import requests
 import random
 import backoff
-from ayush_messages import ayush_surprises
+from functools import lru_cache
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -21,21 +22,21 @@ try:
     from firebase_config import save_chat_message, get_chat_history
     firebase_enabled = True
     logger.info("Firebase integration enabled")
-except ImportError as e:
+except ImportError:
     firebase_enabled = False
-    logger.warning(f"Firebase disabled: {str(e)}")
+    logger.info("Firebase disabled: firebase_config.py not found")
 
 # Load environment variables
 load_dotenv()
 HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
 if not HUGGINGFACE_API_TOKEN:
-    logger.error("HUGGINGFACE_API_TOKEN not found in environment variables.")
-    raise ValueError("HUGGINGFACE_API_TOKEN is required.")
+    logger.error("HUGGINGFACE_API_TOKEN not found in environment variables")
+    raise ValueError("HUGGINGFACE_API_TOKEN is required")
 
 app = Flask(__name__)
 app.config['JSONIFY_PRETTYPRINT_REGULAR'] = True
 
-# Daily messages based on the day of the week (with Bhagavad Gita verses)
+# Daily messages with Bhagavad Gita verses
 DAILY_MESSAGES = {
     0: {"message": "On Sundays, shine like the sun for all!", "verse": "Bhagavad Gita 9.29"},
     1: {"message": "Monday: Do your duty without attachment.", "verse": "Bhagavad Gita 2.47"},
@@ -46,7 +47,7 @@ DAILY_MESSAGES = {
     6: {"message": "Saturday: Find joy in spiritual practice.", "verse": "Bhagavad Gita 6.17"}
 }
 
-# Festival messages for specific dates
+# Festival 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},
@@ -54,7 +55,7 @@ FESTIVAL_MESSAGES = {
 }
 
 def parse_huggingface_response(response):
-    """Parse Hugging Face API response and extract generated text."""
+    """Parse Hugging Face API response."""
     try:
         result = response.json()
         if isinstance(result, list) and result and "generated_text" in result[0]:
@@ -63,20 +64,19 @@ def parse_huggingface_response(response):
             return result["generated_text"].strip()
         elif isinstance(result, str):
             return result.strip()
-        logger.error("Unexpected API response format.")
+        logger.error("Unexpected API response format")
         return None
     except Exception as e:
         logger.error(f"Error parsing API response: {str(e)}")
         return None
 
 def get_daily_message_and_blessing():
-    """Retrieve the daily message and blessing based on the current date."""
+    """Retrieve daily message and blessing."""
     try:
         today = datetime.now()
         day_of_week = (today.weekday() + 1) % 7
         month_day = (today.month, today.day)
 
-        # Check for festival messages
         if month_day in FESTIVAL_MESSAGES:
             daily_message = FESTIVAL_MESSAGES[month_day]
         else:
@@ -102,20 +102,20 @@ def get_daily_message_and_blessing():
             "blessing": "Hare Manavi! May Vrindavan’s joy fill your heart!"
         }
 
-@backoff.on_exception(backoff.expo, (requests.exceptions.RequestException, requests.exceptions.HTTPError), max_tries=12, max_time=900)
-def make_api_request(url, headers, payload, timeout=180):
+@backoff.on_exception(backoff.expo, (requests.exceptions.RequestException, requests.exceptions.HTTPError), max_tries=2, max_time=60)
+def make_api_request(url, headers, payload, timeout=30):
     """Make API requests with exponential backoff."""
     try:
         response = requests.post(url, headers=headers, json=payload, timeout=timeout)
         if response.status_code == 200:
             return response
-        else:
-            logger.error(f"API error: {response.status_code} - {response.text}")
-            raise requests.exceptions.HTTPError(f"API returned status {response.status_code}")
+        logger.error(f"API error: {response.status_code} - {response.text}")
+        raise requests.exceptions.HTTPError(f"API returned status {response.status_code}")
     except requests.exceptions.RequestException as e:
         logger.error(f"API request failed: {str(e)}")
         raise
 
+@lru_cache(maxsize=10)
 def generate_birthday_message():
     """Generate a birthday message for Manavi."""
     try:
@@ -148,12 +148,13 @@ def generate_birthday_message():
         result = parse_huggingface_response(response)
         if result and result.startswith("Happy Birthday, Manavi!"):
             return result
-        logger.warning("Invalid API response; returning auto-generated message.")
+        logger.warning("Invalid API response; using fallback")
         return auto_generate_birthday_message(include_tease=True)
     except Exception as e:
         logger.error(f"Error in generate_birthday_message: {str(e)}")
         return auto_generate_birthday_message(include_tease=True)
 
+@lru_cache(maxsize=10)
 def generate_gift_suggestions():
     """Generate personalized gift suggestions for Manavi."""
     try:
@@ -184,7 +185,7 @@ def generate_gift_suggestions():
         if result:
             suggestions = result.strip().split('\n')
             return [s.strip() for s in suggestions if s.strip()]
-        logger.warning("Returning fallback gift suggestions.")
+        logger.warning("Using fallback gift suggestions")
         return [
             "Hare Manavi! A peacock feather paintbrush for vibrant art!",
             "Hare Manavi! A lotus journal for serene thoughts!",
@@ -198,6 +199,7 @@ def generate_gift_suggestions():
             "Hare Manavi! A flute pendant for spiritual melodies!"
         ]
 
+@lru_cache(maxsize=10)
 def generate_horoscope():
     """Generate a daily horoscope for Manavi."""
     try:
@@ -228,7 +230,7 @@ def generate_horoscope():
         result = parse_huggingface_response(response)
         if result:
             return result
-        logger.warning("Returning fallback horoscope.")
+        logger.warning("Using fallback horoscope")
         return "Hare Manavi! Today, the Yamuna inspires your art—paint with Vrindavan’s colors and let your spirit dance!"
     except Exception as e:
         logger.error(f"Error in generate_horoscope: {str(e)}")
@@ -250,7 +252,7 @@ def home():
                              countdown=countdown_days)
     except Exception as e:
         logger.error(f"Error in / route: {str(e)}")
-        return render_template('error.html', error_message='Failed to load homepage.'), 500
+        return render_template('error.html', error_message='Failed to load homepage. Please try again.'), 500
 
 @app.route('/chat', methods=['GET', 'POST'])
 def chat():
@@ -258,14 +260,14 @@ def chat():
     try:
         if request.method == 'POST':
             if 'message' not in request.form:
-                logger.error("Missing 'message' in POST request.")
+                logger.error("Missing 'message' in POST request")
                 return jsonify({'error': 'Missing message parameter'}), 400
             user_input = escape(request.form['message'].strip())
             if not user_input:
-                logger.error("Empty message received.")
+                logger.error("Empty message received")
                 return jsonify({'error': 'Message cannot be empty'}), 400
             if len(user_input) > 500:
-                logger.error("Message too long.")
+                logger.error("Message too long")
                 return jsonify({'error': 'Message too long (max 500 characters)'}), 400
             logger.info(f"Received chat input: {user_input}")
             reply = get_krishna_response(user_input)
@@ -273,7 +275,7 @@ def chat():
             if firebase_enabled:
                 try:
                     save_chat_message(user_input, reply)
-                    logger.info("Saved chat message to Firebase.")
+                    logger.info("Saved chat message to Firebase")
                 except Exception as e:
                     logger.error(f"Failed to save chat message: {str(e)}")
             return jsonify({'reply': reply})
@@ -286,7 +288,7 @@ def chat():
         return render_template('chat.html', chat_history=chat_history)
     except Exception as e:
         logger.error(f"Error in /chat route: {str(e)}")
-        return render_template('error.html', error_message='Failed to load chat page.'), 500
+        return render_template('error.html', error_message='Failed to load chat page. Please try again.'), 500
 
 @app.route('/adventure', methods=['GET', 'POST'])
 def adventure():
@@ -373,7 +375,7 @@ def adventure():
                 result = parse_huggingface_response(response)
                 story_text = result or scene['prompt'].split('Example: ')[1]
             else:
-                logger.error("Failed to generate adventure scene.")
+                logger.error("Failed to generate adventure scene")
                 story_text = scene['prompt'].split('Example: ')[1]
 
             next_scene = scene['next_scenes'].get(choice)
@@ -384,7 +386,7 @@ def adventure():
         return render_template('adventure.html')
     except Exception as e:
         logger.error(f"Error in /adventure route: {str(e)}")
-        return render_template('error.html', error_message='Failed to load adventure page.'), 500
+        return render_template('error.html', error_message='Failed to load adventure page. Please try again.'), 500
 
 @app.route('/message')
 def message():
@@ -402,7 +404,7 @@ def message():
                              is_birthday=is_birthday)
     except Exception as e:
         logger.error(f"Error in /message route: {str(e)}")
-        return render_template('error.html', error_message='Failed to load birthday message page.'), 500
+        return render_template('error.html', error_message='Failed to load birthday message page. Please try again.'), 500
 
 @app.route('/image', methods=['POST'])
 def image():
@@ -417,7 +419,7 @@ def image():
         return jsonify({'error': 'Failed to generate image'}), 500
     except Exception as e:
         logger.error(f"Error in /image route: {str(e)}")
-        return jsonify({'error': 'Internal Server Error.'}), 500
+        return jsonify({'error': 'Internal Server Error. Please try again.'}), 500
 
 @app.route('/comic', methods=['GET'])
 def comic():
@@ -427,7 +429,7 @@ def comic():
         return jsonify({'comic_images': comic_images})
     except Exception as e:
         logger.error(f"Error in /comic route: {str(e)}")
-        return jsonify({'error': 'Internal Server Error.'}), 500
+        return jsonify({'error': 'Internal Server Error. Please try again.'}), 500
 
 @app.route('/story', methods=['POST'])
 def story():
@@ -459,11 +461,11 @@ def story():
         result = parse_huggingface_response(response)
         if result:
             return jsonify({'story': result})
-        logger.error("Failed to generate story.")
+        logger.error("Failed to generate story")
         return jsonify({'story': f"Hare Manavi! I hid the gopis’ {theme} in a peacock’s nest—what a Vrindavan adventure!'"})
     except Exception as e:
         logger.error(f"Error in /story route: {str(e)}")
-        return jsonify({'error': 'Internal Server Error.'}), 500
+        return jsonify({'error': 'Internal Server Error. Please try again.'}), 500
 
 @app.route('/story_audio', methods=['POST'])
 def story_audio():
@@ -488,11 +490,11 @@ def story_audio():
             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.")
+        logger.error("Failed to generate story audio")
         return jsonify({'error': 'Failed to generate audio'}), 500
     except Exception as e:
         logger.error(f"Error in /story_audio route: {str(e)}")
-        return jsonify({'error': 'Internal Server Error.'}), 500
+        return jsonify({'error': 'Internal Server Error. Please try again.'}), 500
 
 @app.route('/riddle', methods=['GET'])
 def riddle():
@@ -523,11 +525,11 @@ def riddle():
         result = parse_huggingface_response(response)
         if result:
             return jsonify({'riddle': result})
-        logger.error("Failed to generate riddle.")
+        logger.error("Failed to generate riddle")
         return jsonify({'riddle': "Hare Manavi! I play a tune that makes gopis dance—what am I? (Answer: Flute)"})
     except Exception as e:
         logger.error(f"Error in /riddle route: {str(e)}")
-        return jsonify({'error': 'Internal Server Error.'}), 500
+        return jsonify({'error': 'Internal Server Error. Please try again.'}), 500
 
 @app.route('/countdown')
 def countdown():
@@ -537,7 +539,7 @@ def countdown():
         return jsonify({'days': days_left})
     except Exception as e:
         logger.error(f"Error in /countdown route: {str(e)}")
-        return jsonify({'error': 'Internal Server Error.'}), 500
+        return jsonify({'error': 'Internal Server Error. Please try again.'}), 500
 
 if __name__ == '__main__':
     port = int(os.environ.get('PORT', 7860))