2 qestions, fix session init, time error handling

app.py

@@ -95,6 +95,8 @@ async def on_action_display_queries_responses(action):
 @cl.on_chat_start
 async def on_chat_start():
     session_state = SessionState()
+    print("Creating new session state")
+    print(session_state.responses)
     set_session_state_variables(session_state)
     cl.user_session.set("session_state", session_state)
     session_state.llm_model = llm_model

classes.py

@@ -36,6 +36,10 @@ class SessionState:
     opportunity_review_results = None
     opportunity_review_report = None
     def __init__(self):
+        self.company = None
+        self.customer = None
+        self.opportunity = None
+        self.scenario = None
         self.session_stage = "research"
         self.do_evaluation = False
         self.do_opportunity_analysis = False
@@ -103,11 +107,7 @@ class SessionState:
             self.activity = activity
             self.next_steps = next_steps
             
-    def __init__(self):
-        self.company = None
-        self.customer = None
-        self.opportunity = None
-        self.scenario = None
+
 
     def add_company_info(self, name, description, product, product_summary, product_description):
         self.company = self.Company(name, description, product, product_summary, product_description)

utils_objections.py

@@ -22,9 +22,9 @@ async def create_objections(session_state):
 
         objections = [
             "1. Can you provide customer references in the banking and financial services industry?",
-            "2. Second question, what training options are available for our team, given the number of employees and their global distribution?",
-            "3. Last but not least, your pricing seems high compared to some other solutions we've seen. Can you provide any flexibility with HSBC's pricing?",
+            "2. What training options are available for our team, given the number of employees and their global distribution?",  
         ]
+        #  "3. Last but not least, your pricing seems high compared to some other solutions we've seen. Can you provide any flexibility with HSBC's pricing?",
 
     else:
         customer_document_file = session_state.customer_research_report_pdf

utils_output.py

@@ -107,8 +107,6 @@ async def display_evaluation_results(cl, session_state):
 
     for index, resp in enumerate(session_state.responses):
         eval_score = resp.get('evaluation_score', 0)
-        print(eval_score)
-        print(type(eval_score))
         if eval_score == 1:
             eval_output = "Hit"
         elif eval_score == 0:

utils_simulation.py

@@ -67,10 +67,17 @@ async def do_simulation(client, session_state, message):
                 await cl.Message(message_to_rep).send()
             session_state.status = "complete"
             end_time = datetime.now()
-            duration = end_time - session_state.start_time
-            duration_minutes = round(duration.total_seconds() / 60)
             session_state.end_time = end_time
-            session_state.duration_minutes = duration_minutes
+            if session_state.start_time:
+                try:
+                    duration = end_time - session_state.start_time
+                    duration_minutes = round(duration.total_seconds() / 60)
+                except:
+                    print("Error calculating duration")
+                    duration = 130
+                    duration_minutes = 2
+                    
+            session_state.duration_minutes = duration_minutes   
             if session_state.do_evaluation:
                 await display_evaluation_results(cl, session_state)
             else: