Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import uuid
5import random
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11playlists = {}
12recommendations_pool = [
13    {"id": 1, "title": "Sunset Groove", "genre": "house", "tempo": 120, "mood": "chill", "artist": "DJ Lazy"},
14    {"id": 2, "title": "Night Drive", "genre": "synthwave", "tempo": 100, "mood": "dark", "artist": "Retro Rider"},
15    {"id": 3, "title": "Morning Coffee", "genre": "jazz", "tempo": 80, "mood": "happy", "artist": "Mellow Trio"},
16    {"id": 4, "title": "Workout Pump", "genre": "rock", "tempo": 140, "mood": "energetic", "artist": "Iron Will"},
17    {"id": 5, "title": "Rainy Day", "genre": "classical", "tempo": 60, "mood": "sad", "artist": "Chopin AI"},
18    {"id": 6, "title": "Party Starter", "genre": "pop", "tempo": 128, "mood": "energetic", "artist": "DJ Spark"},
19    {"id": 7, "title": "Deep Focus", "genre": "ambient", "tempo": 70, "mood": "chill", "artist": "Zen Master"},
20    {"id": 8, "title": "Road Trip", "genre": "country", "tempo": 110, "mood": "happy", "artist": "Highway Stars"},
21]
22next_playlist_id = 1
23next_user_id = 1
24 
25class SignupRequest(BaseModel):
26    username: str
27    password: str
28 
29class LoginRequest(BaseModel):
30    username: str
31    password: str
32 
33class PlaylistCreate(BaseModel):
34    name: str
35    songs: Optional[List[int]] = []
36 
37class RecommendationQuery(BaseModel):
38    genre: Optional[str] = None
39    tempo: Optional[int] = None
40    mood: Optional[str] = None
41 
42def get_current_user(authorization: str = Header(...)):
43    if not authorization.startswith("Bearer "):
44        raise HTTPException(status_code=401, detail="Invalid auth header")
45    token = authorization[7:]
46    if token not in tokens:
47        raise HTTPException(status_code=401, detail="Invalid token")
48    return tokens[token]
49 
50@app.post("/signup")
51def signup(req: SignupRequest):
52    global next_user_id
53    for u in users.values():
54        if u["username"] == req.username:
55            raise HTTPException(status_code=400, detail="Username taken")
56    user_id = next_user_id
57    next_user_id += 1
58    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
59    return {"id": user_id, "username": req.username}
60 
61@app.post("/login")
62def login(req: LoginRequest):
63    for u in users.values():
64        if u["username"] == req.username and u["password"] == req.password:
65            token = str(uuid.uuid4())
66            tokens[token] = u["id"]
67            return {"token": token}
68    raise HTTPException(status_code=401, detail="Invalid credentials")
69 
70@app.post("/playlists")
71def create_playlist(req: PlaylistCreate, authorization: str = Header(...)):
72    global next_playlist_id
73    user_id = get_current_user(authorization)
74    playlist_id = next_playlist_id
75    next_playlist_id += 1
76    playlists[playlist_id] = {"id": playlist_id, "name": req.name, "songs": req.songs, "user_id": user_id}
77    return playlists[playlist_id]
78 
79@app.get("/playlists/{playlist_id}")
80def get_playlist(playlist_id: int, authorization: str = Header(...)):
81    user_id = get_current_user(authorization)
82    if playlist_id not in playlists:
83        raise HTTPException(status_code=404, detail="Playlist not found")
84    return playlists[playlist_id]
85 
86@app.get("/recommendations")
87def get_recommendations(genre: Optional[str] = None, tempo: Optional[int] = None, mood: Optional[str] = None, authorization: str = Header(...)):
88    user_id = get_current_user(authorization)
89    results = recommendations_pool
90    if genre:
91        results = [s for s in results if s["genre"] == genre]
92    if tempo:
93        results = [s for s in results if abs(s["tempo"] - tempo) <= 10]
94    if mood:
95        results = [s for s in results if s["mood"] == mood]
96    return results[:5] if results else []

1fastapi
2uvicorn