Update app.py

app.py

@@ -1,453 +1,453 @@
-import streamlit as st
-from langchain.agents import AgentType, initialize_agent
-from langchain_community.chat_models import ChatOpenAI
-from langchain.schema import SystemMessage
-from langchain.prompts import MessagesPlaceholder
-from typing import Dict, Any
-from langchain.memory import ConversationBufferMemory
-from langchain.agents import initialize_agent, Tool
-import requests
-from datetime import datetime
-import os
-
-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", "")
-        chat_history = inputs.get("chat_history", [])
-        input_str = f"Human: {input_str}\nChat History: {chat_history}\n"
-        output_str = outputs.get("result", "")
-        return input_str, output_str
-
-
-def get_chat_completion(user_message, content):
-    # Define the API endpoint
-    url = "https://api.openai.com/v1/chat/completions"
-
-    # Define the headers
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {API_O}"
-    }
-
-    # Define the data payload
-    data = {
-        "model": "gpt-4-1106-preview",
-        "messages": [
-            {
-                "role": "system",
-                "content": content
-            },
-            {
-                "role": "user",
-                "content": user_message
-            }
-        ]
-    }
-    # Send the request
-    response = requests.post(url, headers=headers, json=data)
-    # Return the JSON response
-    return str(response.json()["choices"][0]["message"]["content"]).lower()
-
-
-def video_streaming_Test(prompt):
-    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."
-
-    if "One time test" in prompt:
-        if "now" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("now", formatted_date)
-        elif "today" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("today", formatted_date)
-
-        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}}"
-        else:
-            return "There are some infomration needed missing, information needed: Start_Date "
-
-    elif "Periodic test" in prompt or "Driving Mode" in prompt:
-        if "now" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("now", formatted_date)
-        elif "today" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("today", formatted_date)
-
-        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}}"
-        else:
-            return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
-    else:
-        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
-
-
-def wifi_test(prompt):
-    test_name = "wifi_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."
-    if "One time test" in prompt:
-        if "now" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("now", formatted_date)
-        elif "today" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("today", formatted_date)
-
-        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}}"
-        else:
-            return "There are some infomration needed missing, information needed: Start_Date "
-    elif "Periodic test" in prompt or "Driving Mode" in prompt:
-        if "now" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("now", formatted_date)
-        elif "today" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("today", formatted_date)
-
-        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}}"
-        else:
-            return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
-    else:
-        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
-
-
-def iot_test(prompt):
-    test_name = "iot_test"
-
-    content_one_time_test = "you only answer in shematic format,possibile asnwer are YES, NO. YES if in the prompt all the following are present: 1: Start_Date, 2: Mqtt broker address, 3: Quality of Service (QoS), 4: Mqtt Message Count; No if even only one 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, all the following are present: 1: Number_of_tests, 2: Start_Date, 3: Mqtt broker address, 4: Quality of Service (QoS), 5: Mqtt Message Count; 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 CARE ABOUT IT, . ACCEPTED ANSWER : YES / NO."
-    if "One time test" in prompt:
-        if "now" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("now", formatted_date)
-        elif "today" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("today", formatted_date)
-
-        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 }}"
-        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 "
-
-    elif "Periodic test" in prompt or "Driving Mode" in prompt:
-        if "now" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("now", formatted_date)
-        elif "today" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("today", formatted_date)
-
-        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}}"
-        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:
-        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
-
-
-def online_gaming_test(prompt):
-    test_name = "online_gaming_test"
-
-    content_one_time_test = "you only answer in shematic format,possibile asnwer are YES, NO. YES if in the prompt both the Start_Date and the predefined_gaming_servers are present, No if even only one 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, all the following  Number_of_tests, Start_Date, predefined_gaming_servers 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."
-    if "One time test" in prompt:
-        if "now" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("now", formatted_date)
-        elif "today" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("today", formatted_date)
-
-        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}}"
-        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:
-        if "now" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("now", formatted_date)
-        elif "today" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("today", formatted_date)
-
-        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}}"
-
-        else:
-            return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, predefined_gaming_servers "
-    else:
-        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
-
-
-def ping_test(prompt):
-    test_name = "ping_test"
-
-    content_one_time_test = "you only answer in shematic format,possibile asnwer are YES, NO. YES if in the prompt all the following: host_or_ip, ping_count, Start_Date are  present, No if even only one 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, all the following: host_or_ip, ping_count, 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."
-    if "One time test" in prompt:
-        if "now" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("now", formatted_date)
-        elif "today" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("today", formatted_date)
-
-        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  }}"
-        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:
-        if "now" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("now", formatted_date)
-        elif "today" in prompt:
-            now = datetime.now()
-            formatted_date = str(now.strftime("%d/%m/%Y"))
-            prompt = prompt.replace("today", formatted_date)
-
-        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 }}"
-        else:
-            return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, host_or_ip ping_count"
-    else:
-        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
-
-
-tools = [
-    Tool(
-        name="video_streaming_Test",
-        func=lambda prompt: video_streaming_Test(prompt),
-        description="""Use when the user wants to do a video_streaming_Test, 
-                        start the tool IF AND ONLY IF  you have ALL the information needed.
-                        There are three options for this test:
-                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy].)
-                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy].)
-                        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. 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 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="wifi_test",
-        func=lambda prompt: wifi_test(prompt),
-        description="""Use when the user wants to do a wifi_test, 
-                        start the tool IF AND ONLY IF  you have ALL the information needed.
-                        There are three options for this test:
-                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy].)
-                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy].)
-                        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. 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 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="iot_test",
-        func=lambda prompt: iot_test(prompt),
-        description="""Use when the user wants to do a iot_test, 
-                        start the tool IF AND ONLY IF  you have ALL the information needed.
-                        There are three options for this test:
-                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy], Mqtt broker address: str, Quality of Service (QoS): int[default:0], Mqtt Message Count: int[default:200] )
-                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy], Mqtt broker address: str, Quality of Service (QoS): int[default:0], Mqtt Message Count: int[default:200] )
-                        3: Driving Mode. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy] Mqtt broker address: str, Quality of Service (QoS): int[default:0], Mqtt Message Count: int[default:200] )
-                        So first of all you have to understand which tetst does the user wants to do. 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 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="online_gaming_test",
-        func=lambda prompt: online_gaming_test(prompt),
-        description="""Use when the user wants to do a online_gaming_test, 
-                        start the tool IF AND ONLY IF  you have ALL the information needed.
-                        There are three options for this test:
-                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy], predefined_gaming_servers: str[options: FORTNITE/PUBG])
-                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy], predefined_gaming_servers: str[options: FORTNITE/PUBG])
-                        3: Driving Mode. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy] predefined_gaming_servers: str[options: FORTNITE/PUBG])
-                        So first of all you have to understand which tetst does the user wants to do. 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 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="ping_test",
-        func=lambda prompt: ping_test(prompt),
-        description="""Use when the user wants to do a ping_test, 
-                        start the tool IF AND ONLY IF  you have ALL the information needed.
-                        There are three options for this test:
-                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy], host_or_ip: str/int, ping_count: int[predefined: 5])
-                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy], host_or_ip: str/int, ping_count: int[predefined: 5])
-                        3: Driving Mode. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy], host_or_ip: str/int, ping_count: int[predefined: 5])
-                        So first of all you have to understand which tetst does the user wants to do. 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 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)"""
-    )
-
-
-
-]
-
-# Set Streamlit page configuration
-st.set_page_config(page_title="🧠MemoryBot🤖", layout="centered")
-
-hide_st_style = """
-            <style>
-            #MainMenu {visibility: hidden;}
-            footer {visibility: hidden;}
-            header {visibility: hidden;}
-            </style>
-            """
-st.markdown(hide_st_style, unsafe_allow_html=True)
-
-# Initialize session states
-if "generated" not in st.session_state:
-    st.session_state["generated"] = []
-if "past" not in st.session_state:
-    st.session_state["past"] = []
-if "input" not in st.session_state:
-    st.session_state["input"] = ""
-if "stored_session" not in st.session_state:
-    st.session_state["stored_session"] = []
-
-# Define function to get user input
-
-
-def get_text():
-    """
-    Get the user input text.
-
-    Returns:
-        (str): The text entered by the user
-    """
-    input_text = st.text_input(
-        "You: ",
-        st.session_state["input"],
-        key="input",
-        placeholder="Enter a command or query here...",
-        label_visibility="hidden",
-    )
-    return input_text
-
-
-def new_chat():
-    """
-    Clears session state and starts a new chat.
-    """
-    save = []
-    for i in range(len(st.session_state["generated"]) - 1, -1, -1):
-        save.append("User:" + st.session_state["past"][i])
-        save.append("Bot:" + st.session_state["generated"][i])
-    st.session_state["stored_session"].append(save)
-    st.session_state["generated"] = []
-    st.session_state["past"] = []
-    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.title("🧠 Memory Bot 🤖")
-st.markdown(
-    """ 
-        > :black[**A Chatbot that remembers**     Made By - Bridge2e]
-        """
-)
-
-# Ask the user to enter their OpenAI API key
-
-# Session state storage would be ideal
-if API_O:
-    # Load the tools
-    # Initialize the memory object
-    memory = CustomMemory()
-    # Initialize the agent with the memory object
-    agent = initialize_agent(
-        tools,
-        ChatOpenAI(temperature=0, model_name=MODEL, openai_api_key=API_O),
-        agent=AgentType.CONVERSATIONAL_REACT_DESCRIPTION,
-        memory=memory,
-        verbose=True,
-    )
-
-# Add a button to start a new chat
-st.sidebar.button("New Chat", on_click=new_chat, type="primary")
-
-# Get the user input
-user_input = get_text()
-
-# Generate the output using the agent and the user input, and add the input/output to the session
-if user_input:
-    with st.spinner("Processing..."):
-        output = agent.run(
-            input=user_input, chat_history=st.session_state.past)
-    st.session_state.past.append(user_input)
-    st.session_state.generated.append(output)
-
-download_str = []
-# Display the conversation history using an expander, and allow the user to download it
-with st.expander("Conversation", expanded=True):
-    for i in range(len(st.session_state["generated"]) - 1, -1, -1):
-        st.info(st.session_state["past"][i], icon="🧐")
-        st.success(st.session_state["generated"][i], icon="🤖")
-        download_str.append(st.session_state["past"][i])
-        download_str.append(st.session_state["generated"][i])
-
-    # Can throw error - requires fix
-    # download_str = "\n".join(download_str)
-    # if download_str:
-    #    st.download_button("Download", download_str)
-
-# Display stored conversation sessions in the sidebar
-for i, sublist in enumerate(st.session_state.stored_session):
-    with st.sidebar.expander(label=f"Conversation-Session:{i}"):
-        st.write(sublist)
-
-# Allow the user to clear all stored conversation sessions
-if st.session_state.stored_session:
-    if st.sidebar.checkbox("Clear-all"):
-        del st.session_state.stored_session
-
-with st.sidebar:
-    st.markdown("---")
+import streamlit as st
+from langchain.agents import AgentType, initialize_agent
+from langchain_community.chat_models import ChatOpenAI
+from langchain.schema import SystemMessage
+from langchain.prompts import MessagesPlaceholder
+from typing import Dict, Any
+from langchain.memory import ConversationBufferMemory
+from langchain.agents import initialize_agent, Tool
+import requests
+from datetime import datetime
+import os
+
+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", "")
+        chat_history = inputs.get("chat_history", [])
+        input_str = f"Human: {input_str}\nChat History: {chat_history}\n"
+        output_str = outputs.get("result", "")
+        return input_str, output_str
+
+
+def get_chat_completion(user_message, content):
+    # Define the API endpoint
+    url = "https://api.openai.com/v1/chat/completions"
+
+    # Define the headers
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {API_O}"
+    }
+
+    # Define the data payload
+    data = {
+        "model": "gpt-4-1106-preview",
+        "messages": [
+            {
+                "role": "system",
+                "content": content
+            },
+            {
+                "role": "user",
+                "content": user_message
+            }
+        ]
+    }
+    # Send the request
+    response = requests.post(url, headers=headers, json=data)
+    # Return the JSON response
+    return str(response.json()["choices"][0]["message"]["content"]).lower()
+
+
+def video_streaming_Test(prompt):
+    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."
+
+    if "One time test" in prompt:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        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}}"
+        else:
+            return "There are some infomration needed missing, information needed: Start_Date "
+
+    elif "Periodic test" in prompt or "Driving Mode" in prompt:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        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}}"
+        else:
+            return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
+    else:
+        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
+
+
+def wifi_test(prompt):
+    test_name = "wifi_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."
+    if "One time test" in prompt:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        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}}"
+        else:
+            return "There are some infomration needed missing, information needed: Start_Date "
+    elif "Periodic test" in prompt or "Driving Mode" in prompt:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        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}}"
+        else:
+            return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
+    else:
+        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
+
+
+def iot_test(prompt):
+    test_name = "iot_test"
+
+    content_one_time_test = "you only answer in shematic format,possibile asnwer are YES, NO. YES if in the prompt all the following are present: 1: Start_Date, 2: Mqtt broker address, 3: Quality of Service (QoS), 4: Mqtt Message Count; No if even only one 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, all the following are present: 1: Number_of_tests, 2: Start_Date, 3: Mqtt broker address, 4: Quality of Service (QoS), 5: Mqtt Message Count; 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 CARE ABOUT IT, . ACCEPTED ANSWER : YES / NO."
+    if "One time test" in prompt:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        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 }}"
+        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 "
+
+    elif "Periodic test" in prompt or "Driving Mode" in prompt:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        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}}"
+        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:
+        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
+
+
+def online_gaming_test(prompt):
+    test_name = "online_gaming_test"
+
+    content_one_time_test = "you only answer in shematic format,possibile asnwer are YES, NO. YES if in the prompt both the Start_Date and the predefined_gaming_servers are present, No if even only one 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, all the following  Number_of_tests, Start_Date, predefined_gaming_servers 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."
+    if "One time test" in prompt:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        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}}"
+        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:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        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}}"
+
+        else:
+            return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, predefined_gaming_servers "
+    else:
+        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
+
+
+def ping_test(prompt):
+    test_name = "ping_test"
+
+    content_one_time_test = "you only answer in shematic format,possibile asnwer are YES, NO. YES if in the prompt all the following: host_or_ip, ping_count, Start_Date are  present, No if even only one 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, all the following: host_or_ip, ping_count, 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."
+    if "One time test" in prompt:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        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  }}"
+        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:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        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 }}"
+        else:
+            return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date, host_or_ip ping_count"
+    else:
+        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
+
+
+tools = [
+    Tool(
+        name="video_streaming_Test",
+        func=lambda prompt: video_streaming_Test(prompt),
+        description="""Use when the user wants to do a video_streaming_Test, 
+                        start the tool IF AND ONLY IF  you have ALL the information needed.
+                        There are three options for this test, FIRST OF ALL YOU ALWAYS MUST KNOW WHICH ONE THE USER WANT, IF YOU DON'T KNOW, ASK. NEVER MAKE UP ONE:
+                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy].)
+                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy].)
+                        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.IF IS MISSING, ask back. 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 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="wifi_test",
+        func=lambda prompt: wifi_test(prompt),
+        description="""Use when the user wants to do a wifi_test, 
+                        start the tool IF AND ONLY IF  you have ALL the information needed.
+                        There are three options for this test, FIRST OF ALL YOU ALWAYS MUST KNOW WHICH ONE THE USER WANT, IF YOU DON'T KNOW, ASK. NEVER MAKE UP ONE:
+                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy].)
+                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy].)
+                        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.IF IS MISSING, ask back. 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 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="iot_test",
+        func=lambda prompt: iot_test(prompt),
+        description="""Use when the user wants to do a iot_test, 
+                        start the tool IF AND ONLY IF  you have ALL the information needed.
+                        There are three options for this , FIRST OF ALL YOU ALWAYS MUST KNOW WHICH ONE THE USER WANT, IF YOU DON'T KNOW, ASK. NEVER MAKE UP ONE:
+                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy], Mqtt broker address: str, Quality of Service (QoS): int[default:0], Mqtt Message Count: int[default:200] )
+                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy], Mqtt broker address: str, Quality of Service (QoS): int[default:0], Mqtt Message Count: int[default:200] )
+                        3: Driving Mode. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy] Mqtt broker address: str, Quality of Service (QoS): int[default:0], Mqtt Message Count: int[default:200] )
+                        So first of all you have to understand which tetst does the user wants to do. IF IS MISSING, ask back. 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 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="online_gaming_test",
+        func=lambda prompt: online_gaming_test(prompt),
+        description="""Use when the user wants to do a online_gaming_test, 
+                        start the tool IF AND ONLY IF  you have ALL the information needed.
+                        There are three options for this test, FIRST OF ALL YOU ALWAYS MUST KNOW WHICH ONE THE USER WANT, IF YOU DON'T KNOW, ASK. NEVER MAKE UP ONE:
+                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy], predefined_gaming_servers: str[options: FORTNITE/PUBG])
+                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy], predefined_gaming_servers: str[options: FORTNITE/PUBG])
+                        3: Driving Mode. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy] predefined_gaming_servers: str[options: FORTNITE/PUBG])
+                        So first of all you have to understand which tetst does the user wants to do. IF IS MISSING, ask back. 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 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="ping_test",
+        func=lambda prompt: ping_test(prompt),
+        description="""Use when the user wants to do a ping_test, 
+                        start the tool IF AND ONLY IF  you have ALL the information needed.
+                        There are three options for this test, FIRST OF ALL YOU ALWAYS MUST KNOW WHICH ONE THE USER WANT, IF YOU DON'T KNOW, ASK. NEVER MAKE UP ONE:
+                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy], host_or_ip: str/int, ping_count: int[predefined: 5])
+                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy], host_or_ip: str/int, ping_count: int[predefined: 5])
+                        3: Driving Mode. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy], host_or_ip: str/int, ping_count: int[predefined: 5])
+                        So first of all you have to understand which tetst does the user wants to do. IF IS MISSING, ask back. 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 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)"""
+    )
+
+
+
+]
+
+# Set Streamlit page configuration
+st.set_page_config(page_title="🧠MemoryBot🤖", layout="centered")
+
+hide_st_style = """
+            <style>
+            #MainMenu {visibility: hidden;}
+            footer {visibility: hidden;}
+            header {visibility: hidden;}
+            </style>
+            """
+st.markdown(hide_st_style, unsafe_allow_html=True)
+
+# Initialize session states
+if "generated" not in st.session_state:
+    st.session_state["generated"] = []
+if "past" not in st.session_state:
+    st.session_state["past"] = []
+if "input" not in st.session_state:
+    st.session_state["input"] = ""
+if "stored_session" not in st.session_state:
+    st.session_state["stored_session"] = []
+
+# Define function to get user input
+
+
+def get_text():
+    """
+    Get the user input text.
+
+    Returns:
+        (str): The text entered by the user
+    """
+    input_text = st.text_input(
+        "You: ",
+        st.session_state["input"],
+        key="input",
+        placeholder="Enter a command or query here...",
+        label_visibility="hidden",
+    )
+    return input_text
+
+
+def new_chat():
+    """
+    Clears session state and starts a new chat.
+    """
+    save = []
+    for i in range(len(st.session_state["generated"]) - 1, -1, -1):
+        save.append("User:" + st.session_state["past"][i])
+        save.append("Bot:" + st.session_state["generated"][i])
+    st.session_state["stored_session"].append(save)
+    st.session_state["generated"] = []
+    st.session_state["past"] = []
+    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.title("🧠 Memory Bot 🤖")
+st.markdown(
+    """ 
+        > :black[**A Chatbot that remembers**     Made By - Bridge2e]
+        """
+)
+
+# Ask the user to enter their OpenAI API key
+
+# Session state storage would be ideal
+if API_O:
+    # Load the tools
+    # Initialize the memory object
+    memory = CustomMemory()
+    # Initialize the agent with the memory object
+    agent = initialize_agent(
+        tools,
+        ChatOpenAI(temperature=0, model_name=MODEL, openai_api_key=API_O),
+        agent=AgentType.CONVERSATIONAL_REACT_DESCRIPTION,
+        memory=memory,
+        verbose=True,
+    )
+
+# Add a button to start a new chat
+st.sidebar.button("New Chat", on_click=new_chat, type="primary")
+
+# Get the user input
+user_input = get_text()
+
+# Generate the output using the agent and the user input, and add the input/output to the session
+if user_input:
+    with st.spinner("Processing..."):
+        output = agent.run(
+            input=user_input, chat_history=st.session_state.past)
+    st.session_state.past.append(user_input)
+    st.session_state.generated.append(output)
+
+download_str = []
+# Display the conversation history using an expander, and allow the user to download it
+with st.expander("Conversation", expanded=True):
+    for i in range(len(st.session_state["generated"]) - 1, -1, -1):
+        st.info(st.session_state["past"][i], icon="🧐")
+        st.success(st.session_state["generated"][i], icon="🤖")
+        download_str.append(st.session_state["past"][i])
+        download_str.append(st.session_state["generated"][i])
+
+    # Can throw error - requires fix
+    # download_str = "\n".join(download_str)
+    # if download_str:
+    #    st.download_button("Download", download_str)
+
+# Display stored conversation sessions in the sidebar
+for i, sublist in enumerate(st.session_state.stored_session):
+    with st.sidebar.expander(label=f"Conversation-Session:{i}"):
+        st.write(sublist)
+
+# Allow the user to clear all stored conversation sessions
+if st.session_state.stored_session:
+    if st.sidebar.checkbox("Clear-all"):
+        del st.session_state.stored_session
+
+with st.sidebar:
+    st.markdown("---")