Update app.py

app.py

@@ -10,6 +10,8 @@ import requests
 from datetime import datetime
 import os
 #TODO aggiungre la frequenza per tutti i test, 
+import uuid
+
 
 API_O = os.environ.get("OPENAI_API_KEY")
 class CustomMemory(ConversationBufferMemory):
@@ -52,7 +54,7 @@ def get_chat_completion(user_message, content):
 
 
 def video_streaming_Test(prompt):
-    test_name = "video_streaming_Test"
+    test_name = "video_streaming_test"
     content_one_time_test = "you only answer in shematic format,possibile asnwer are YES, NO. YES if in the prompt the Start_Date is present, No if is missing. IF 'null' is in the input , the answer is NO. ACCEPTED ANSWER : YES / NO."
     content_multiple_tests = "you only answer in shematic format,possibile asnwer are YES, NO. YES if in the prompt, both  Number_of_tests, Start_Date are present, No if even only one is missing.IF 'null' is in the input , the answer is NO. THERE CAN BE MORE IN THE PROMPT, YOU DON'T HAVE TO CAARE ABOUT IT. ACCEPTED ANSWER : YES / NO."
 
@@ -69,7 +71,7 @@ def video_streaming_Test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"first run the new_chat tool , then report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
+            return f"1. run the new_chat tool, 2. report to the user in json format: test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date "
 
@@ -86,7 +88,7 @@ def video_streaming_Test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"first run the new_chat tool , then report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
+            return f"1. run the new_chat tool, 2. report to the user in json format: test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
     else:
@@ -111,7 +113,7 @@ def wifi_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date "
     elif "Periodic test" in prompt or "Driving Mode" in prompt:
@@ -127,7 +129,7 @@ def wifi_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
     else:
@@ -152,7 +154,7 @@ def iot_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, Mqtt broker address: selected_Mqtt_broker_address, Quality of Service (QoS): selected_Quality_of_Service, Mqtt Message Count: selected_Mqtt_Message_Count }}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, Mqtt broker address: selected_Mqtt_broker_address, Quality of Service (QoS): selected_Quality_of_Service, Mqtt Message Count: selected_Mqtt_Message_Count }}"
         else:
             return "There are some infomration needed missing, information needed: 1: Start_Date, 2: Mqtt broker address, 3: Quality of Service (QoS), 4: Mqtt Message Count "
 
@@ -169,7 +171,7 @@ def iot_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Number_of_tests: selected_number_of_tests,  Start_Date: selected_start_date, Mqtt broker address: selected_Mqtt_broker_address, Quality of Service (QoS): selected_Quality_of_Service, Mqtt Message Count}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Number_of_tests: selected_number_of_tests,  Start_Date: selected_start_date, Mqtt broker address: selected_Mqtt_broker_address, Quality of Service (QoS): selected_Quality_of_Service, Mqtt Message Count}}"
         else:
             return "There are some infomration needed missing, information needed: 1: Number_of_tests, 2: Start_Date, 3: Mqtt broker address, 4: Quality of Service (QoS), 5: Mqtt Message "
     else:
@@ -194,7 +196,7 @@ def online_gaming_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, game_servers: predefined_gaming_servers}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, game_servers: predefined_gaming_servers}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date, predefined_gaming_servers "
     elif "Periodic test" in prompt or "Driving Mode" in prompt:
@@ -210,7 +212,7 @@ def online_gaming_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, game_servers: predefined_gaming_servers}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, game_servers: predefined_gaming_servers}}"
 
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, predefined_gaming_servers "
@@ -235,7 +237,7 @@ def youtube_streaming_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, video_quality: selected_video_quality}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, video_quality: selected_video_quality}}"
         else:
             return "There are some infomration needed missing, information needed: video_quality ,Start_Date"
     elif "Periodic test" in prompt or "Driving Mode" in prompt:
@@ -251,7 +253,7 @@ def youtube_streaming_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, video_quality: selected_video_quality}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, video_quality: selected_video_quality}}"
 
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, video_quality "
@@ -276,7 +278,7 @@ def ping_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date,host_or_ip: selected_host_or_ip, ping_count: selected_ping_count  }}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date,host_or_ip: selected_host_or_ip, ping_count: selected_ping_count  }}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date ,host_or_ip, ping_count"
     elif "Periodic test" in prompt or "Driving Mode" in prompt:
@@ -292,7 +294,7 @@ def ping_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date,host_or_ip: selected_host_or_ip, ping_count: selected_ping_count }}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date,host_or_ip: selected_host_or_ip, ping_count: selected_ping_count }}"
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, host_or_ip ping_count"
     else:
@@ -316,7 +318,7 @@ def speed_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, Speed_type: selected_Speed_type}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, Speed_type: selected_Speed_type}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date, Speed_type "
     elif "Periodic test" in prompt or "Driving Mode" in prompt:
@@ -332,7 +334,7 @@ def speed_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, Speed_type: selected_Speed_type}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, Speed_type: selected_Speed_type}}"
 
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, Speed_type"
@@ -357,7 +359,7 @@ def web_browsing_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, Host_list: selected_Host_list}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, Host_list: selected_Host_list}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date, Host_list"
     elif "Periodic test" in prompt or "Driving Mode" in prompt:
@@ -373,7 +375,7 @@ def web_browsing_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, Host_list: selected_Host_list}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, Host_list: selected_Host_list}}"
 
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, Host_list"
@@ -398,7 +400,7 @@ def call_receive_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date "
 
@@ -415,7 +417,7 @@ def call_receive_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
     else:
@@ -439,7 +441,7 @@ def sms_receive_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date "
 
@@ -456,7 +458,7 @@ def sms_receive_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
     else:
@@ -481,7 +483,7 @@ def voice_call_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, Phone_number: selected_Phone_number}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, Phone_number: selected_Phone_number}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date, Phone_number"
     elif "Periodic test" in prompt or "Driving Mode" in prompt:
@@ -497,7 +499,7 @@ def voice_call_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, Phone_number: selected_Phone_number}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, Phone_number: selected_Phone_number}}"
 
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, Phone_number"
@@ -524,7 +526,7 @@ def sms_send_test(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, phone_number: selected_phone_number, sms_message: selected_sms_message}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date, phone_number: selected_phone_number, sms_message: selected_sms_message}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date, phone_number, sms_message"
     elif "Periodic test" in prompt or "Driving Mode" in prompt:
@@ -540,7 +542,7 @@ def sms_send_test(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, phone_number: selected_phone_number, sms_message: selected_sms_message}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date, phone_number: selected_phone_number, sms_message: selected_sms_message}}"
 
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, phone_number, sms_message"
@@ -565,7 +567,7 @@ def service_mode_tests(prompt):
         response = get_chat_completion(prompt, content_one_time_test)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date "
 
@@ -582,7 +584,7 @@ def service_mode_tests(prompt):
         response = get_chat_completion(prompt, content_multiple_tests)
         if "yes" in str(response).lower():
             print(prompt)
-            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
+            return "1. run the new_chat tool, 2. report to the user in json format: test started successfully:  {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
     else:
@@ -739,7 +741,7 @@ tools = [
         Tool(
             name="new_chat",
             func=lambda prompt: new_chat(),
-            description="use this tool always when the agent made to finish a succesfull action, when the chain is finished, this tool is used to clear the chat and to create a new one"
+            description="use this tool always when the agent made to finish a succesfull action, when the chain is finished, this tool is used to clear the chat and to create a new one. Use whrn the agent decides to use the new_chat tool"
         )
 ]
 
@@ -765,7 +767,7 @@ if "input" not in st.session_state:
 if "stored_session" not in st.session_state:
     st.session_state["stored_session"] = []
 
-# Define function to get user input
+# Define tool to get user input
 
 
 def get_text():
@@ -777,7 +779,7 @@ def get_text():
     """
     input_text = st.text_input(
         "You: ",
-        st.session_state["input"],
+        st.session_state.get("user_input", ""),  # Retrieve user input from new key
         key="input",
         placeholder="Enter a command or query here...",
         label_visibility="hidden",
@@ -796,7 +798,9 @@ def new_chat():
     st.session_state["stored_session"].append(save)
     st.session_state["generated"] = []
     st.session_state["past"] = []
-    st.session_state["input"] = ""
+    st.session_state["user_input"] = ""  # Clear user input
+    st.session_state["bot_response"] = ""  # Clear bot response
+
 
 
 st.sidebar.image("./assets/logo.png")
@@ -804,7 +808,7 @@ st.sidebar.image("./assets/logo.png")
 st.title("🧠 Memory Bot 🤖")
 st.markdown(
     """ 
-        > :black[**A Chatbot that remembers**     Made By - Palo Rimoldi & Bridge2e]
+        > :black[**A Chatbot that remembers**     Made By - Bridge2e]
         """
 )
 
@@ -813,6 +817,7 @@ st.markdown(
 # Session state storage would be ideal
 if API_O:
     MODEL= "gpt-4-1106-preview"
+
     # Load the tools
     # Initialize the memory object
     memory = CustomMemory()