main.py

import streamlit as st
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import yfinance as yf
from io import BytesIO


st.title('Учебный проект по визуализации данных')
st.write("Статистическая выдержка предприятия питания за 2023 г.")

tips = pd.read_csv('tips.csv')

st.write(tips.head(10))


# 1
st.subheader('Вариативность размера общего чека')
fig, ax = plt.subplots()
sns.histplot(data=tips, x='total_bill', ax=ax, kde=True, color='#0d88e6')
plt.xlabel('Total Bill, USD')
plt.ylabel('Number of orders')
st.pyplot(fig)

buf = BytesIO()
fig.savefig(buf, format="png")
buf.seek(0)

st.download_button('Скачать график', 
                   data=buf, 
                   file_name='Вариативность размера Общего Чека.png', 
)


# 2
st.subheader('Тренд размера чаевых по дням недели')
st.bar_chart(data=tips, 
             x='day_of_week', 
             y='tip',
             x_label='Day of the week',
             y_label='Size of the tip, USD',
             color='#0d88e6'
)


# 3
st.subheader('Чаевые vs. Общий Чек')
st.scatter_chart(data=tips, 
                 x='total_bill', 
                 y='tip', 
                 x_label='Total Bill, USD', 
                 y_label='Tip, USD',
                 color='#0d88e6'
)


# 4
st.subheader('Отношение Общего Чека и Размера Чаевых к Объему заказа')
g = sns.relplot(data=tips, 
                x='total_bill', 
                y='tip', 
                size='size',
                aspect=1.4,
                color='#0d88e6',
                legend=False
);
plt.xlabel('Total Bill')
plt.ylabel('Tip, USD')
st.pyplot(g)

buf = BytesIO()
fig.savefig(buf, format="png")
buf.seek(0)

st.download_button('Скачать график', 
                   data=buf, 
                   file_name='Отношение Общего Чека и Размера Чаевых к Объему заказа.png', 
)