Update backend/app.py

backend/app.py

@@ -2,7 +2,7 @@ from flask import Flask, render_template, request, jsonify
 import os
 from dotenv import load_dotenv
 from chatbot import get_krishna_response
-from dalle_api import generate_krishna_image
+from dalle_api import generate_krishna_image, generate_comic_strip
 from countdown import get_countdown
 
 load_dotenv()
@@ -28,11 +28,18 @@ def message():
 def image():
     prompt = request.json['prompt']
     image_url = generate_krishna_image(prompt)
-    return jsonify({'image_url': image_url})
+    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})
 
 @app.route('/countdown')
 def countdown():
-    days_left = get_countdown()  # Days until April 19, 2025
+    days_left = get_countdown()
     return jsonify({'days': days_left})
 
 if __name__ == '__main__':