Update app.py

app.py

@@ -1,12 +1,65 @@
+
 import streamlit as st
 import pandas as pd
+import altair as alt
+
+# Load the datasets
+child_mortality_path = "child_mortality.csv"
+population_path = "pop.csv"
+
+child_mortality = pd.read_csv(child_mortality_path)
+population = pd.read_csv(population_path)
+
+# Clean population data (handle k, M, B suffixes)
+def convert_population(value):
+    if isinstance(value, str):
+        if 'B' in value:
+            return float(value.replace('B', '')) * 1_000_000_000
+        elif 'M' in value:
+            return float(value.replace('M', '')) * 1_000_000
+        elif 'k' in value:
+            return float(value.replace('k', '')) * 1_000
+        else:
+            return float(value)
+    return value
+
+population.iloc[:, 1:] = population.iloc[:, 1:].applymap(convert_population)
+
+# Streamlit app title
+st.title("Child Mortality vs Population Across Years")
+
+# Sidebar: Year selection
+year = st.sidebar.slider("Select Year", min_value=1800, max_value=2020, value=2020, step=1)
+
+# Filter datasets for the selected year
+mortality_year = child_mortality[['country', str(year)]].rename(columns={str(year): 'child_mortality'})
+population_year = population[['country', str(year)]].rename(columns={str(year): 'population'})
+
+# Merge datasets
+merged_data = pd.merge(mortality_year, population_year, on='country', how='inner').dropna()
+
+# Sidebar: Country dropdown
+countries = ['All'] + sorted(merged_data['country'].unique())
+selected_country = st.sidebar.selectbox("Select Country", countries)
+
+if selected_country != 'All':
+    merged_data = merged_data[merged_data['country'] == selected_country]
+
+# Scatter plot
+scatter_chart = alt.Chart(merged_data).mark_circle(size=60).encode(
+    x=alt.X('population', title='Population'),
+    y=alt.Y('child_mortality', title='Child Mortality Rate'),
+    tooltip=['country', 'population', 'child_mortality']
+).properties(
+    width=700,
+    height=400,
+    title=f"Child Mortality vs Population in {year}"
+)
 
-# Direct path to the uploaded file (ensure the file is in the same directory as app.py or specify the correct relative path)
-file_path = "pop.csv"  # Adjust this if the file is in a subfolder (e.g., "data/pop.csv")
+# Display chart
+st.altair_chart(scatter_chart, use_container_width=True)
 
-# Read the CSV file into a DataFrame
-df = pd.read_csv(file_path)
+# Data preview
+st.subheader("Data Preview")
+st.write(merged_data)
 
-# Display the first 5 rows of the dataset
-st.write("First 5 rows of the dataset:")
-st.write(df.head())