Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import random
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11playlists = {}
12songs = {
13    1: {"artist": "Tycho", "bpm": 90, "genre": "ambient"},
14    2: {"artist": "Bonobo", "bpm": 100, "genre": "downtempo"},
15    3: {"artist": "Nujabes", "bpm": 85, "genre": "lofi hip hop"},
16    4: {"artist": "FKJ", "bpm": 95, "genre": "jazz house"},
17    5: {"artist": "Tom Misch", "bpm": 105, "genre": "neo soul"},
18    6: {"artist": "Moby", "bpm": 88, "genre": "electronic"},
19    7: {"artist": "Massive Attack", "bpm": 75, "genre": "trip hop"},
20    8: {"artist": "Air", "bpm": 80, "genre": "ambient"},
21    9: {"artist": "Boards of Canada", "bpm": 82, "genre": "ambient"},
22    10: {"artist": "Aphex Twin", "bpm": 110, "genre": "experimental"}
23}
24next_song_id = 11
25next_user_id = 1
26 
27def get_user_from_token(authorization: Optional[str] = Header(None)):
28    if not authorization:
29        raise HTTPException(status_code=401, detail="Missing auth header")
30    token = authorization.replace("Bearer ", "")
31    if token not in tokens:
32        raise HTTPException(status_code=401, detail="Invalid token")
33    return tokens[token]
34 
35@app.post("/signup")
36def signup(username: str, password: str):
37    global next_user_id
38    for u in users.values():
39        if u["username"] == username:
40            raise HTTPException(status_code=400, detail="User exists")
41    user_id = next_user_id
42    next_user_id += 1
43    users[user_id] = {"id": user_id, "username": username, "password": hashlib.sha256(password.encode()).hexdigest()}
44    return {"id": user_id, "username": username}
45 
46@app.post("/login")
47def login(username: str, password: str):
48    for u in users.values():
49        if u["username"] == username and u["password"] == hashlib.sha256(password.encode()).hexdigest():
50            token = secrets.token_hex(16)
51            tokens[token] = u["id"]
52            return {"token": token}
53    raise HTTPException(status_code=401, detail="Invalid credentials")
54 
55@app.get("/playlist")
56def get_playlist(mood: str, authorization: Optional[str] = Header(None)):
57    user_id = get_user_from_token(authorization)
58    mood_map = {
59        "chill": ["ambient", "downtempo", "lofi hip hop", "trip hop"],
60        "happy": ["neo soul", "jazz house"],
61        "energetic": ["electronic", "experimental"]
62    }
63    genres = mood_map.get(mood.lower(), ["ambient"])
64    filtered = {k: v for k, v in songs.items() if v["genre"] in genres}
65    return {"playlist": filtered, "user_id": user_id}
66 
67@app.get("/recommend")
68def recommend(song_id: int, authorization: Optional[str] = Header(None)):
69    user_id = get_user_from_token(authorization)
70    if song_id not in songs:
71        raise HTTPException(status_code=404, detail="Song not found")
72    target = songs[song_id]
73    scored = []
74    for sid, s in songs.items():
75        if sid == song_id:
76            continue
77        score = 0
78        if s["genre"] == target["genre"]:
79            score += 5
80        bpm_diff = abs(s["bpm"] - target["bpm"])
81        if bpm_diff <= 10:
82            score += 3
83        elif bpm_diff <= 20:
84            score += 1
85        scored.append((score, sid, s))
86    scored.sort(reverse=True)
87    top3 = [(sid, s) for _, sid, s in scored[:3]]
88    return {"recommendations": {sid: s for sid, s in top3}, "user_id": user_id}
89 
90@app.get("/song/{song_id}")
91def get_song(song_id: int, authorization: Optional[str] = Header(None)):
92    user_id = get_user_from_token(authorization)
93    if song_id not in songs:
94        raise HTTPException(status_code=404, detail="Song not found")
95    return songs[song_id]
96 
97@app.post("/song")
98def create_song(artist: str, bpm: int, genre: str, authorization: Optional[str] = Header(None)):
99    global next_song_id
100    user_id = get_user_from_token(authorization)
101    song_id = next_song_id
102    next_song_id += 1
103    songs[song_id] = {"artist": artist, "bpm": bpm, "genre": genre}
104    return {"id": song_id, "artist": artist, "bpm": bpm, "genre": genre}
105 
106@app.get("/user/{user_id}")
107def get_user(user_id: int, authorization: Optional[str] = Header(None)):
108    current_user = get_user_from_token(authorization)
109    if user_id not in users:
110        raise HTTPException(status_code=404, detail="User not found")
111    return {"id": users[user_id]["id"], "username": users[user_id]["username"]}

1fastapi
2uvicorn