import streamlit as st
import openai 
import sys, getopt
from datetime import datetime
from streamlit.components.v1 import html
import boto3

from main import chatgpt_prompt, get_chatgpt_resp, generate_kyc_output, gsearch, save_to_s3

# Function to perform the search
# This is a placeholder function, replace it with your actual search implementation
def perform_search(pname, keywords, num_results):
    # record current timestamp
    start_time = datetime.now()
    
    # Google search for the person name and get the first 20 query links 
    query = pname + " " + keywords
    search_links = gsearch(query, num_results)

    # Construct the prompt 
    prompt_text = chatgpt_prompt(pname, search_links)
    #get ChatGPT response 
    resp = get_chatgpt_resp(prompt_text)
    # Create PDF with links and summary 
    rep_txt= generate_kyc_output(query, search_links, resp, start_time)
    return (rep_txt)

main_tab, help_tab = st.tabs(["Run the Bot", "Help"])

with main_tab: 
    # Streamlit app
    st.title("Adverse News Detection Assistant")

    # Input fields
    names_txt = st.text_input("Enter party name (or multiple names separated by ,)")
    plc_text = "laundering OR terrorist OR fraud OR corrupt OR criminal OR investigation OR prosecute OR evasion OR bribe OR sanction"
    keywords = st.text_input("Enter other search words:", value=plc_text)

    st.sidebar.markdown("## Controls")
    st.sidebar.markdown("Choose your **search** *parameters*")
    num_results = st.sidebar.slider("Choose the number of search results:", 5, 30, 20, 5)
    st.sidebar.markdown("## Model")
    st.sidebar.markdown("GPT v3.5")
    st.sidebar.markdown("## App")
    st.sidebar.markdown("v0.3")

    # Search button
    if st.button("Search"):
        names = names_txt.split(",")
        # Perform the search and display the results
        if names and keywords:
            search_results = ""
            for name in names:
                #print("trying for name {} \n".format(name))
                search_results += perform_search(name, keywords, num_results)

            html(f"<pre>{search_results}</pre>", height=200, scrolling=True)
            st.download_button('Download Report',search_results)
            try:
                date_time = datetime.now()
                save_to_s3(search_results,date_time )
                print ("Completed processing for {} names: {} at {} \n".format(len(names), names_txt, str(date_time)))
            except: 
                print ("Completed processing with S3 write error for {} names: {} at {} \n".format(len(names),names_txt, str(date_time)))
        else:
            st.error("Please enter your name and search keywords before searching.")

with help_tab:
    st.title("FAQ")
    
    st.markdown("Q. How do I get a count of number of adverse news?")
    st.markdown("A. This functionality isnt implemented yet. A workaround is to manually count the number of links with adverse news")

    st.markdown("Q. How do I summarise all the adverse news?")
    st.markdown("A. This functionality isnt implemented yet. A workaround is to aggregate the summary of all adverse news items manually, and get a sumary from ChatGPT (chat.openai.com")

    st.markdown("Q. Can I search in other lauguages?")
    st.markdown("A. This functionality isnt implemented yet. We are planning to test this feature out with Chinese first")