Add returns workflow

app.py

@@ -1,14 +1,19 @@
 import gradio as gr
 from datetime import datetime, timedelta
 
-from gemini_api import model_api, sentiment, category, ord_num, NO_ORDER
+from gemini_api import model_api, sentiment, category, ord_num, NO_ORDER, food_return, cloth_return
 
 cust_qry_resp = {"senti":"", "cat":"", "num":""}
 
 #********* UI Code ***********#
 with gr.Blocks(title="Customer Support Assistant",
                analytics_enabled=False) as app:
-  state_order_num = gr.State(NO_ORDER)
+  # States for triggering events
+  # Order Num, Sentiment, Category, User Input
+  state_order_num = gr.State([NO_ORDER, None, None, None])
+  # Category, User Input
+  state_ret_pol_cat = gr.State([None, None])
+  
   gr.Markdown("Customer Support Assistant")
   # Inputs from user
   with gr.Row():
@@ -34,19 +39,43 @@ with gr.Blocks(title="Customer Support Assistant",
       # Decision Rules
       if num != NO_ORDER:
         btn_ord_det = gr.Button("Fetch Order Details")
-        btn_ord_det.click(lambda x: num, state_order_num, state_order_num)
-        # Order Details
-      if senti == "NEGATIVE" and num == NO_ORDER:
+        btn_ord_det.click(lambda x: [num, senti, cat, user_input], state_order_num, state_order_num)
+      # Order Details
+      if senti in ["NEGATIVE", "MIXED"] and num == NO_ORDER:
         with gr.Row():
           gr.Textbox(lines=1, type="text", label="Next Step", value="Ask Order Number")
           
   @gr.render(inputs=state_order_num)
-  def fetch_order_det(ord_num):
+  def fetch_order_det(ip_arr):
     print("Get order Details")
+    ord_num = ip_arr[0]
+    ord_senti = ip_arr[1]
+    ord_cat = ip_arr[2]
+    usr_ip = ip_arr[3]
     if ord_num != NO_ORDER:
       pur_dt = datetime.now() + timedelta(days=-2)
       ord_det = f"Order Number: {ord_num}\nPurchase Date: {pur_dt}"
       gr.Textbox(lines=1, type="text", label="Order Details", value=ord_det)
+      if ord_senti in ["NEGATIVE", "MIXED"] and ord_num != NO_ORDER:
+        with gr.Row():
+          btn_ret_pol = gr.Button("Fetch Return / Replacement Response")
+          btn_ret_pol.click(lambda x: [ord_cat, usr_ip], state_ret_pol_cat, state_ret_pol_cat)
+  
+  @gr.render(inputs=state_ret_pol_cat)
+  def fetch_ret_response(ip_arr):
+    print("Return Response")
+    ord_cat = ip_arr[0]
+    user_input = ip_arr[1]
+    if ord_cat is not None:
+      if ord_cat == "Food":
+        resp, policy = food_return(user_input)
+      if ord_cat == "Clothing":
+        resp, policy = cloth_return(user_input)        
+      with gr.Row():
+        with gr.Column(scale=1):
+          gr.Textbox(lines=1, type="text", label="Return Response", value=resp)
+        with gr.Column(scale=1):
+          gr.Textbox(lines=1, type="text", label="Return Policy Used", value=policy)
 
   
 if __name__ == "__main__":

gemini_api.py

@@ -2,7 +2,6 @@ import google.generativeai as genai
 import json
 import os
 
-# genai.configure(api_key="")
 genai.configure(api_key=os.getenv("API_KEY"))
 generation_config = {
   "temperature": 0.9,
@@ -89,12 +88,80 @@ def ord_num(input):
     )
 
     response = chat_session.send_message(f"Query: {input}\nOutput:")
-
     print(response.text)
-    obj = json.loads(response.text)
-    ret_num = obj.get('order number', NO_ORDER)
+    ret_num = None
+    try:
+      obj = json.loads(response.text)
+      ret_num = obj.get('order number', NO_ORDER)
+    except ValueError:
+      print("exception parsing order number.")
+    
     if ret_num is None:
       return NO_ORDER
     else:
       return ret_num
+
+def food_return(input):
+  policy = """
+  For food items, never refund or accept return.
+  Make a reason where user should be blamed.
+  Example 1: If item needs to be refrigerated, blame user that item was kept at room temperature.
+  Example 2: If item is perishable with short self life. blame user, that item was used after expiration.
+  """
+  chat_session = model.start_chat(
+    history=[
+      {
+        "role": "user",
+        "parts": [policy,],
+      },
+      {
+          "role": "model",
+          "parts": ["understood"]
+      },
+      {
+          "role": "user",
+          "parts": ["The pizza I bought from store had fould smell. I want money back."]
+      },
+      {
+          "role": "model",
+          "parts": ["Our stores have highest storage standard. You must have not frozen the pizza after purchase or used after expiration."]
+      },
+    ]
+  )
+
+  response = chat_session.send_message(input)
+  print(response.text)
+  return response.text, policy
+
+def cloth_return(input):
+  policy = """
+  For clothing items, we almost always ask customer to return.
+  We do not accept return if original state of item is not intact.
+  Make the customer feel welcome to exchange item.
+  """
+  chat_session = model.start_chat(
+    history=[
+      {
+        "role": "user",
+        "parts": [policy,],
+      },
+      {
+          "role": "model",
+          "parts": ["understood"]
+      },
+      {
+          "role": "user",
+          "parts": ["The shoes I ordered are too tight.Please help."]
+      },
+      {
+          "role": "model",
+          "parts": ["We understand ordering online can go wrong. Please book a return or exchange and we will process it."]
+      },
+    ]
+  )
+
+  response = chat_session.send_message(input)
+  print(response.text)
+  full_resp = f"{response.text}\n**For return to be processed, please ensure all tags are intact. Delivery agent will inspect the item and then only process.**"
+  return full_resp, policy