', unsafe_allow_html=True)
st.markdown('
VibeSync
', unsafe_allow_html=True)
st.markdown('
Your personality, translated into sound.
', unsafe_allow_html=True)
# Spotify Connection Status
user = check_spotify_auth()
if user:
st.markdown(f'
✓ Connected as {user["display_name"]}
', unsafe_allow_html=True)
elif SPOTIFY_CLIENT_ID:
st.markdown('
Connect Spotify to export your magic.
', unsafe_allow_html=True)
col1, col2, col3 = st.columns([1, 1, 1])
with col2:
if st.button("🎵 Connect Spotify", key="connect_spotify"):
# Generate auth URL
sp_oauth = get_spotify_oauth()
auth_url = sp_oauth.get_authorize_url()
st.markdown(f"""
""", unsafe_allow_html=True)
st.markdown("""
🧠
AI Analysis
Deep mood mapping
✨
Curated Vibes
20+ perfect tracks
🎧
Instant Export
Direct to Spotify
""", unsafe_allow_html=True)
col1, col2, col3 = st.columns([1, 2, 1])
with col2:
if st.button("🚀 Start Your Journey", key="start_quiz"):
st.session_state.stage = 'quiz'
st.rerun()
st.markdown('
', unsafe_allow_html=True)
st.markdown('
VibeSync Quiz
', unsafe_allow_html=True)
st.markdown(f'
Question {st.session_state.quiz_index + 1} of {len(QUIZ_QUESTIONS)}
', unsafe_allow_html=True)
# Progress bar
progress = (st.session_state.quiz_index) / len(QUIZ_QUESTIONS)
st.progress(progress)
st.markdown('
', unsafe_allow_html=True)
st.markdown(f'
{current_q["question"]}
', unsafe_allow_html=True)
if current_q.get("type") == "multiselect":
answer = st.multiselect("Choose languages:", current_q["options"], key=f"q_{st.session_state.quiz_index}")
else:
answer = st.radio("Choose one:", current_q["options"], key=f"q_{st.session_state.quiz_index}", label_visibility="collapsed")
col1, col2, col3 = st.columns([1, 2, 1])
with col2:
if st.button("Next Question ➡️", key=f"next_{st.session_state.quiz_index}"):
# Save answer
st.session_state.quiz_answers[current_q["key"]] = answer
if st.session_state.quiz_index < len(QUIZ_QUESTIONS) - 1:
st.session_state.quiz_index += 1
st.rerun()
else:
# Quiz completed, generate playlist with AI
st.session_state.playlist = generate_playlist_with_groq(st.session_state.quiz_answers)
st.session_state.stage = 'results'
st.rerun()
st.markdown('
', unsafe_allow_html=True)
st.markdown('
Your Sonic Identity
', unsafe_allow_html=True)
st.markdown(f"
A {len(st.session_state.playlist)}-track journey curated by AI.
", unsafe_allow_html=True)
# Display playlist
for idx, song in enumerate(st.session_state.playlist, 1):
display_song_card(song, idx)
# Add more songs with AI
st.markdown("
", unsafe_allow_html=True)
st.markdown("
Curious for more? 🎵
", unsafe_allow_html=True)
user_request = st.text_input("Describe what kind of songs you'd like to add:", placeholder="e.g., more upbeat songs, slower tempo, different genre...")
if st.button("✨ Get AI Suggestions"):
if user_request:
groq_client = get_groq_client()
if groq_client:
prompt = f"""Based on the user's request and their current playlist, suggest 5 additional songs.
User's request: {user_request}
Current playlist context:
{json.dumps(st.session_state.playlist[:3], indent=2)}
Recommend 5 songs that match their request. Format as JSON array:
[
{{
"title": "Song Title",
"artist": "Artist Name",
"genre": "Genre",
"reasoning": "Why this matches their request (MAX 10 words)"
}}
]"""
try:
with st.spinner("✨ AI is finding more gems..."):
response = groq_client.chat.completions.create(
model="llama-3.3-70b-versatile",
messages=[
{"role": "system", "content": "You are a music expert. Always respond with valid JSON."},
{"role": "user", "content": prompt}
],
temperature=0.8,
max_tokens=1000
)
content = response.choices[0].message.content.strip()
if "```json" in content:
content = content.split("```json")[1].split("```")[0].strip()
elif "```" in content:
content = content.split("```")[1].split("```")[0].strip()
new_songs = json.loads(content)
st.session_state.temp_suggestions = new_songs
except Exception as e:
st.error(f"Error getting AI suggestions: {str(e)}")
else:
st.warning("⚠️ Groq API not configured")
# Display suggestions if they exist
if 'temp_suggestions' in st.session_state:
st.markdown("
AI Suggestions:
", unsafe_allow_html=True)
for idx, song in enumerate(st.session_state.temp_suggestions, 1):
col1, col2 = st.columns([5, 1])
with col1:
st.markdown(f"""
{song['title']}
{song['artist']}
“{song.get('reasoning', 'Matches your request.')}”
""", unsafe_allow_html=True)
with col2:
st.markdown("
", unsafe_allow_html=True)
if st.button("➕", key=f"add_{idx}_{song['title'][:10]}"):
st.session_state.playlist.append(song)
# Add to Spotify if connected
if st.session_state.spotify_playlist_id:
sp = get_spotify_client()
if sp:
try:
query = f"{song['title']} {song['artist']}"
results = sp.search(q=query, type='track', limit=1)
if results['tracks']['items']:
sp.playlist_add_items(
st.session_state.spotify_playlist_id,
[results['tracks']['items'][0]['uri']]
)
st.toast(f"✅ Added {song['title']} to Spotify!")
except:
pass
# Remove from suggestions after adding
st.session_state.temp_suggestions.pop(idx-1)
if not st.session_state.temp_suggestions:
del st.session_state.temp_suggestions
st.rerun()
# Export to Spotify
st.markdown("
", unsafe_allow_html=True)
if st.session_state.spotify_connected and not st.session_state.spotify_playlist_id:
st.markdown("""
🎧 Ready to export to Spotify!
""", unsafe_allow_html=True)
col1, col2, col3 = st.columns([1, 2, 1])
with col2:
playlist_name = st.text_input("Playlist Name:", value="My VibeSync Playlist")
if st.button("📤 Create Spotify Playlist"):
playlist_id = create_spotify_playlist(playlist_name)
if playlist_id:
if search_and_add_to_spotify_playlist(st.session_state.playlist, playlist_id):
st.session_state.spotify_playlist_id = playlist_id
st.success("✅ Playlist created and songs added to your Spotify account!")
st.balloons()
st.rerun()
elif st.session_state.spotify_playlist_id:
st.markdown("""
✅ Playlist exported to Spotify!
Check your Spotify account
""", unsafe_allow_html=True)
# Action buttons
st.markdown("
", unsafe_allow_html=True)
col1, col2 = st.columns(2)
with col1:
if st.button("🔄 Start Over"):
st.session_state.stage = 'landing'
st.session_state.quiz_index = 0
st.session_state.quiz_answers = {}
st.session_state.playlist = []
st.session_state.spotify_playlist_id = None
if 'temp_suggestions' in st.session_state:
del st.session_state.temp_suggestions
st.rerun()
with col2:
if st.button("✅ I'm Happy with My Playlist"):
st.balloons()
st.success(f"🎉 Awesome! Your playlist of {len(st.session_state.playlist)} songs is ready!")
st.markdown('