import json
import requests
import streamlit as st
from decouple import config
import random

# Create random user_id
if 'user_id' not in st.session_state:
    user_id = 'st' + str(random.randint(100000, 999999))
    st.session_state['user_id'] = user_id

# Read URL from .env file in the directory
url = config('RASA_SERVER_URL')

# Page header texts
st.set_page_config(page_title="NaBot")

st.markdown("<h1 style='text-align: center; color:grey;'>\
             NaBot</h1>", unsafe_allow_html=True)

st.markdown("<h3 style='text-align: center; color: grey;'>\
             Ask your medical questions 👩‍⚕️ </h3>", unsafe_allow_html=True)
st.write('')

# Show GitHub page badge
"""
[![Star](https://img.shields.io/github/stars/MedBot-team/NaBot.svg?logo=github&style=social)](https://gitHub.com/MedBot-team/NaBot)
"""

st.markdown('___')

# Page sidebar 
# sidebar header
st.sidebar.header('About NaBot')
st.sidebar.write('')
# sidebar text
st.sidebar.markdown("<h4 style='text-align: left; color:grey;'>\
                    NaBot is a Rasa-based, smart medical chatbot.\
                    It can retrieve medicines and lab tests information\
                    for users just by chatting with it.</h4>", unsafe_allow_html=True)


st.sidebar.write('')
st.sidebar.write('')
st.sidebar.write('')

# add expander to the sidebar for website detail
site_expander = st.sidebar.expander('Website')
site_expander.markdown(
    "[NaBot instruction](https://medbot-team.github.io/NaBot)")

# add expander to the sidebar for some conversation examples
exm_expander = st.sidebar.expander('Conversation Example')
exm_expander.write("What is the globulin test?")
exm_expander.write("What are the Gemfibrozil side-effects?")


# Questions
st.markdown("<h2 style='text-align: left; color:#F63366;'><b>Ask your question<b></h2>",
            unsafe_allow_html=True)
st.write('')

question = st.text_area("We support medicine and lab test related questions. For more information visit our website",
                        max_chars=500, value="Can you give me dosage information of Acetaminophen?")


# Ask question button
ask_button = st.button('Give me the answer')

# Function to use when feedback button is clicked
def send_feedback(feedback):
    payload = json.dumps({
        "message": f"{feedback}",
        "sender": f"{st.session_state['user_id']}"
        })
    requests.request("POST", url, headers=headers, data=payload)
    st.success('Feedback submitted successfully')

# When ask question button is clicked    
if ask_button:
    if question == '':
        st.error('Please ask your question')
    else:
        with st.spinner('Wait for it...'):
            # Pass the question to the chatbot server
            result = question.title()

            payload = json.dumps({
                "message": f"{question}",
                "sender": f"{st.session_state['user_id']}"
            })

            headers = {
                'Content-Type': 'application/json'
            }

            response = requests.request(
                "POST", url, headers=headers, data=payload)
            
            # Display feedback buttons
            if len(response.json()) != 0:
                if 'buttons' in response.json()[0].keys():
                    for button in response.json()[0]['buttons']:
                        b = st.button(
                            label=button['title'],
                            on_click=send_feedback,
                            args = [button['payload']],)
                                
                # Give the chatbot answer to the user
                st.markdown(response.json()[0]['text'].replace('\n', '<br>'),
                            unsafe_allow_html=True)
            else:
                st.markdown('Unfortunately, I don\'t know the answer to this question yet')
            
st.markdown('___')