Camel AGI first code commit

server/webserver.py

@@ -0,0 +1,66 @@
+from flask import Flask
+from flask_migrate import Migrate
+from database import *
+import os
+import urllib.parse
+from flask import jsonify,request,session,render_template,redirect,url_for,Blueprint
+from requests_oauthlib import OAuth2Session
+from flask_login import UserMixin, LoginManager, login_required, login_user, current_user, logout_user
+import secrets
+from datetime import datetime, date, timedelta, timezone
+import os
+import json
+import random
+import requests
+from agent_convo import rp
+
+try:
+    import config
+except ModuleNotFoundError:
+    pass
+
+
+app = Flask(__name__)
+app.register_blueprint(rp)
+app.secret_key = 'autogptsamurai@123'
+
+app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///./test.db'
+db.init_app(app)
+db.app = app
+migrate = Migrate(app, db, compare_type=True,
+                  render_as_batch=True)
+
+login_manager = LoginManager(app)
+
+
+@login_manager.user_loader
+def load_user(user_id):
+    return Admin.query.filter_by(id=user_id).first()
+
+os.environ['OAUTHLIB_INSECURE_TRANSPORT'] = '1'
+
+@app.route("/change_model", methods=['POST'])
+def change_model():
+    model = request.values["model"]
+    current_user.gpt_model = model
+    db.session.commit()
+    return "Success"
+
+@app.route("/logout", methods=['GET'])
+def logout():
+    logout_user()
+    return "Success", 200
+    
+@app.route("/store_key", methods=['POST'])
+def store_key():
+    key = request.json['key']
+    getAdmin = Admin.query.filter_by(id=current_user.id).first()
+    if len(key) == 51:
+        getAdmin.openai_key = key
+        db.session.commit()
+        return jsonify(True)
+    else:
+        return jsonify(False), 400      
+
+if __name__ == '__main__':
+    app.run(host="0.0.0.0", debug=True)