fetch order details

app.py

@@ -1,15 +1,14 @@
 import gradio as gr
+from datetime import datetime, timedelta
 
-from gemini_api import model_api, sentiment, category, ord_num
+from gemini_api import model_api, sentiment, category, ord_num, NO_ORDER
 
 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)
   gr.Markdown("Customer Support Assistant")
   # Inputs from user
   with gr.Row():
@@ -28,17 +27,27 @@ with gr.Blocks(title="Customer Support Assistant",
       cat = model_api(user_input, category)
       num = model_api(user_input, ord_num)
       # Output response
-      with gr.Row():
-        gr.Textbox(lines=1, type="text", label="Customer Sentiment", value=senti)
-      with gr.Row():
-        gr.Textbox(lines=1, type="text", label="Order Category", value=cat)
-      with gr.Row():
-        gr.Textbox(lines=1, type="text", label="Order Number", value=num)
-      if num != "Order Number not provided.":
+      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)
+      
+      # Decision Rules
+      if num != NO_ORDER:
         btn_ord_det = gr.Button("Fetch Order Details")
-      if senti == "NEGATIVE" and num == "Order Number not provided.":
-        gr.Textbox(lines=1, type="text", label="Next Step", value="Ask Order Number")
-  
+        btn_ord_det.click(lambda x: num, state_order_num, state_order_num)
+        # Order Details
+      if senti == "NEGATIVE" 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):
+    print("Get order Details")
+    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 __name__ == "__main__":
   app.launch(server_name="0.0.0.0")

gemini_api.py

@@ -2,7 +2,8 @@ import google.generativeai as genai
 import json
 import os
 
-genai.configure(api_key=os.getenv("API_KEY"))
+genai.configure(api_key="AIzaSyAT-eqTQUIpoVpmM-PL5hR5el1n6YqZRV4")
+# genai.configure(api_key=os.getenv("API_KEY"))
 generation_config = {
   "temperature": 0.9,
   "top_p": 1,
@@ -34,6 +35,7 @@ model = genai.GenerativeModel(
   safety_settings=safety_settings,
   generation_config=generation_config,
 )
+NO_ORDER = "Order Number not provided."
 
 def model_api(input, prompt_type):
     return prompt_type(input)
@@ -90,5 +92,9 @@ def ord_num(input):
 
     print(response.text)
     obj = json.loads(response.text)
-    return obj.get('order number', 'Order Number not provided.')
+    ret_num = obj.get('order number', NO_ORDER)
+    if ret_num is None:
+      return NO_ORDER
+    else:
+      return ret_num