app.py

import os
import base64
from flask import Flask, redirect, request, jsonify, Response
import requests
from flask_cors import CORS

import firebasefunctions
import helperfunctions
from firebaseconfig import *
from firebasefunctions import *
import random

app = Flask(__name__)
cors = CORS(app)



access_token = ""
refresh_token = ""
seeds = []


@app.route("/")
def get_artist():
    # new_artist = artist()
    return "Hello World"


@app.route("/login")
def get_login():
    client_id = f'client_id={os.environ.get("client_id")}'
    redirect_uri = f'redirect_uri={os.environ.get("redirect_uri")}'
    state = 'state=12355444444'
    scope = 'scope=user-read-private user-read-email user-top-read playlist-modify-public playlist-modify-private'
    response_type = 'response_type=code'
    return redirect(
        f'https://accounts.spotify.com/authorize?{client_id}&{redirect_uri}&{state}&{scope}&{response_type}')


@app.route("/callback")
def get_callback():
    code = request.args.get('code')
    state = request.args.get('state')
    payload = {'grant_type': 'authorization_code', 'code': code, 'redirect_uri': {os.environ.get("redirect_uri")}}
    message = f'{os.environ.get("client_id")}:{os.environ.get("client_secret")}'
    headers = {'Content-Type': 'application/x-www-form-urlencoded',
               'Authorization': "Basic " + base64.b64encode(message.encode("utf-8")).decode("utf-8")}

    r = requests.post('https://accounts.spotify.com/api/token', data=payload, headers=headers)
    print("gabiiii2", r.text)

    rj = r.json()
    print("gabiiii", rj)



    current_access_token = rj["access_token"]
    current_refresh_token = rj["refresh_token"]

    user_id = helperfunctions.get_me(current_access_token)

    print("AT", current_access_token) 
    print("UI", user_id)
    firebasefunctions.set_user(user_id, current_refresh_token)
                               
    return redirect(f'{os.environ.get("uri")}/home?UI={user_id}&AT={current_access_token}')



@app.route("/refresh")
def refresh(): 
    message = f'{os.environ.get("client_id")}:{os.environ.get("client_secret")}'
    headers = {'Content-Type': 'application/x-www-form-urlencoded',
               'Authorization': "Basic " + base64.b64encode(message.encode("utf-8")).decode("utf-8")}
    global access_token
    global refresh_token

    payload = {'grant_type': 'refresh_token', 'refresh_token': refresh_token}
    r = requests.post('https://accounts.spotify.com/api/token', data=payload, headers=headers)
    rj = r.json()
    access_token = rj["access_token"]
    return "Refreshed Token"


@app.route("/me/artists")
def get_me_artists():
    current_access_token = request.headers.get('AT')
    current_user_id = request.headers.get('UI')
    headers = {'Authorization': "Bearer " + current_access_token}
    r = requests.get('https://api.spotify.com/v1/me/top/artists', headers=headers)
    rj = r.json()
    items_arr = []
    artists = []
    genres = []
    items = rj.get('items')
    if items:
        for item in items:
            curr = {'name': item.get('name'), 'img': item.get('images')[1].get('url'), 'genres': item.get('genres'),
                    'id': item.get('id')}
    
            if curr.get('genres') is None or len(curr.get('genres')) == 0:
                curr['genres'] = ['pop']
            if curr.get('id') is None:
                curr['id' ] = '06HL4z0CvFAxyc27GXpf02'
            if curr['name'] != 'Gibi ASMR':
                items_arr.append(curr)
                artists.append(curr.get('id'))
                genres.append(curr.get('genres')[0])
        print(current_user_id, "app ui")
        response = firebasefunctions.update_user_artists(current_user_id, artists[:5], genres[:5])
        if response:
            print("Artists updated successfully")
        else:
            print("Error on update")
        return items_arr
    return Response(
        f'Could not get artists from user {current_user_id}',
        status= 400)


@app.route("/me/tracks")
def get_me_tracks():
    current_access_token = request.headers.get('AT')
    current_user_id = request.headers.get('UI')
    headers = {'Authorization': "Bearer " + current_access_token}
    r = requests.get('https://api.spotify.com/v1/me/top/tracks', headers=headers)
    rj = r.json()
    items_arr = []
    tracks = []
    items = rj.get('items')
    if items:
        for item in items:
            curr = {'artist': item.get('artists')[0].get('name'), 'name': item.get('name'),
                    'img': item.get('album').get('images')[1].get('url'), 'id': item.get('id')}
            # Second Image 300x300
            if curr['artist'] != 'Gibi ASMR':
                items_arr.append(curr)
                tracks.append(curr.get('id'))
        firebasefunctions.update_user_tracks(current_user_id, tracks[:5])
        return items_arr
    return Response(
        f'Could not get tracks from user {current_user_id}',
        status= 400)


@app.route("/me/recommendations")
def get_me_recommendations():
    current_access_token = request.headers.get('AT')
    headers = {'Authorization': "Bearer " + current_access_token}
    params = {'seed_artists': '4NHQUGzhtTLFvgF5SZesLK', 'seed_genres': 'country,pop,electropop',
              'seed_tracks': '0c6xIDDpzE81m2q797ordA'}
    r = requests.get('https://api.spotify.com/v1/recommendations', headers=headers, params=params)
    rj = r.json()
    items_arr = []
    items = rj.get('tracks')
    for item in items:
        # Second Image 300x300
        curr = {'artist': item.get('artists')[0].get('name'), 'name': item.get('name'),
                'img': item.get('album').get('images')[1].get('url'), 'id': item.get('id')}
        items_arr.append(curr)
    return items_arr


@app.route("/groups/create")
def create_group():
    adjectives = ["sweet 🥹", "bitter 😑", "delicious 🤤", "pretty 😚", "beautiful 😘"]
    fruits = ["apple 🍎", "orange 🍊", "pineapple 🍍", "strawberry 🍓", "grape 🍇"]
    group_name = f'{random.choice(adjectives)} {random.choice(fruits)}'
    current_user_id = request.headers.get('UI')
    group_id =  firebasefunctions.create_group(group_name, current_user_id)
    return Response(
        group_id,
        status= 200)


@app.route("/groups/join")
def join_group():
    group_id = request.args.get('id')
    current_user_id = request.headers.get('UI')
    status = firebasefunctions.join_group(group_id, current_user_id)
    if status:
        return Response(
        f'Joined group {group_id} successfully',
        status= 200
        )
    return Response (
            f'Error: Could not join group {group_id}',
            status= 400
            )



@app.route("/groups/recommendations")
def get_recommendations():
    group_id = request.args.get('id')
    group_genre = request.args.get('genre')
    current_access_token = request.headers.get('AT')
    group_tracks, group_artists, _ = firebasefunctions.get_group_seeds(group_id)
    headers = {'Authorization': "Bearer " + current_access_token}
    params = {'seed_artists': f'{random.choice(group_artists)},{random.choice(group_artists)}', 'seed_genres': group_genre,
              'seed_tracks': f'{random.choice(group_tracks)},{random.choice(group_tracks)}' }
    r = requests.get('https://api.spotify.com/v1/recommendations', headers=headers, params=params)
    rj = r.json()
    items_arr = []
    items = rj.get('tracks')
    for item in items:
        curr = {'artist': item.get('artists')[0].get('name'), 'name': item.get('name'),
                'img': item.get('album').get('images')[1].get('url'), 'id': item.get('id')}
        items_arr.append(curr)
    response = firebasefunctions.update_group_playlist(group_id, items_arr)
    if not response:
        print("Playlist not updated")
    else: print("Playlist updated")
    return items_arr
  


@app.route("/groups/group")
def get_group():
    group_id = request.args.get('id')
    current_access_token = request.headers.get('AT')
    response = firebasefunctions.get_group(group_id, current_access_token)
    if response:
        return response
    return Response(
        f'Could not get group {group_id}',
        status= 400)


@app.route("/groups/mine")
def get_my_groups():
    current_user_id = request.headers.get('UI')
    current_access_token  = request.headers.get('AT')

    response = firebasefunctions.get_my_groups(current_user_id, current_access_token)
    if response:
        return response
    return Response(
        f'Could not get groups form user {current_user_id}',
        status= 400)


@app.route("/groups/add/playlist")
def add_playlist():
    current_access_token = request.headers.get('AT')
    current_user_id = request.headers.get('UI')
    current_group_id = request.args.get('groupid')
    current_group_name =  request.args.get('name')

    
    payload = {'name': current_group_name, 'description': 'Playlist generated by Spomix'}
    headers = {'Authorization': "Bearer " + current_access_token}
    r = requests.post(f'https://api.spotify.com/v1/users/{current_user_id}/playlists', headers=headers, json=payload)
    rj = r.json()
    playlist_id = rj.get("id")

    group_playlist = firebasefunctions.get_group_playlist(current_group_id)
    playlist_json = {"uris": []}

    for track in group_playlist:
        playlist_json["uris"].append(f'spotify:track:{track.get("id")}') 

    r2 = requests.post(f'https://api.spotify.com/v1/playlists/{playlist_id}/tracks', headers=headers, json=playlist_json)
    r2j = r2.json()

    return r2j



@app.route("/test")
def test():
    return "HET"


if __name__ == "__main__":
    app = app
    app.run()