from flask import Flask, request, jsonify, render_template
from dotenv import load_dotenv
import google.generativeai as genai
import os

load_dotenv()
os.getenv("langchain_google_genai")
os.environ['GOOGLE_API_KEY'] = 'AIzaSyAVFvEwx35owbCnkCpP5w2jbk6SXppFegs'
genai.configure(api_key="AIzaSyAVFvEwx35owbCnkCpP5w2jbk6SXppFegs")

app = Flask(__name__)

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

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

def question_answer(question):
    model = genai.GenerativeModel(model_name="gemini-pro")
    response = model.generate_content(question)
    return response.text

@app.route('/send-message', methods=['POST'])
def send_message():
    message = request.json.get('message')
    print(f"User message: {message}")

    response = question_answer(message)
    print(f"Ours: {response}")
    return jsonify({'response': response})

@app.route('/login', methods=['POST'])
def login():
    data = request.get_json()
    email = data.get('email')
    print(data)
    password = data.get('password')

    if email == 'abc@gmail' and password == 'abc@gmail':
        message = f"Login successful with {email}"
        print(message)
    else:
        message = "Invalid email or password"

    return jsonify({'message': message})


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=7860, debug=True)