import streamlit as st 
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import plotly.express as px 

st.set_page_config(layout='wide')
df = pd.read_csv('india.csv')
st.write(df)

# je veux avoir une carte dans laquelle je vais choisir la carte , la ville et la population de chaquec ville

list_of_states = list(df['State'].unique())
list_of_states.insert(0,'Overall India')

selected_state = st.sidebar.selectbox('Select a state', list_of_states)
primary = st.sidebar.selectbox('Select a primary parameter',sorted(df.columns[5:]))
secondary = st.sidebar.selectbox('Select a secondary parameter',sorted(df.columns[5:]))

plot = st.sidebar.button('Plot Graph')

if plot:
    st.text('Size represents primary parameter')
    st.text('Color Represent secondary parameter')
    if selected_state == 'Overall India':
        # Plot for india country
        fig = px.scatter_mapbox(df,lat = 'Latitude', lon = 'Longitude', size = primary, 
                                color = secondary, zoom=4,size_max=35,
                                mapbox_style='carto-position',width=1200,height=700,hover_name='District')
        st.plotly_chart(fig,use_container_width=True)
    else:
        state_df = df[df['State'] == selected_state]
        fig = px.scatter_mapbox(state_df,lat = 'Latitude', lon = 'Longitude', size = primary, 
                                color = secondary, zoom=6,size_max=35,
                                mapbox_style='carto-position',width=1200,height=700,hover_name='District')
        st.plotly_chart(fig,use_container_width=True)