Update app.py

app.py

@@ -9,10 +9,10 @@ from langchain.agents import initialize_agent, Tool
 import requests
 from datetime import datetime
 import os
-
-API_O = os.environ.get("OPENAI_API_KEY")
+#TODO aggiungre la frequenza per tutti i test, 
 
 
+API_O = os.environ.get("OPENAI_API_KEY")
 class CustomMemory(ConversationBufferMemory):
     def _get_input_output(self, inputs: Dict[str, Any], outputs: Dict[str, Any]) -> tuple:
         input_str = inputs.get("input", "")
@@ -70,7 +70,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"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"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}}"
         else:
             return "There are some infomration needed missing, information needed: Start_Date "
 
@@ -87,7 +87,7 @@ def video_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}}"
+            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}}"
         else:
             return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
     else:
@@ -737,6 +737,11 @@ tools = [
                         3: Driving Mode. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy].)
                         So first of all you have to understand which tetst does the user wants to do.ask DIRECTLLY the user which test options they want to perform, don't try to guess it. Then you have to find the informaton needed for that test.Users alway forget something, if you are missing some info, just pass none.In the field date,IF AND ONLY IF users says now or today, just pass now or today. NEVER CREATE INPUT IF NOT WRITTEN BY THE USER. IN THE INPUT PASS THE INFORMATION IN JSON FORMAT. IN THE INPUR PASS THE SPECIFIED TET TYPE, ask if missing test_type:(One time test/Periodic test/Driving Mode)"""
     ),
+        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"
+        )
 ]
 
 # Set Streamlit page configuration
@@ -795,20 +800,12 @@ def new_chat():
     st.session_state["input"] = ""
 
 
-with st.sidebar.expander(" 🛠️ Settings ", expanded=False):
-    MODEL = st.selectbox(
-        label="Model",
-        options=[
-            "gpt-4-1106-preview",
-            "gpt-4-turbo",
-            "gpt-4",
-        ],
-    )
+st.sidebar.image(r"assets\logo.png")
 
 st.title("🧠 Memory Bot 🤖")
 st.markdown(
     """ 
-        > :black[**A Chatbot that remembers**     Made By - Bridge2e]
+        > :black[**A Chatbot that remembers**     Made By - Palo Rimoldi & Bridge2e]
         """
 )
 
@@ -816,6 +813,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()