main.py

from flask import Flask, render_template, request,redirect, send_from_directory,abort, g
import pymysql
import pymysql.cursors
from flask_login import LoginManager, login_required, login_user, current_user, logout_user
from transformers import AutoTokenizer, AutoModelForCausalLM
from transformers import pipeline



login_manager = LoginManager()



app = Flask(__name__)
login_manager.init_app(app)



app.config['SECRET_KEY'] = 'something_random'


@login_manager.user_loader
def user_loader(user_id):
     cursor = get_db().cursor()

     cursor.execute("SELECT * from `Users` WHERE `id` =%s ", (user_id)),
    
     result = cursor.fetchone()

     if result is None:
          return None
     
     return User(result['id'], result['username'], result['banned'])

#User login manager^^^

fincialissues= ["How do i save","is investing in a card a good choice"]


@app.route("/todo")
def todo():
    cursor =  get_db().cursor()
    cursor.execute("SELECT * FROM `User_Questions`")
    results = cursor.fetchall()

    return render_template(
        "todo.html.jinja",
        fincialissues=results,
        my_variable="2023"
    )

#IMPORTANT!!!! This is the AI functionality code.

def ai_stuff(user_question):
    tokenizer = AutoTokenizer.from_pretrained("cerebras/Cerebras-GPT-2.7B")
    model = AutoModelForCausalLM.from_pretrained("cerebras/Cerebras-GPT-2.7B")
  


    pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
    generated_text = pipe(user_question, max_length=200, do_sample=False, no_repeat_ngram_size=2)[0]
    ai_response = generated_text['generated_text']


    return ai_response

@app.route("/add", methods=['POST'])
def add():
    cursor =  get_db().cursor()

    user_question = request.form['new_question']

    ai_response = ai_stuff(user_question)
    


    cursor.execute("INSERT INTO `User_Questions`(`description`, `answers`) VALUES(%s,%s)", (user_question, ai_response))
    

    fincialissues.append(user_question)
    
    return redirect(('/todo'))







  


#Below this comment is my user page functionality code. 



@app.get('/media/<path:path>')
def send_media(path):
    return send_from_directory('media',path)

#Home page
@app.route("/")
def index():

    return render_template(
        "home.html.jinja"
        
    )

# Sign Out snippet code
@app.route('/sign-out')
def sign_out():
     logout_user()

     return redirect('/sign-in')


#Sign In page 
@app.route('/sign-in', methods = ['POST', 'GET'])  
def sign_in():
      if current_user.is_authenticated:
           return redirect('/todo')
      

      if request.method == 'POST':
           cursor = get_db().cursor()


           cursor.execute("SELECT * FROM `Users` WHERE `username` = %s", (request.form['username']))
           result = cursor.fetchone()

           if result is None:
                return render_template("sign.in.html.jinja")
           
           
           if request.form['password'] == result['password']:
                user = User(result['id'], result['username'], result['banned'])

                login_user(user)

                return redirect('/todo')


           return request.form

      elif request.method == 'GET':
        return render_template("sign.in.html.jinja")

#Sign Up apge
@app.route('/sign-up', methods=['POST', 'GET'])
def sign_up():
       
      if current_user.is_authenticated:
        return redirect('/todo')
       
      if request.method == 'POST':
        cursor = get_db().cursor()

        photo = request.files['photo']

        file_name = photo.filename

        file_extension = file_name.split('.')[-1]

        print(file_extension)

        if file_extension in ['jpg', 'jpeg', 'png', 'gif']:
             
             photo.save('media/users/' + file_name)

        else:

            raise Exception('Invalid file type')

        cursor.execute("""
            INSERT INTO `Users` (`username`, `password`, `email`, `date_of_birth`, `phone_number`, `photo`, `display_name`)
            VALUES(%s,%s,%s,%s,%s,%s,%s)
        """, (request.form['username'], request.form['password'],request.form['email'],request.form['brithday'],request.form['phone_number'],file_name ,request.form['display_name']))

        
        return redirect('/todo')
      elif request.method == 'GET':
      
        return render_template("sign.up.html.jinja")
      

#404 page
@app.errorhandler(404)
def not_found_error(error):
    return render_template('404.html.jinja'),404     



class User:
     def __init__(self,id,username, banned):
          self.is_authenticated = True
          self.is_anonymous = False
          self.is_active = not banned

          self.username = username
          self.id = id
     def get_id(self):
        return str(self.id)
          
#Data base
def connect_db():
    return pymysql.connect(
        host="10.100.33.60",
        user="gabidemi",
        password="244655536",
        database="gabidemi_capstone",
        cursorclass=pymysql.cursors.DictCursor,
        autocommit=True
    )

def get_db():
    '''Opens a new database connection per request.'''        
    if not hasattr(g, 'db'):
        g.db = connect_db()
    return g.db    

@app.teardown_appcontext
def close_db(error):
    '''Closes the database connection at the end of request.'''    
    if hasattr(g, 'db'):
        g.db.close() 

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