from flask import Flask, jsonify, request
from flask_cors import CORS
import g4f

app = Flask(__name__)

CORS(app)

@app.route('/chat-bot', methods=['POST'])
def chatbot():
    if request.method == 'POST':
        data = request.get_json()
        query = data.get('query')

        print(data)

        response = g4f.ChatCompletion.create(
            model="gpt-3.5-turbo",
            provider=g4f.Provider.You,
            messages=[{"role": "user", "content": query}],
        )

        print(response)

        return jsonify({'response': str(response)})