from flask import Flask, request, jsonify
import reka
import api_info

app = Flask(__name__)

@app.route('/')
def initial():
    return '<pre>' + api_info.default_info + '</pre>'

@app.route("/available_models", methods=['GET'])
def available_models():
    return jsonify(api_info.available_models)

@app.route("/endpoints", methods=['GET'])
def endpoints():
    return jsonify(api_info.endpoint)

@app.route("/developer_info", methods=['GET'])
def developer_info():
    return jsonify(api_info.developer_info)

@app.route('/generate', methods=['GET'])
def generate():
    
    query = request.args.get('query') 
    system_prompt = str(request.args.get('system', "Be Helpful and Friendly. Keep your response straightfoward, short and concise"))  # Optional parameter with default value
    model = str(request.args.get('model', "reka-core"))  # Optional parameter with default value
    use_search_engine = bool(request.args.get('use_search_engine', False))  # Optional parameter with default value
    use_code_interpreter = bool(request.args.get('use_code_interpreter', False))  # Optional parameter with default value

    if query:
        response = reka.generate(query, model=model, system_prompt=system_prompt, use_search_engine=use_search_engine, use_code_interpreter=use_code_interpreter)
        return jsonify([{'response': response}, {'developer_info': api_info.developer_info}]), 200

    else:
        return jsonify(api_info.error_message), 400

if __name__ == '__main__':
    app.run(debug=True)