import streamlit as st

import meta
import fitz
import requests
from bs4 import BeautifulSoup

from best_seller import get_items
from pharmap_url import get_link_mapped
from summarize import get_summary_app
from utils.results_utils import get_brief, ReadPDFFile, PopulateDict
from utils.utils import load_image_from_local, local_css, pure_comma_separation, remote_css
from examples import CATEGORY_LIST, LINK_LIST, STUDY_LIST, FILE_LIST

with open("bestseller.html", "r") as f:
	html_content = f.read()

# html_content[:500]





def main():
	st.set_page_config(
		page_title="Tracer",
		page_icon="‍👻",
		layout="wide",
		initial_sidebar_state="expanded"
	)
	remote_css("https://fonts.googleapis.com/css2?family=Montserrat:wght@400;600&family=Poppins:wght@600&display=swap")
	local_css("asset/css/style.css")

	col1, col2 = st.columns([5, 5])
	with col2:
		# st.image(load_image_from_local("asset/images/main_logo.png"), width=300)
		# st.markdown("Discover --> Extract --> Sense --> Decide --> Render", unsafe_allow_html=True)
		st.image(load_image_from_local("asset/images/task_theme.png"), width=500)
		with st.expander("Design & Development", expanded=True):
			st.markdown(meta.SIDEBAR_INFO, unsafe_allow_html=True)

		with st.expander("What does Tracer do?", expanded=True):
			st.markdown(meta.STORY, unsafe_allow_html=True)

		st.markdown(meta.CONCEPT_INFO, unsafe_allow_html=True)
		st.image(load_image_from_local("asset/images/side_bar.png"), width=600)

	with col1:
		st.markdown(meta.HEADER_INFO, unsafe_allow_html=True)

		st.markdown(meta.CHEF_INFO, unsafe_allow_html=True)

		# st.write('<style>div.row-widget.stRadio > div{flex-direction:row;justify-content: center;} </style>',
		#          unsafe_allow_html=True)
		#
		# st.write('<style>div.st-bf{flex-direction:column;} div.st-ag{font-weight:bold;padding-left:2px;}</style>',
		#          unsafe_allow_html=True)

		choose = st.radio("Choose Task", ("Get Best Sellers - Amazon",
		                                  "Get Therapy Area Mapped for Given Disease/s",
		                                  "Generate Summary for Clinical Trials",
		                                  "Generate Financial Brief for given 10-Q file"))

		if choose == 'Get Best Sellers - Amazon':

			content = BeautifulSoup(html_content, "html.parser")
			mydivs = content.find_all("a", {"class": "a-link-normal"})
			bs = get_items(mydivs)

			prompts = list(CATEGORY_LIST.keys()) + ["Custom"]
			prompt = st.selectbox(
				'Examples (select from this list)',
				prompts,
				# index=len(prompts) - 1,
				index=0,

			)

			if prompt == "Custom":
				prompt_box = ""
			else:
				prompt_box = CATEGORY_LIST[prompt]

			items = st.text_area(
				'Selected Best Seller Category: ',
				prompt_box,

			)
			# items = pure_comma_separation(items, return_list=False)
			entered_items = st.empty()
			result_button = st.button('Get Best Selling Items!')

			st.markdown(
				"<hr />",
				unsafe_allow_html=True
			)


			if result_button:
				with st.spinner("Getting the best sellers"):
					st.write(items)
					st.write(bs[items])

		if choose == 'Get Therapy Area Mapped for Given Disease/s':

			prompts = list(LINK_LIST.keys()) + ["Custom"]
			prompt = st.selectbox(
				'Examples (select from this list)',
				prompts,
				# index=len(prompts) - 1,
				index=0
			)

			if prompt == "Custom":
				prompt_box = ""
			else:
				prompt_box = LINK_LIST[prompt]

			items = st.text_area(
				'Selected Sample Webpage Link: ',
				prompt_box,
			)
			# items = pure_comma_separation(items, return_list=False)
			entered_items = st.empty()
			result_button = st.button('Get Therapy Areas!')

			st.markdown(
				"<hr />",
				unsafe_allow_html=True
			)

			if result_button:
				get_link_mapped(items)

		if choose == 'Generate Summary for Clinical Trials':

			prompts = list(STUDY_LIST.keys()) + ["Custom"]
			prompt = st.selectbox(
				'Examples (select from this list)',
				prompts,
				# index=len(prompts) - 1,
				index=0
			)

			if prompt == "Custom":
				prompt_box = ""
			else:
				prompt_box = STUDY_LIST[prompt]

			items = st.text_area(
				'Selected Sample Study ID (Clinical Trial.Gov): ',
				prompt_box,
			)
			# items = pure_comma_separation(items, return_list=False)
			# st.write(items)
			entered_items = st.empty()
			result_button = st.button('Get Summary!')

			st.markdown(
				"<hr />",
				unsafe_allow_html=True
			)

			if result_button:
				headline_output, summary_output = get_summary_app(items)

				# headline = f"""
				# <p class="story-box font-body">{headline_output}</p>
				# """
				# summary = f"""
				# <p class="story-box font-body">{summary_output}</p>
				# """
				# st.markdown(summary, unsafe_allow_html=True)

				# st.write(headline_output)
				st.write(summary_output)

		if choose == 'Generate Financial Brief for given 10-Q file':

			prompts = list(FILE_LIST.keys()) + ["Custom"]
			prompt = st.selectbox(
				'Examples (select from this list)',
				prompts,
				# index=len(prompts) - 1,
				index=0
			)
			if prompt == "Custom":
				file = st.file_uploader("Choose a file")
				uploaded_file  = open(file, 'rb')
				doc = fitz.open(uploaded_file)
				result_button = st.button('Get Earnings Brief!')

				st.markdown(
					"<hr />",
					unsafe_allow_html=True
				)
				print("before custom result button")
				if result_button and uploaded_file is not None:
					with st.spinner("Generating Brief..."):
						DictText = {}
						Extracttext = ReadPDFFile(doc, 19, "(Dollars in millions except per share amounts)", "")
						DictText = PopulateDict(Extracttext, DictText)
						Extracttext = ReadPDFFile(doc, 20, "Overview (continued)",
						                          "Results may not sum due to rounding")
						DictText = PopulateDict(Extracttext, DictText)
						Extracttext = ReadPDFFile(doc, 21, "(Dollars in millions)", "Three months ended")
						DictText = PopulateDict(Extracttext, DictText)
						# st.write(DictText)
						st.write(get_brief(DictText))

			else:
				prompt_box = FILE_LIST[prompt]

				items = st.text_area(
					'Selected File Name: ',
					prompt_box,
				)
				# items = pure_comma_separation(items, return_list=False)
				# st.write(items)
				entered_items = st.empty()
				result_button = st.button('Get Earnings Brief!')

				st.markdown(
					"<hr />",
					unsafe_allow_html=True
				)

				if result_button:
					uploaded_file = open("asset/data/TF-Q1'22-10Q.pdf", 'rb')
					doc = fitz.open(uploaded_file)
					with st.spinner("Generating Brief..."):
						DictText = {}
						Extracttext = ReadPDFFile(doc, 19, "(Dollars in millions except per share amounts)", "")
						DictText = PopulateDict(Extracttext, DictText)
						Extracttext = ReadPDFFile(doc, 20, "Overview (continued)",
						                          "Results may not sum due to rounding")
						DictText = PopulateDict(Extracttext, DictText)
						Extracttext = ReadPDFFile(doc, 21, "(Dollars in millions)", "Three months ended")
						DictText = PopulateDict(Extracttext, DictText)
						# st.write(DictText)
						st.write(get_brief(DictText))


if __name__ == '__main__':
	main()