app.py

from flask import Flask, jsonify, redirect, url_for, request
from src.setup import setup
from src.account import create_user, login_user, request_verify_user, verify_user, send_to_mobile_func
from src.tldr import check_version, tldr
from dotenv import load_dotenv
load_dotenv()
import os, cohere
import src.nlp as nlp
import src.scraper as scraper
import json

os.system("curl --create-dirs -o $HOME/.postgresql/root.crt -O https://cockroachlabs.cloud/clusters/02d0901b-5fb2-4a67-b68c-654b2c8c7731/cert")

app = Flask(__name__)
api_key = os.getenv('COHERE_KEY')
co = cohere.Client(api_key)

@app.route('/')
def api(): 
    return {"up": True}

@app.route('/signup', methods = ['POST'])
def signup():
    if request.method == 'POST':
        content = request.json
        user_email = content['user_email']
        user_password = content['user_password']
        user_phone = content['user_phone']
        
    resp = create_user(email=user_email, password=user_password, phone_number=user_phone)
    return resp

@app.route('/login', methods = ['POST'])
def login():
    if request.method == 'POST':
        content = request.json
        user_email = content['user_email']
        user_password = content['user_password']

    resp = login_user(email=user_email, password=user_password)
    return resp

@app.route('/request_verify', methods = ['POST'])
def request_verify():
    if request.method == 'POST':
        content = request.json
        user_email = content['user_email']
    
    resp = request_verify_user(email=user_email)
    return resp

@app.route('/verify', methods = ['POST'])
def verify():
    if request.method == 'POST':
        content = request.json
        user_email = content['user_email']
        user_token = content['user_token']
    
    resp = verify_user(email=user_email, token=user_token)
    return resp

@app.route('/send_to_mobile', methods = ['POST'])
def send_to_mobile():
    if request.method == 'POST':
        content = request.json
        user_email = content['user_email']
        user_content = content['tldr']
    resp = send_to_mobile_func(user_email, user_content)
    return resp

@app.route('/analyze', methods = ['POST'])
def analyze():
    version_num = -1
    if request.method == 'POST':
        print(request)
        content = request.json
        user_id = content['user_id']
        if 'version' in content:
            version_num = content['version']
        if 'input_data' in content:
            input_data = content['input_data']
        else:
            return {"response": "No input"}

    # check for existing version
    if (check_version(user_id, input_data) == 1):
        if (version_num == 1 or version_num == -1): # user requesting new version
            # fetch original text 
            print("new")    
            return
        else: # user requesting old version
            # fetch text
            print("old")
            return 
            
        
    # transform into text
    urlUsed = scraper.validateUrl(input_data)
    if (urlUsed):
        url = input_data
        input_data = scraper.scrape(input_data)
    
    # generate TLDR
    tldr_text = nlp.generateSummery(co, input_data)
    # print(tldr_text)
    # generate sentiment (pos, neg, neut)
    sentiment_res = nlp.generateSentiment(co, input_data)
    sentiment = sentiment_res[0]
    negVal = sentiment_res[1]['negative']
    posVal = sentiment_res[1]['positive']
    neuVal = sentiment_res[1]['neutral']

    if (urlUsed):
        resp = tldr(user_id, "", tldr_text.replace("'", "\""), sentiment, negVal, posVal, neuVal, version_num, url)
    else:
        resp = tldr(user_id, input_data.replace("'", "\""), tldr_text.replace("'", "\""), sentiment, negVal, posVal, neuVal, version_num, "")
    
    return {"response": resp, "tldr": tldr_text, "sentiment_obj": sentiment_res[1]}


# @app.route('/revise', methods = ['POST'])
# def revise():
#     if request.method == 'POST':
#         print(request)
#         content = request.json
#         user_id = content['user_id']
#         if 'input_data' in content:
#             user_data = content['input_data']
#         else:
#             return {"response": "No input"}
#         original_text = content['original_text']

#     # check for existing version
#     if (check_version(user_id, original_text) == 1):
        
            
        
  

if __name__ == '__main__':
    # setup()
    app.run(port=8000, debug=True)