Spaces:
Runtime error
Runtime error
File size: 2,790 Bytes
fa652ec 502394e fa652ec 502394e fa652ec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
import config
import openai
import sys, getopt
from datetime import datetime
import streamlit as st
def get_chatgpt_resp(question):
openai.api_key = st.secrets['OPENAI_API_KEY']
response = openai.ChatCompletion.create(
model='gpt-3.5-turbo',
messages=[
{"role":"system","content":"You are a chatbot"},
{"role":"system","content":question}]
)
result = ''
for choice in response.choices:
result+=choice.message.content
return (result)
def gsearch(pname, num_results):
try:
from googlesearch import search
except ImportError:
print("No module named 'google' found")
# Google Search and return 10 links
query = pname
search_links = []
for j in search(query, tld="com", num=num_results, stop=num_results, pause=2):
search_links.append(j)
return(search_links)
def chatgpt_prompt(pname, search_links):
all_links = '\n'.join(map(str,search_links))
prompt_text = "You are a expert KYC analyst. I need help to identify if there is any adverse news about {}\
in the following links. \n {}. \n. In the reply include a 20 word summary of the text in each link and if you find any adverse\
news (Yes or No)".format(pname, all_links)
return(prompt_text)
def generate_kyc_output(pname, search_links, chat_response):
rep_txt = ''
rep_txt += 'Summary of Google Search for {} \n'.format(pname)
rep_txt += '\n'
rep_txt += "Report generated on {} \n".format(datetime.now())
#rep_txt += "----------------------------------------------------- \n"
rep_txt += '\n'
rep_txt += "Top Google Search Links "
rep_txt += '\n'
rep_txt += '\n'.join(map(str,search_links))
#rep_txt += "\n----------------------------------------------------- \n"
rep_txt += '\n'
rep_txt+= "\n Summary of searches and adverse news findings \n"
#rep_txt += "----------------------------------------------------- \n"
rep_txt += chat_response
return(rep_txt)
def main(argv):
try:
opts, args = getopt.getopt(argv,"i:", ["person="])
except getopt.GetoptError:
print ('Usage: python app.py --person=<person name>')
sys.exit(2)
for opt, arg in opts:
if opt == '--person':
pname = arg
# Google search for the person name and get the first 20 query links
search_links = gsearch(pname)
# 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(pname, search_links, resp)
print(rep_txt)
if __name__ == "__main__":
main(sys.argv[1:]) |