Update app.py

app.py

@@ -18,7 +18,7 @@ def openai_query(
     
     return openai.Completion.create(
         engine='text-davinci-002',
-        prompt="Write a professional email to my " + recipient + " starting with Hello " + recipient_name + ", about the subject " + context + " and the email should be based on this draft: " + input,
+        prompt="Write a professional email to my " + recipient.lower() + " starting with Hello " + recipient_name + ", about the subject " + context + " and the email should be based on this draft: " + input,
         temperature = random_state,
         max_tokens= len,
         frequency_penalty=0.25,
@@ -26,18 +26,15 @@ def openai_query(
         best_of=1
     ).get("choices")[0]['text'].strip()
 
-
 def query(payload, API_URL):
-    print()
     response = requests.request("POST", API_URL, json=payload)
     return response.json()
 
-
-def pre_query(model_id, context, input, dates, sender, recipient, recipient_name):
+def pre_query(recipient, recipient_name, sender, context, dates, input, model_id):
   API_URL = "https://api-inference.huggingface.co/models/" + model_id
   
   if model_id == "bigscience/T0pp":
-    input_string = "Write a professional email to my " + recipient + " starting with Hello " + recipient_name + ", about the subject " + context + " and the email should be based on this draft: " + input
+    input_string = "Write a professional email to my " + recipient.lower() + " starting with Hello " + recipient_name + ", about the subject " + context + " and the email should be based on this draft: " + input
     data = query(input_string, API_URL)
     if type(data) is dict:
       return data['error']
@@ -45,7 +42,7 @@ def pre_query(model_id, context, input, dates, sender, recipient, recipient_name
       return data[0]['generated_text']
   
   if model_id == "bigscience/bloom":
-    input_string = "Write a professional email to my " + recipient + " starting with Hello " + recipient_name + ", about the subject " + context + " and the email should be based on this draft: " + input + ": Hello " + recipient_name + ",\n\n"
+    input_string = "Write a professional email to my " + recipient.lower() + " starting with Hello " + recipient_name + ", about the subject " + context + " and the email should be based on this draft: " + input + ": Hello " + recipient_name + ",\n\n"
     data = query({
         "inputs":input_string,
         "parameters":{"max_new_tokens":96,
@@ -54,9 +51,9 @@ def pre_query(model_id, context, input, dates, sender, recipient, recipient_name
     if type(data) is dict:
       return data['error']
     else:
-      return "Hello " + recipient_name + ",\n\n" + data[0]['generated_text'].replace(input_string,'')
-  
-  if model_id == "EleutherAI/gpt-neox-20b":
+          return "Hello " + recipient_name + ",\n\n" + data[0]['generated_text'].replace(input_string,'')
+
+  if model_id == "EleutherAI/gpt-neo-2.7B":
     input_string = "Write a professional email to my " + recipient + " starting with Hello " + recipient_name + ", about the subject " + context + " and the email should be based on this draft: " + input
     data = query(input_string, API_URL)
 
@@ -70,20 +67,31 @@ def pre_query(model_id, context, input, dates, sender, recipient, recipient_name
 
   return
 
-title = "Email Assistant"
+demo = gr.Blocks()
 
-interface = gr.Interface(
-    fn = pre_query,
+with demo:
+    gr.Markdown(
+      """
+      # <center> Email Assistant 
+      Please fill out the fields below!
+      """)
+    with gr.Row():
+      with gr.Column():
+        with gr.Group():
+          with gr.Row():
+            sender = gr.Dropdown(["student", "professor", "employee", "employer", "coworker", "applicant", "recruiter"], label="From", placeholder="I am a...")
+            recipient = gr.Dropdown(["student", "professor", "employee", "employer", "coworker", "applicant", "recruiter"], label="Recipient", placeholder="I am sending to my...")
+            recipient_name = gr.Textbox(label="Recipient Name", placeholder = "Their name is...")
     
-    inputs=[gr.Dropdown(["GPT-3", "bigscience/T0pp", "bigscience/bloom", "EleutherAI/gpt-neox-20b"] ,label = "model_id"),
-            gr.Dropdown([ "Requesting a meeting", "Conflict with scheduled meeting time", "Requesting clarification", "Requesting to leave early", "Requesting a leave of absence", "Requesting a letter of recommendation", "Requesting a referral for a job application"], label= "Subject/Context"),
-            gr.Textbox(label="Input", lines=10, placeholder="Enter your Message Here!"),
-            gr.Textbox(label="Relevant Dates", placeholder ="MM/DD/YYYY"), 
-            gr.Dropdown(["student", "employee", "applicant", "recruiter", "boss"], label="Sender"),
-            gr.Dropdown(["professor", "supervisor", "coworker", "recruiter", "boss"], label="Recipient"),
-            gr.Textbox(label="Recipient Name", placeholder = "George")],
-    
-    outputs=[gr.Textbox(lines=10, label = "Result")],
-    
-    title = title,
-).launch(debug=True)
+    subject = gr.Dropdown([ "Requesting a meeting", "Conflict with scheduled meeting time", "Requesting clarification", "Requesting to leave early", "Requesting a leave of absence", "Requesting a letter of recommendation", "Requesting a referral for a job application"], label= "Subject/Context")
+    dates = gr.Textbox(label="Relevant Dates", placeholder ="MM/DD/YYYY")
+    email = gr.Textbox(label="Input", lines=10, placeholder="Enter your Message Here!")
+    model_id = gr.Dropdown(["GPT-3", "bigscience/T0pp", "bigscience/bloom", "EleutherAI/gpt-neo-2.7B"] ,label = "model_id")
+    submit_button = gr.Button("Generate my email!")
+    text_output = gr.Textbox(lines=10, label = "Email", placeholder = "Your generated email!", interactive = True)
+
+    input_list = [sender, recipient, recipient_name, subject, dates, email, model_id]
+
+    submit_button.click(pre_query, inputs = input_list, outputs=text_output)
+
+demo.launch(debug=True)