Spaces:
Runtime error
Runtime error
File size: 1,337 Bytes
90653e1 93b834c 90653e1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
import os
from datetime import datetime
import openai
from flask import Flask, redirect, render_template, request, session, url_for
from models.bloom import initialize_index
app = Flask(__name__)
index = None
@app.route("/")
def home():
if 'access_token' not in session:
return render_template('auth.html')
global index
try:
index = initialize_index("index.json")
except:
return render_template("auth.html", error="Invalid Token Key")
return render_template('chat.html')
@app.route("/login", methods=["GET", "POST"])
def login():
if request.method == "POST":
username = request.form.get("username")
access_token = request.form.get("access_token")
openai.api_key = access_token
session['access_token'] = access_token
return redirect(url_for('home'))
return render_template('auth.html')
@app.route("/logout")
def logout():
session.pop('access_token', None)
return redirect(url_for('home'))
@app.route("/chat")
def chat():
query_text = request.args.get("msg", None)
if query_text is None:
return "Invalid input"
if 'access_token' not in session:
return redirect(url_for('login'))
response = index.query(query_text)
return str(response), 200
if __name__ == "__main__":
app.run(debug=True)
|