Update backend/app.py

backend/app.py

@@ -2,9 +2,15 @@ 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, generate_comic_strip
+from image_api import generate_krishna_image, generate_comic_strip
 from countdown import get_countdown
+try:
+    from firebase_config import save_chat_message, get_chat_history
+    firebase_enabled = True
+except ImportError:
+    firebase_enabled = False
 
+# Load environment variables (Hugging Face Space secrets)
 load_dotenv()
 app = Flask(__name__)
 
@@ -17,8 +23,12 @@ def chat():
     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})
-    return render_template('chat.html')
+    # Fetch chat history for display
+    chat_history = get_chat_history() if firebase_enabled else []
+    return render_template('chat.html', chat_history=chat_history)
 
 @app.route('/message')
 def message():