Initial commit

app.py

@@ -0,0 +1,108 @@
+import gradio as gr
+import json
+from collections import Counter, defaultdict
+import os
+
+def analyze_spotify_data(files):
+    # files: list of file objects
+    # We'll parse each JSON file and aggregate the data
+    
+    all_records = []
+    for f in files:
+        try:
+            data = json.load(open(f))
+            if isinstance(data, list):
+                all_records.extend(data)
+            else:
+                # If the JSON file isn't a list at top-level, skip or handle differently
+                continue
+        except:
+            # If there's an error in loading JSON, skip that file
+            continue
+    
+    # If no valid data found
+    if not all_records:
+        return "No valid data found in the uploaded files."
+
+    # Aggregate listening stats
+    artist_counter = Counter()
+    track_counter = Counter()
+    album_counter = Counter()  # Note: album info not provided in the sample data, will only work if album is in the data
+    
+    # We also want to consider total listening time per artist/track/album
+    artist_time = defaultdict(int)
+    track_time = defaultdict(int)
+    album_time = defaultdict(int)
+
+    # Attempt to detect if albumName is present
+    album_present = all("albumName" in record for record in all_records if isinstance(record, dict))
+
+    for record in all_records:
+        if not isinstance(record, dict):
+            continue
+        artist = record.get("artistName", "Unknown Artist")
+        track = record.get("trackName", "Unknown Track")
+        ms_played = record.get("msPlayed", 0)
+        # Album may not be present; handle gracefully
+        album = record.get("albumName", "Unknown Album") if album_present else None
+
+        artist_counter[artist] += 1
+        track_counter[track] += 1
+        artist_time[artist] += ms_played
+        track_time[track] += ms_played
+        
+        if album_present and album is not None:
+            album_counter[album] += 1
+            album_time[album] += ms_played
+
+    # Determine top artists by number of tracks played (frequency) and also by time
+    top_artists_by_count = artist_counter.most_common(10)
+    top_artists_by_time = sorted(artist_time.items(), key=lambda x: x[1], reverse=True)[:10]
+
+    # Determine top tracks by frequency and by time
+    top_tracks_by_count = track_counter.most_common(10)
+    top_tracks_by_time = sorted(track_time.items(), key=lambda x: x[1], reverse=True)[:10]
+
+    # Determine top albums if available
+    if album_present:
+        top_albums_by_count = album_counter.most_common(10)
+        top_albums_by_time = sorted(album_time.items(), key=lambda x: x[1], reverse=True)[:10]
+    else:
+        top_albums_by_count = [("No album data found", 0)]
+        top_albums_by_time = [("No album data found", 0)]
+
+    # Format the results into a readable output
+    def format_list(title, data_list, time_data=False):
+        result = f"**{title}**\n"
+        if not time_data:
+            for i, (name, count) in enumerate(data_list, 1):
+                result += f"{i}. {name} ({count} plays)\n"
+        else:
+            for i, (name, ms) in enumerate(data_list, 1):
+                hours = ms / (1000*60*60)
+                result += f"{i}. {name} ({hours:.2f} hours)\n"
+        result += "\n"
+        return result
+
+    output = ""
+    output += format_list("Top Artists by Play Count", top_artists_by_count, time_data=False)
+    output += format_list("Top Artists by Listening Time", top_artists_by_time, time_data=True)
+    output += format_list("Top Tracks by Play Count", top_tracks_by_count, time_data=False)
+    output += format_list("Top Tracks by Listening Time", top_tracks_by_time, time_data=True)
+    output += format_list("Top Albums by Play Count", top_albums_by_count, time_data=False)
+    output += format_list("Top Albums by Listening Time", top_albums_by_time, time_data=True)
+
+    return output
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Spotify Listening Data Analyzer")
+    gr.Markdown("Upload your Spotify JSON files (e.g., 'StreamingHistory0.json', 'StreamingHistory1.json', etc.) to get an overview of your top artists, albums, and tracks.")
+
+    file_input = gr.File(file_count="multiple", type="filepath", label="Upload JSON files")
+    analyze_button = gr.Button("Analyze")
+    output_box = gr.Markdown()
+
+    analyze_button.click(fn=analyze_spotify_data, inputs=file_input, outputs=output_box)
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1 @@
+gradio