Update app.py

app.py

@@ -1,73 +1,24 @@
-import sqlite3
 import streamlit as st
+import pandas as pd
+import plotly.express as px
 
-# Function to query the database and get the number of movies
-def get_movie_count():
-    # Connect to the SQLite database
-    conn = sqlite3.connect('imdb.db')
-    cursor = conn.cursor()
+# Load your dataframes
+df1= pd.read_csv('/content/series_after_cleaning.csv')
+df2= pd.read_csv('/content/movie_after_cleaning.csv')
 
-    # Query to count the number of movies
-    cursor.execute('SELECT COUNT(*) FROM title_basics WHERE titleType = "movie"')
-    movie_count = cursor.fetchone()[0]
+# Function to generate treemap
+def create_treemap(df, title):
+    fig = px.treemap(df, path=['parentalguide'], title=title)
+    return fig
 
-    # Close the connection
-    conn.close()
+# Streamlit app
+st.title('Parental Guide Treemaps')
 
-    return movie_count
+# Add buttons to switch between dataframes
+option = st.selectbox('Select DataFrame', ['DataFrame 1', 'DataFrame 2'])
 
-# Function to query the database and get the number of unique languages
-def get_language_count():
-    # Connect to the SQLite database
-    conn = sqlite3.connect('imdb.db')
-    cursor = conn.cursor()
-
-    # Query to count the number of unique languages
-    cursor.execute('SELECT COUNT(DISTINCT language) FROM title_basics')
-    language_count = cursor.fetchone()[0]
-
-    # Close the connection
-    conn.close()
-
-    return language_count
-
-# Function to query the database and get the number of unique countries
-def get_country_count():
-    # Connect to the SQLite database
-    conn = sqlite3.connect('imdb.db')
-    cursor = conn.cursor()
-
-    # Query to count the number of unique countries
-    cursor.execute('SELECT COUNT(DISTINCT country) FROM title_basics')
-    country_count = cursor.fetchone()[0]
-
-    # Close the connection
-    conn.close()
-
-    return country_count
-
-# Main Streamlit app
-def main():
-    # Title
-    st.title('IMDb Dataset Statistics')
-
-    # Create a 3-column layout using Streamlit
-    col1, col2, col3 = st.columns(3)
-
-    # Column 1: Number of Movies
-    with col1:
-        st.markdown('<span class="material-icons-outlined" style="font-size: 48px;">movie</span>', unsafe_allow_html=True)
-        st.markdown(f'<div style="text-align: center; font-size: 24px;">Movies: <strong>{get_movie_count()}</strong></div>', unsafe_allow_html=True)
-
-    # Column 2: Number of Languages
-    with col2:
-        st.markdown('<span class="material-icons-outlined" style="font-size: 48px;">language</span>', unsafe_allow_html=True)
-        st.markdown(f'<div style="text-align: center; font-size: 24px;">Languages: <strong>{get_language_count()}</strong></div>', unsafe_allow_html=True)
-
-    # Column 3: Number of Countries
-    with col3:
-        st.markdown('<span class="material-icons-outlined" style="font-size: 48px;">public</span>', unsafe_allow_html=True)
-        st.markdown(f'<div style="text-align: center; font-size: 24px;">Countries: <strong>{get_country_count()}</strong></div>', unsafe_allow_html=True)
-
-if __name__ == '__main__':
-    main()
+# Display the corresponding treemap based on the selected dataframe
+if option == 'DataFrame 1':
+    st.plotly_chart(create_treemap(df1, 'Parental Guide - DataFrame 1'))
+else:
+    st.plotly_chart(create_treemap(df2, 'Parental Guide - DataFrame 2'))