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 countdown import get_countdown

load_dotenv()
app = Flask(__name__)

@app.route('/')
def home():
    return render_template('home.html')

@app.route('/chat', methods=['GET', 'POST'])
def chat():
    if request.method == 'POST':
        user_input = request.form['message']
        reply = get_krishna_response(user_input)
        return jsonify({'reply': reply})
    return render_template('chat.html')

@app.route('/message')
def message():
    return render_template('message.html')

@app.route('/image', methods=['POST'])
def image():
    prompt = request.json['prompt']
    image_url = generate_krishna_image(prompt)
    return jsonify({'image_url': image_url})

@app.route('/countdown')
def countdown():
    days_left = get_countdown()  # Days until April 19, 2025
    return jsonify({'days': days_left})

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000)