| import openai | |
| import gradio as gr | |
| import os | |
| open_ai_key = os.environ["open_ai_key"] | |
| openai.api_key = open_ai_key | |
| def openai_chat(prompt): | |
| completion = openai.completions.create( | |
| model="text-davinci-003", | |
| prompt=prompt, | |
| max_tokens=2048, | |
| temperature=0.5, | |
| top_p=1, | |
| frequency_penalty=0, | |
| presence_penalty=0 | |
| ) | |
| message = completion.choices[0].text | |
| return message.strip() | |
| def build_type_1(content_url, Email_Type, industry,job_role,prospect_context): | |
| return f"Write a {Email_Type} email to sell Bluemetra solutions to a prospect in {industry} based on their information extracted from {prospect_context} by analyzing and summarizing this content {content_url} and is relevant for job {job_role}" | |
| def build_type_2(content_url, Email_Type, industry,job_role,prospect_context): | |
| return f"Write a {Email_Type} email to sell Bluemetra solutions to a prospect in {industry} based on their information extracted from {prospect_context} by analyzing and summarizing this content {content_url} and is relevant for job {job_role}.Start with congratulating {job_role} for the new role" | |
| def call_email(Email_Type,industry,content_url,job_role_change, job_role,prospect_context, history=[]): | |
| if job_role_change=="Yes": | |
| input_agg = build_type_2(content_url,Email_Type, industry,job_role,prospect_context) | |
| elif job_role_change=="No": | |
| input_agg = build_type_1(content_url,Email_Type, industry,job_role,prospect_context) | |
| print(input_agg) | |
| output = openai_chat(input_agg) | |
| history=[] | |
| history.append((Email_Type, output)) | |
| return history, history | |
| gr.Interface(fn = call_email, | |
| inputs = [gr.Radio(["First Touch", "Persuasive", "Followup","Event Invitation","New Launch Communication"],value="First Touch"),"text","text",gr.Dropdown(["Yes", "No"],value="No"),"text",'text','state'], | |
| outputs = ["chatbot",'state']).launch(debug = True) | |
| gr.launch(share=True) |