support adds

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from datetime import datetime, timedelta
 
-from gemini_api import model_api, sentiment, category, ord_num, NO_ORDER, food_return, cloth_return
+from gemini_api import model_api, sentiment, category, ord_num, NO_ORDER, NO_ITEM, food_return, cloth_return, item_identy, item_match
 
 cust_qry_resp = {"senti":"", "cat":"", "num":""}
 
@@ -10,11 +10,13 @@ with gr.Blocks(title="Customer Support Assistant",
                analytics_enabled=False) as app:
   # States for triggering events
   # Order Num, Sentiment, Category, User Input
-  state_order_num = gr.State([NO_ORDER, None, None, None])
+  state_order_num = gr.State([NO_ORDER, None, None, None, None])
   # Category, User Input
   state_ret_pol_cat = gr.State([None, None])
+  # Item, Category
+  state_match_item = gr.State([None, None])
   
-  gr.Markdown("Customer Support Assistant")
+  gr.Markdown("# Customer Support Assistant")
   # Inputs from user
   with gr.Row():
     cust_qry = gr.Textbox(lines=5, type="text", label="Customer Query")
@@ -31,19 +33,25 @@ with gr.Blocks(title="Customer Support Assistant",
       senti = model_api(user_input, sentiment)
       cat = model_api(user_input, category)
       num = model_api(user_input, ord_num)
+      item = model_api(user_input, item_identy)
       # Output response
       gr.Textbox(lines=1, type="text", label="Customer Sentiment", value=senti)
       gr.Textbox(lines=1, type="text", label="Order Category", value=cat)
       gr.Textbox(lines=1, type="text", label="Order Number", value=num)
+      gr.Textbox(lines=1, type="text", label="Order Item", value=item)
       
       # Decision Rules
-      if num != NO_ORDER:
-        btn_ord_det = gr.Button("Fetch Order Details")
-        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")
+      if senti in ["NEGATIVE", "MIXED"]:
+        if num != NO_ORDER:
+          btn_ord_det = gr.Button("Fetch Order Details")
+          btn_ord_det.click(lambda x: [num, senti, cat, user_input, item], state_order_num, state_order_num)
+        else:
+          with gr.Row():
+            gr.Textbox(lines=1, type="text", label="Next Step", value="Ask Order Number")
+      else:
+        btn_item_match = gr.Button("Fetch Matching Items from Adds Store")
+        btn_item_match.click(lambda x: [item, cat], state_match_item, state_match_item)  
           
   @gr.render(inputs=state_order_num)
   def fetch_order_det(ip_arr):
@@ -52,9 +60,10 @@ with gr.Blocks(title="Customer Support Assistant",
     ord_senti = ip_arr[1]
     ord_cat = ip_arr[2]
     usr_ip = ip_arr[3]
+    ord_item = ip_arr[4]
     if ord_num != NO_ORDER:
       pur_dt = datetime.now() + timedelta(days=-2)
-      ord_det = f"Order Number: {ord_num}\nPurchase Date: {pur_dt}"
+      ord_det = f"Order Number: {ord_num}\nPurchase Date: {pur_dt}\nItem Ordered: {ord_item}"
       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():
@@ -76,6 +85,20 @@ with gr.Blocks(title="Customer Support Assistant",
           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)
+  
+  @gr.render(inputs=state_match_item)
+  def fetch_ret_response(ip_arr):
+    print("Return Matching Items")
+    ord_item = ip_arr[0]
+    ord_cat = ip_arr[1]
+    if ord_item is not None:
+      ret_item, ret_theme = item_match(ord_item)
+      with gr.Row():
+        gr.Textbox(lines=1, type="text", label="Nearest Matching Item from Adds Store", value=ret_item)
+        gr.Textbox(lines=1, type="text", label="Add Theme", value=ret_theme)
+      with gr.Row():
+        age_cat = gr.Radio(["less than 13 years", "14 to 30 years", "30 to 50 years", "more than 50 years"])
+        btn_gen_add = gr.Button("Generate Add")
 
   
 if __name__ == "__main__":

gemini_api.py

@@ -34,7 +34,19 @@ model = genai.GenerativeModel(
   safety_settings=safety_settings,
   generation_config=generation_config,
 )
+
 NO_ORDER = "Order Number not provided."
+NO_ITEM = "No item present."
+
+adds_store = {
+  "pasta": "buy 1 get 1 free on penne",
+  "chocolate": "50 percent off on premium dark chocolate",
+  "biryani": "100 rupees off on all biryani orders",
+  "shoe": "get 1 pair socks free",
+  "skirt": "new exclusive summer collection",
+  "socks": "buy 2 get 3 free",
+  "saree": "10 percent off on silk collection"
+}
 
 def model_api(input, prompt_type):
     return prompt_type(input)
@@ -101,6 +113,33 @@ def ord_num(input):
     else:
       return ret_num
 
+def item_identy(input):
+    chat_session = model.start_chat(
+    history=[
+        {
+        "role": "user",
+        "parts": [
+            "Identify the item being discussed.\nQuery: The shoes I ordered are small. Size is not what was in the catalog. Order number is 11223\nOutput:{\"item\": \"shoe\"}",
+            "Identify the item being discussed.\nQuery: The pizza had strange smell. Order number is 11223\nOutput:{\"item\": \"pizza\"}",
+        ],
+        },
+    ]
+    )
+
+    response = chat_session.send_message(f"Query: {input}\nOutput:")
+    print(response.text)
+    ret_item = None
+    try:
+      obj = json.loads(response.text)
+      ret_item = obj.get('item', NO_ITEM)
+    except ValueError:
+      print("exception parsing item.")
+    
+    if ret_item is None:
+      return NO_ITEM
+    else:
+      return ret_item
+
 def food_return(input):
   policy = """
   For food items, never refund or accept return.
@@ -164,4 +203,31 @@ def cloth_return(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
+
+def item_match(ip_item):
+  food_items = ["pasta", "chocolate", "biryani"]
+  cloth_items = [ "shoe", "skirt", "socks", "saree"]
+
+  ex1 = f'Get nearest matching item. Item list is {food_items}.\nQuery: Find item matching pizza.\nOutput:{{"item": "pasta"}}'
+  ex2 = f'Get nearest matching item. Item list is {cloth_items}.\nQuery: Find item matching shoe.\nOutput:{{"item": "shoe"}}'
+  ex3 = f'Get nearest matching item. Item list is {food_items}.\nQuery: Find item matching cake.\nOutput:{{"item": "chocolate"}}'
+  ex4 = f'Get nearest matching item. Item list is {cloth_items}.\nQuery: Find item matching dress.\nOutput:{{"item": "skirt"}}'
+
+  chat_session = model.start_chat(
+    history=[
+      {
+        "role": "user",
+        "parts": [
+          ex1, ex2, ex3, ex4
+        ],
+      },
+    ]
+  )
+
+  response = chat_session.send_message(f"Query: Find item matching {ip_item}.\nOutput:")
+  print(response.text)
+  obj = json.loads(response.text)
+  ret_item = obj['item']
+  theme = adds_store.get(ret_item, "5 percent discount to all members. Become a member")
+  return ret_item, theme