Update app.py

app.py

@@ -1,52 +1,152 @@
-from flask import Flask, render_template, request, jsonify
 import os
+from flask import Flask, render_template, request, jsonify
 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
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Initialize Firebase if available
 try:
     from firebase_config import save_chat_message, get_chat_history
     firebase_enabled = True
-except ImportError:
+    logger.info("Firebase integration enabled")
+except ImportError as e:
     firebase_enabled = False
+    logger.warning(f"Firebase disabled: {str(e)}")
 
-# Load environment variables (Hugging Face Space secrets)
+# Load environment variables
 load_dotenv()
+
 app = Flask(__name__)
+app.config['JSONIFY_PRETTYPRINT_REGULAR'] = True
 
 @app.route('/')
 def home():
+    """Render the home page"""
     return render_template('home.html')
 
 @app.route('/chat', methods=['GET', 'POST'])
 def chat():
+    """Handle chat interactions"""
     if request.method == 'POST':
-        user_input = request.form['message']
-        reply = get_krishna_response(user_input)
-        if firebase_enabled:
-            save_chat_message(user_input, reply)
-        return jsonify({'reply': reply})
+        try:
+            user_input = request.form['message']
+            logger.info(f"Received message: {user_input}")
+            
+            reply = get_krishna_response(user_input)
+            
+            if firebase_enabled:
+                save_chat_message(user_input, reply)
+                
+            return jsonify({
+                'reply': reply,
+                'status': 'success'
+            })
+            
+        except Exception as e:
+            logger.error(f"Chat error: {str(e)}")
+            return jsonify({
+                'error': 'Failed to process message',
+                'status': 'error'
+            }), 500
+            
+    # GET request - load chat history
     chat_history = get_chat_history() if firebase_enabled else []
     return render_template('chat.html', chat_history=chat_history)
 
 @app.route('/message')
-def message():
+def special_message():
+    """Render special message page"""
     return render_template('message.html')
 
 @app.route('/image', methods=['POST'])
-def image():
-    prompt = request.json['prompt']
-    image_url = generate_krishna_image(prompt)
-    if image_url:
-        return jsonify({'image_url': image_url})
-    return jsonify({'error': 'Failed to generate image'}), 500
-
-@app.route('/comic', methods=['GET'])
-def comic():
-    comic_images = generate_comic_strip()
-    return jsonify({'comic_images': comic_images})
+def generate_image():
+    """Generate Krishna image from prompt"""
+    try:
+        prompt = request.json.get('prompt', 'Lord Krishna playing flute')
+        logger.info(f"Generating image for prompt: {prompt}")
+        
+        image_url = generate_krishna_image(prompt)
+        
+        if not image_url:
+            raise ValueError("Image generation failed")
+            
+        return jsonify({
+            'image_url': image_url,
+            'status': 'success'
+        })
+        
+    except Exception as e:
+        logger.error(f"Image generation error: {str(e)}")
+        return jsonify({
+            'error': 'Image generation failed',
+            'status': 'error'
+        }), 500
+
+@app.route('/comic')
+def generate_comic():
+    """Generate comic strip"""
+    try:
+        comic_images = generate_comic_strip()
+        return jsonify({
+            'comic_images': comic_images,
+            'status': 'success'
+        })
+    except Exception as e:
+        logger.error(f"Comic generation error: {str(e)}")
+        return jsonify({
+            'error': 'Comic generation failed',
+            'status': 'error'
+        }), 500
 
 @app.route('/countdown')
 def countdown():
-    days_left = get_countdown()
-    return jsonify({'days': days_left})
+    """Get days until special event"""
+    try:
+        days_left = get_countdown()
+        return jsonify({
+            'days': days_left,
+            'status': 'success'
+        })
+    except Exception as e:
+        logger.error(f"Countdown error: {str(e)}")
+        return jsonify({
+            'error': 'Could not calculate countdown',
+            'status': 'error'
+        }), 500
+
+@app.route('/health')
+def health_check():
+    """Health check endpoint"""
+    return jsonify({
+        'status': 'healthy',
+        'services': {
+            'firebase': firebase_enabled,
+            'chatbot': True,
+            'image_generation': True
+        }
+    })
+
+@app.errorhandler(404)
+def not_found(e):
+    return jsonify({
+        'error': 'Endpoint not found',
+        'status': 'error'
+    }), 404
+
+@app.errorhandler(500)
+def server_error(e):
+    logger.error(f"Server error: {str(e)}")
+    return jsonify({
+        'error': 'Internal server error',
+        'status': 'error'
+    }), 500
+
+if __name__ == '__main__':
+    port = int(os.environ.get('PORT', 7860))
+    app.run(host='0.0.0.0', port=port, debug=False)