Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,57 +1,39 @@
-import pickle
-import pandas as pd
 import streamlit as st
+import pandas as pd
+import pickle
+import numpy as np
 
 st.set_page_config(layout="centered", page_title="Predict Music Genre")
 
 # DATA
 data = [
-    {"key": "popularity", "label": "Popularity", "min": 1,
-        "max": 100, "default": 44, "type": "number", "format": "%1f"},
-    {"key": "danceability", "label": "Danceability", "min": 0.00,
-        "max": 1.00, "default": 0.66, "type": "number"},
-    {"key": "mode", "label": "Mode", "default": 0, "type": "bool"},
-    {"key": "valence", "label": "Valence", "min": 0.00,
-        "max": 1.00, "default": 0.48, "type": "number"},
-    {"key": "speechiness", "label": "Speechiness", "min": 0.00,
-        "max": 1.00, "default": 0.79, "type": "number"},
-    {"key": "acousticness", "label": "Acousticness", "min": 0.00,
-        "max": 1.00, "default": 0.34, "type": "number"},
-    {"key": "liveness", "label": "Liveness", "min": 0.00,
-        "max": 1.00, "default": 0.19, "type": "number"},
-    {"key": "instrumentalness", "label": "Instrumentalness",
-        "min": 0.00, "max": 1.00, "default": 0.17, "type": "number"},
-    {"key": "duration_in_min_sg", "label": "Duration in seg",
-        "min": 0, "max": 1000000, "default": 317, "type": "number"},
+ {"key": "popularity", "label": "Popularity", "min": 1, "max": 100, "default": 44, "type": "number", "format": "%1f" },
+ {"key": "danceability", "label": "Danceability", "min": 0.00, "max": 1.00, "default": 0.66, "type": "number" },
+ {"key": "mode", "label": "Mode", "default": 0, "type": "bool" },
+ {"key": "valence", "label": "Valence", "min": 0.00, "max": 1.00, "default": 0.48, "type": "number" },
+ {"key": "speechiness", "label": "Speechiness", "min": 0.00, "max": 1.00, "default": 0.79, "type": "number" },
+ {"key": "acousticness", "label": "Acousticness", "min": 0.00, "max": 1.00, "default": 0.34, "type": "number" },
+ {"key": "liveness", "label": "Liveness", "min": 0.00, "max": 1.00, "default": 0.19, "type": "number" },
+ {"key": "instrumentalness", "label": "Instrumentalness", "min": 0.00, "max": 1.00, "default": 0.17, "type": "number" },
+ {"key": "duration_in_min_sg", "label": "Duration in seg", "min": 0, "max": 1000000, "default": 317, "type": "number" },
 ]
 
 response = [
-    {"label": "Acoustic/Folk", "descripcion": "Acoustic/Folk music emphasizes unplugged instruments like guitars and violins. It often features narrative storytelling and heartfelt lyrics, drawing on traditional sounds and cultural roots. This genre evokes a sense of intimacy and authenticity, focusing on raw musical expression.",
-        "image": "https://i.pinimg.com/736x/03/80/eb/0380eb05f1ce83ea0f429413f8daebef.jpg"},
-    {"label": "Alternative Music", "descripcion": "Alternative music emerged from the indie underground, rejecting mainstream rock and pop conventions. It encompasses diverse styles, often characterized by unconventional song structures, experimental sounds, and emotionally charged lyrics. Alt_Music prioritizes artistic expression over commercial appeal.",
-     "image": "https://i.pinimg.com/736x/87/68/3c/87683cd8c9dafe80b6bd927167efcbaa.jpg"},
-    {"label": "Blues", "descripcion": "Blues is a foundational genre born from African-American spirituals, work songs, and chants. It's known for its melancholic themes, expressive vocal delivery, and distinctive 12-bar chord progressions. Blues music powerfully conveys feelings of hardship, resilience, and hope.",
-     "image": "https://i.pinimg.com/736x/69/89/03/6989032f191d50fe98f76a3e7503e79a.jpg"},
-    {"label": "Bollywood", "descripcion": "Bollywood music is the vibrant soundtrack of Indian cinema, a fusion of traditional Indian melodies with Western pop, folk, and classical influences. It's highly energetic, often featuring elaborate orchestral arrangements, rich vocals, and rhythmic dance numbers. Bollywood music is integral to the storytelling and emotional core of the films.",
-     "image": "https://i.pinimg.com/736x/6d/d2/f1/6dd2f14df0e77a432df1cd9703a7d3f7.jpg"},
-    {"label": "Country", "descripcion": "Country music originates from the American South and Southwest, blending folk, blues, and gospel traditions. It typically features themes of rural life, love, loss, and patriotism, often driven by instruments like guitars, banjos, and fiddles. Country music tells relatable stories with a down-to-earth sensibility.",
-     "image": "https://i.pinimg.com/736x/58/a3/ad/58a3ad5a34925b1735d29c617e4e55d6.jpg"},
-    {"label": "HipHop",
-     "descripcion": "HipHop is a cultural movement originating in the Bronx, New York, known for its rhythmic spoken word (rapping) over sampled beats. It incorporates elements of DJing, breakdancing, and graffiti art, serving as a powerful medium for social commentary and storytelling. HipHop is diverse, influential, and constantly evolving.", "image": "https://i.pinimg.com/736x/0f/5e/6d/0f5e6d5466ac7e0b7a99a9aa4df88d89.jpg"},
-    {"label": "Indie Alternative", "descripcion": "Indie Alternative music emphasizes independent production and distribution, fostering a unique sound distinct from major labels. It often combines raw, DIY aesthetics with experimental sounds and introspective lyrics. This genre celebrates artistic freedom and a non-conformist approach to music creation.",
-     "image": "https://i.pinimg.com/736x/8c/10/6d/8c106dfea2b51acfad3eaa6d12b886d0.jpg"},
-    {"label": "Instrumental", "descripcion": "Instrumental music focuses solely on musical instruments without the presence of vocals. It can span any genre, allowing for pure melodic and harmonic exploration. This genre allows listeners to interpret the music based on its sonic qualities and emotional resonance, often used for background or cinematic purposes.",
-     "image": "https://i.pinimg.com/736x/ce/36/d7/ce36d70c2b71b931bceb7b744f871944.jpg"},
-    {"label": "Metal", "descripcion": "Metal music is characterized by its aggressive, powerful sound, typically featuring distorted guitars, heavy drumming, and often intense vocals. Subgenres vary widely, but common themes include fantasy, mythology, and social issues. Metal is known for its strong subculture and energetic live performances.",
-     "image": "https://i.pinimg.com/736x/bd/0c/dc/bd0cdc59569c0bd45876f80ad673ad70.jpg"},
-    {"label": "Pop", "descripcion": "Pop music is a broad genre characterized by catchy melodies, simple song structures, and widespread appeal. It's designed for mass consumption, often incorporating elements from various genres and focusing on relatable themes like love and everyday life. Pop music is highly commercial and constantly evolving with trends.",
-     "image": "https://i.pinimg.com/736x/f1/63/5e/f1635e44415b7eb86876ab0d181049b8.jpg"},
-    {"label": "Rock", "descripcion": "Rock music emerged from blues and country, known for its prominent electric guitar, bass, drums, and often strong vocals. It's a diverse genre encompassing many subgenres, often exploring themes of rebellion, love, and social change. Rock music is celebrated for its energy and cultural impact.",
-     "image": "https://i.pinimg.com/736x/37/e4/e1/37e4e1add7dbe95fb90b962d015edd26.jpg"},
+  {"label": "Acoustic/Folk", "descripcion": "Acoustic/Folk music emphasizes unplugged instruments like guitars and violins. It often features narrative storytelling and heartfelt lyrics, drawing on traditional sounds and cultural roots. This genre evokes a sense of intimacy and authenticity, focusing on raw musical expression.", "image": "https://i.pinimg.com/736x/03/80/eb/0380eb05f1ce83ea0f429413f8daebef.jpg"},
+  {"label": "Alternative Music", "descripcion": "Alternative music emerged from the indie underground, rejecting mainstream rock and pop conventions. It encompasses diverse styles, often characterized by unconventional song structures, experimental sounds, and emotionally charged lyrics. Alt_Music prioritizes artistic expression over commercial appeal.", "image": "https://i.pinimg.com/736x/87/68/3c/87683cd8c9dafe80b6bd927167efcbaa.jpg"},
+  {"label": "Blues", "descripcion": "Blues is a foundational genre born from African-American spirituals, work songs, and chants. It's known for its melancholic themes, expressive vocal delivery, and distinctive 12-bar chord progressions. Blues music powerfully conveys feelings of hardship, resilience, and hope.", "image": "https://i.pinimg.com/736x/69/89/03/6989032f191d50fe98f76a3e7503e79a.jpg"},
+  {"label": "Bollywood", "descripcion": "Bollywood music is the vibrant soundtrack of Indian cinema, a fusion of traditional Indian melodies with Western pop, folk, and classical influences. It's highly energetic, often featuring elaborate orchestral arrangements, rich vocals, and rhythmic dance numbers. Bollywood music is integral to the storytelling and emotional core of the films.", "image": "https://i.pinimg.com/736x/6d/d2/f1/6dd2f14df0e77a432df1cd9703a7d3f7.jpg"},
+  {"label": "Country", "descripcion": "Country music originates from the American South and Southwest, blending folk, blues, and gospel traditions. It typically features themes of rural life, love, loss, and patriotism, often driven by instruments like guitars, banjos, and fiddles. Country music tells relatable stories with a down-to-earth sensibility.", "image": "https://i.pinimg.com/736x/58/a3/ad/58a3ad5a34925b1735d29c617e4e55d6.jpg"},
+  {"label": "HipHop", "descripcion": "HipHop is a cultural movement originating in the Bronx, New York, known for its rhythmic spoken word (rapping) over sampled beats. It incorporates elements of DJing, breakdancing, and graffiti art, serving as a powerful medium for social commentary and storytelling. HipHop is diverse, influential, and constantly evolving.", "image": "https://i.pinimg.com/736x/0f/5e/6d/0f5e6d5466ac7e0b7a99a9aa4df88d89.jpg"},
+  {"label": "Indie Alternative", "descripcion": "Indie Alternative music emphasizes independent production and distribution, fostering a unique sound distinct from major labels. It often combines raw, DIY aesthetics with experimental sounds and introspective lyrics. This genre celebrates artistic freedom and a non-conformist approach to music creation.", "image": "https://i.pinimg.com/736x/8c/10/6d/8c106dfea2b51acfad3eaa6d12b886d0.jpg"},
+  {"label": "Instrumental", "descripcion": "Instrumental music focuses solely on musical instruments without the presence of vocals. It can span any genre, allowing for pure melodic and harmonic exploration. This genre allows listeners to interpret the music based on its sonic qualities and emotional resonance, often used for background or cinematic purposes.", "image": "https://i.pinimg.com/736x/ce/36/d7/ce36d70c2b71b931bceb7b744f871944.jpg"},
+  {"label": "Metal", "descripcion": "Metal music is characterized by its aggressive, powerful sound, typically featuring distorted guitars, heavy drumming, and often intense vocals. Subgenres vary widely, but common themes include fantasy, mythology, and social issues. Metal is known for its strong subculture and energetic live performances.", "image": "https://i.pinimg.com/736x/bd/0c/dc/bd0cdc59569c0bd45876f80ad673ad70.jpg"},
+  {"label": "Pop", "descripcion": "Pop music is a broad genre characterized by catchy melodies, simple song structures, and widespread appeal. It's designed for mass consumption, often incorporating elements from various genres and focusing on relatable themes like love and everyday life. Pop music is highly commercial and constantly evolving with trends.", "image": "https://i.pinimg.com/736x/f1/63/5e/f1635e44415b7eb86876ab0d181049b8.jpg"},
+  {"label": "Rock", "descripcion": "Rock music emerged from blues and country, known for its prominent electric guitar, bass, drums, and often strong vocals. It's a diverse genre encompassing many subgenres, often exploring themes of rebellion, love, and social change. Rock music is celebrated for its energy and cultural impact.", "image": "https://i.pinimg.com/736x/37/e4/e1/37e4e1add7dbe95fb90b962d015edd26.jpg"},
 ]
 
-# LOGIC
-@st.cache_resource
+## LOGIC
+#@st.cache_resource
 def load_model(path="./src/xgb_classifier_pipeline.pkl"):
     try:
         with open(path, "rb") as f:
@@ -72,7 +54,7 @@ prediction = -1
 datos_bool = []
 
 
-# VIEW
+### VIEW
 
 # Custom CSS for styling
 st.markdown("""<style>
@@ -226,11 +208,9 @@ with col1:
 
     for data_point in data:
       if data_point["type"] == "number":
-        input_data_original[data_point["key"]] = st.number_input(
-            label=data_point["label"], min_value=data_point["min"], max_value=data_point["max"], value=data_point["default"])
+        input_data_original[data_point["key"]] = st.number_input(label=data_point["label"], min_value=data_point["min"], max_value=data_point["max"], value=data_point["default"])
       elif data_point["type"] == "bool":
-        input_data_original[data_point["key"]] = st.checkbox(
-            label=data_point["label"], value=data_point["default"])
+        input_data_original[data_point["key"]] = st.checkbox(label=data_point["label"], value=data_point["default"])
         datos_bool.append(data_point["key"])
 
     if st.button("Predict Genre"):
@@ -239,15 +219,13 @@ with col1:
           input_data_original[key] = int(input_data_original[key])
 
         input_df_model = pd.DataFrame([input_data_original])
-        input_df_model = pd.DataFrame(
-            input_df_model, columns=input_df_model.columns)
+        input_df_model = pd.DataFrame(input_df_model, columns=input_df_model.columns)
         prediction = int(model.predict(input_df_model.head(1)))
       except Exception as e:
         st.error(f"An error occurred during prediction: {e}")
 
 with col2:
-  st.markdown('<h2 class="predicted-genre-title">Predicted Genre</h2>',
-              unsafe_allow_html=True)
+  st.markdown('<h2 class="predicted-genre-title">Predicted Genre</h2>', unsafe_allow_html=True)
 
   if prediction != -1:
     st.markdown(f"""