Update temp.py

temp.py

@@ -1,7 +1,7 @@
 from typing import Dict, Any, Tuple
 
-from dotenv import load_dotenv
-load_dotenv('.env')
+# from dotenv import load_dotenv
+# load_dotenv('.env')
 import gradio as gr
 import json
 from datetime import datetime, timedelta
@@ -214,6 +214,7 @@ class MailChatBot:
                                                 cost=cost
                                             )
         session.available_data['delivery_time'] = delivery_time
+        session.available_data['pickup_time'] = pickup_time.strftime('%H:%M')
 
         return email
     
@@ -224,6 +225,14 @@ class MailChatBot:
         print('SESSION: ', req.session_hash)
         session: Session = self.sessions.get(req.session_hash,
                                              Session(id=req.session_hash)).copy()
+        # general_chain = GeminiChain.from_system_prompt(template=prompts.ROUTER_CHAIN_PROMPT,
+        #                                input_variables=['query'],
+        #                                model_name=get_env_variable("OPENAI_MODEL_TURBO"),
+        #                                temperature=self.temperature, api_key=get_env_variable("OPENAI_API_KEY"),
+        #                                verbose=self.verbose, output_parser=self.parser
+        #                             )
+        # # output = general_chain.invoke({'query': message})['text']
+        # # return output
         if session.state == 'done':
             return 'We have your order in processing. If you have any queries Kindly contact us at 123-456-7890'
         if session.state == 'rejected':
@@ -244,7 +253,7 @@ class MailChatBot:
                     session.state = 'confirmation'
                     response = self._get_confirmation_email(session)
                     data = session.available_data.copy()
-                    data['pickup_time'] = data["pickup_date"][0]["date"] + " " + data["pickup_time"]
+                    data['pickup_time'] = data["pickup_date"][0]["date"] + " " + data['pickup_time']
 
                     cost, row_id = update_data_in_gsheet(data,
                                                          sheet_id=SHEET_ID,
@@ -288,15 +297,19 @@ class MailChatBot:
             if session.available_data.get('pickup_location') != output.get('pickup_location') or \
                 session.available_data.get('delivery_location') != output.get('delivery_location'):
                 session.state = "update"
-            
+
+        reasons = output.get("reasoning", {})    
+        output = output.get("required_data", {})    
         output = Data(**output)
         print('OUTPUT: ', output)
+        print('REASONS: ', reasons)
         output, missing_data = self._validate_response(output)
-        session.update_data(output)        
+        session.update_data(output)
+        
         if missing_data:
             if 'pickup_date' in missing_data:
                 chain = self._init_chain('missing_data')
-                data = {"required_data": "\n".join([f"{idx+1}. {val}"\
+                data = {"required_data": "\n".join([f"{idx+1}. {val}: {reasons[val]}"\
                     for idx, val in enumerate(missing_data)]),
                         "cdate": current_date,
                         "cday": current_day,
@@ -304,7 +317,7 @@ class MailChatBot:
                         }
             else:
                 chain = self._init_chain('missing_data_with_date')
-                data = {"required_data": "\n".join([f"{idx+1}. {val}"\
+                data = {"required_data": "\n".join([f"{idx+1}. {val}: {reasons[val]}"\
                     for idx, val in enumerate(missing_data)]),
                         'date': output.pickup_date[0]['date'],
                         'day': output.pickup_date[0]['day_of_week'],