generate add

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, NO_ITEM, food_return, cloth_return, item_identy, item_match
+from gemini_api import model_api, sentiment, category, ord_num, NO_ORDER, NO_ITEM, food_return, cloth_return, item_identy, item_match, generate_add
 from openai_api import model_api as openai_model_api, sentiment as openai_sentiment
 
 cust_qry_resp = {"senti":"", "cat":"", "num":""}
@@ -16,6 +16,8 @@ with gr.Blocks(title="Customer Support Assistant",
   state_ret_pol_cat = gr.State([None, None])
   # Item, Category
   state_match_item = gr.State([None, None])
+  # Item, theme, age group
+  state_gen_add = gr.State([None, None, None])
   
   gr.Markdown("# Customer Support Assistant")
   llm_api = gr.Radio(["gemini-1.0-pro", "gpt-3.5-turbo"], label="Choose LLM", value="gemini-1.0-pro")
@@ -39,10 +41,14 @@ with gr.Blocks(title="Customer Support Assistant",
       cat = model_api(user_input, category)
       num = model_api(user_input, ord_num)
       item = model_api(user_input, item_identy)
+      
+      def btn_ord_det_clk(ord_num):
+        return [ord_num, senti, cat, user_input, item]
+      
       # 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)
+      ord_num_txt = gr.Textbox(lines=1, type="text", label="Order Number", value=num, interactive=True)
       gr.Textbox(lines=1, type="text", label="Order Item", value=item)
       
       # Decision Rules
@@ -50,7 +56,7 @@ with gr.Blocks(title="Customer Support Assistant",
       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)
+          btn_ord_det.click( btn_ord_det_clk, ord_num_txt, state_order_num)
         else:
           with gr.Row():
             gr.Textbox(lines=1, type="text", label="Next Step", value="Ask Order Number")
@@ -92,19 +98,34 @@ with gr.Blocks(title="Customer Support Assistant",
           gr.Textbox(lines=1, type="text", label="Return Policy Used", value=policy)
   
   @gr.render(inputs=state_match_item)
-  def fetch_ret_response(ip_arr):
+  def fetch_match_item(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)
+      def btn_clk(rd_ip):
+        return [ret_item, ret_theme, rd_ip]
+      
       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)
+        gr.Textbox(lines=1, type="text", label="Add Theme", value=ret_theme, interactive=True)
       with gr.Row():
-        age_cat = gr.Radio(["less than 13 years", "14 to 30 years", "30 to 50 years", "more than 50 years"])
+        age_cat = gr.Radio(["less than 13 years", "14 to 30 years", "30 to 50 years", "more than 50 years"],
+                           label="Select target age group", value="30 to 50 years", interactive=True)
         btn_gen_add = gr.Button("Generate Add")
-
+        btn_gen_add.click(btn_clk, age_cat, state_gen_add)
+  
+  @gr.render(inputs=state_gen_add)
+  def gen_add(ip_arr):
+    print("Generate Add")
+    ord_item = ip_arr[0]
+    theme = ip_arr[1]
+    age_grp = ip_arr[2]
+    if ord_item is not None:
+      ret_add = generate_add(ord_item, theme, age_grp)
+      with gr.Row():
+        gr.Textbox(type="text", label="Genarated Add", value=ret_add, interactive=True)
   
 if __name__ == "__main__":
   app.launch(server_name="0.0.0.0")

gemini_api.py

@@ -230,4 +230,23 @@ def item_match(ip_item):
   ret_item = obj['item']
   theme = adds_store.get(ret_item, "5 percent discount to all members. Become a member")
   return ret_item, theme
+
+def generate_add(item, add_theme, age_group):
+  print(str(age_group))
+  chat_session = model.start_chat(
+    history=[
+      {
+          "role": "user",
+          "parts": ["Generate a catchy advertisement. promoting socks, buy 2 get 3 free for age group more than 50 years"]
+      },
+      {
+          "role": "model",
+          "parts": ["Our super warm and comfortable socks are now availble for a discount for the winter season. Just buy 2 pairs and you get 3 free. The socks are suited for senior citizens as they provide extra warmth."]
+      },
+    ]
+  )
+
+  add_prompt = f"Generate a catchy advertisement.Promoting {item}, {add_theme} for age group {age_group}"
+  response = chat_session.send_message(add_prompt)
+  return response.text