import numpy as np
import math
import matplotlib.pyplot as plt
import seaborn as sns
plt.style.use('seaborn-white')
import pandas as pd
from matplotlib import animation, rc
import torch.nn.functional as F
import torch
import torch.nn as nn
import torch.optim as optim
plt.rcParams.update({'pdf.fonttype': 'truetype'})
import pickle
pc2 = pickle.load(open('price.pkl','rb'))
import streamlit as st
st.title("Price Optimization")

def to_tensor(x):
    return torch.from_numpy(np.array(x).astype(np.float32))
def prediction(price_max,price_step,policy_net):
    price_grid = np.arange(price_step, price_max, price_step)
    sample_state = [0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0., \
                1.,     0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.]
    Q_s = policy_net(to_tensor(sample_state))
    a_opt = Q_s.max(0)[1].detach()
    
    return price_grid[a_opt],Q_s.detach().numpy()
def fun():
    st.header("Optimal Price Action")
    st.subheader(str(a))
    
    return 
st.header("Enter the Specification")
max_value = st.number_input('Enter the Maximum Value of Price',min_value=50,value = 500,step=1)
step = st.number_input('Enter the Price step',min_value = 10,value = 10,step=1)
a,b = prediction(max_value,step,pc2)
if st.button('Predict'):
    fun()
    chart_data = pd.DataFrame(a,b
    
    columns=["a", "b", "c"],x="Price action ($)",y="Q ($)",width=6)

    st.bar_chart(chart_data)