newFlaskTest.py

from flask import Flask, request, jsonify, render_template, session
import copy
import os
import dm_ai

app = Flask(__name__)
app.secret_key = os.urandom(24)

@app.route('/')
def index():
    session.pop('chat', None)
    session['chat'] = session.get('chat', copy.deepcopy(conversation_history))
    return render_template('newIndex.html')

@app.route('/submit_form', methods=['POST'])
def submit_form():
    global client

    response_message = dm_ai.generate_chat(client, request.form['message'], session['chat'])
    session.modified = True
    print("Chat history:", session['chat'])

    response_data = {
        'status': 'success',
        'message': response_message
    }
    return jsonify(response_data)

if __name__ == '__main__':
    client = dm_ai.create_openai_client()
    conversation_history = [{"role": "system", "content": dm_ai.system_prompt}]
    dm_ai.system_token = dm_ai.num_tokens_from_messages(conversation_history, model=dm_ai.MODEL)
    dm_ai.encoding = dm_ai.tiktoken.encoding_for_model(dm_ai.MODEL)
    app.run(debug=True)