update readme

Reviewed-on: #8

README.md

@@ -1,4 +1,22 @@
-# Steps to init
+# Steps to initialize the project
-- just need to do this one time
 
-`python -m venv .venv`
+```sh
+python -m venv .venv
+```
+
+```sh
+source .venv/bin/active
+```
+
+## Steps to install dependencies
+
+```sh
+pip install -r requirements.txt
+```
+
+## Seps to run
+
+```sh
+streamlit run main.py
+```
+Make sure that [rust server](https://git.ssdd.dev/ssdd/hackpsuchatbot) is active before asking a prompt.

app.py

@@ -1,85 +0,0 @@
-import streamlit as st
-import json
-import os
-import datetime
-
-# Define the path to the local JSON file where conversation will be stored
-FILE_PATH = "/home/sapan/PycharmProjects/maargdarshak/conversation_history.json"
-
-
-# Function to save prompt and response to local JSON file
-def save_to_local(prompt, response):
-    data = {
-        "prompt": prompt,
-        "response": response,
-        "timestamp": str(datetime.datetime.now())
-    }
-
-    # If the file exists, load existing data, handle empty or corrupt files
-    if os.path.exists(FILE_PATH):
-        try:
-            with open(FILE_PATH, 'r') as file:
-                history = json.load(file)
-        except json.JSONDecodeError:
-            history = []  # If the file is empty or corrupted, start with an empty list
-    else:
-        history = []
-
-    # Append the new conversation entry
-    history.append(data)
-
-    # Save the updated history back to the JSON file
-    with open(FILE_PATH, 'w') as file:
-        json.dump(history, file, indent=4)
-
-
-# Function to load conversation history from local JSON file
-def load_from_local():
-    if os.path.exists(FILE_PATH):
-        try:
-            with open(FILE_PATH, 'r') as file:
-                return json.load(file)
-        except json.JSONDecodeError:
-            return []  # Return empty list if the file is empty or corrupt
-    return []
-
-
-# Initialize session state to store conversation history for the session
-if "conversation" not in st.session_state:
-    st.session_state["conversation"] = []
-
-# Streamlit App
-st.title("Chatbot with Local Chat History")
-
-# Text input for user prompt
-user_input = st.text_input("Ask the chatbot something:")
-
-# If the user submits input
-if user_input:
-    # Simulated response (for now it's echoing the input)
-    response = f"Echo: {user_input}"
-
-    # Save the current prompt and response to session state for chat history
-    st.session_state["conversation"].append({"prompt": user_input, "response": response})
-
-    # Also save the conversation to local storage (JSON file)
-    save_to_local(user_input, response)
-
-# Display the chat history (conversation for the session)
-st.write("### Chat History")
-for convo in st.session_state["conversation"]:
-    st.write(f"**You**: {convo['prompt']}")
-    st.write(f"**Bot**: {convo['response']}")
-    st.write("---")
-
-# Button to show entire history from local JSON file
-if st.button("Show Full Local Conversation History"):
-    st.write("### Full Local Conversation History")
-
-    # Load history from the local file and display it
-    full_history = load_from_local()
-    for convo in full_history:
-        st.write(f"**You**: {convo['prompt']}")
-        st.write(f"**Bot**: {convo['response']}")
-        st.write(f"Timestamp: {convo['timestamp']}")
-        st.write("---")

conversation_history.json

@@ -1,32 +0,0 @@
-[
-    {
-        "prompt": "sa",
-        "response": "Echo: sa",
-        "timestamp": "2024-10-13 02:30:26.507681"
-    },
-    {
-        "prompt": "sadsas",
-        "response": "Echo: sadsas",
-        "timestamp": "2024-10-13 02:30:27.815409"
-    },
-    {
-        "prompt": "sadsasdaswdewf",
-        "response": "Echo: sadsasdaswdewf",
-        "timestamp": "2024-10-13 02:30:28.942786"
-    },
-    {
-        "prompt": "ffhj",
-        "response": "Echo: ffhj",
-        "timestamp": "2024-10-13 02:31:17.158006"
-    },
-    {
-        "prompt": "ffhj;lkjhg",
-        "response": "Echo: ffhj;lkjhg",
-        "timestamp": "2024-10-13 02:31:19.262145"
-    },
-    {
-        "prompt": "ffhj;lkjhglkjh",
-        "response": "Echo: ffhj;lkjhglkjh",
-        "timestamp": "2024-10-13 02:31:21.288840"
-    }
-]

main.py

@@ -1,6 +1,101 @@
-class HelloWorld:
+import streamlit as st
-    def __init__(self):
+import requests
-        print("Hello, World!")
+import datetime
+import json
+import os
+from streamlit_cookies_manager import CookieManager
 
-if __name__ == "__main__":
+# Initialize the Cookie Manager (no encryption)
-    hello = HelloWorld()
+cookies = CookieManager(prefix="ktosiek/streamlit-cookies-manager/")
+if not cookies.ready():
+    st.stop()
+
+# Define the file path for storing user data
+user_data_file = "user_data.json"
+
+# Helper function to load user data from JSON file
+def load_user_data():
+    if os.path.exists(user_data_file):
+        with open(user_data_file, 'r') as file:
+            return json.load(file)
+    else:
+        save_user_data({})
+        return {}
+
+# Helper function to save user data to JSON file
+def save_user_data(data):
+    with open(user_data_file, 'w') as file:
+        json.dump(data, file, indent=4)
+
+# Load existing user data
+user_data = load_user_data()
+
+# Generate a user ID and store it in cookies if not already set
+if cookies.get('user_id') is None:  # Correct check for user_id in cookies
+    cookies['user_id'] = str(datetime.datetime.now().timestamp())  # Use timestamp as a unique ID
+    cookies.save()
+
+user_id = cookies.get('user_id')
+# st.write(f"Current User ID: {user_id}")
+
+# Check if the user exists in the JSON file
+if user_id not in user_data:
+    timestamp = str(datetime.datetime.now().isoformat())  # Unique timestamp for each user
+    user_data[user_id] = {
+        "timestamp": timestamp,
+        "messages": []  # Empty chat history
+    }
+    save_user_data(user_data)
+    st.write("New user detected! Created user entry in the JSON file.")
+else:
+    # If user exists, load the chat history from the JSON file
+    st.session_state.messages = user_data[user_id]["messages"]
+    # st.write("Returning user detected! Loaded chat history from the JSON file.")
+
+# Function to send POST request to a local server
+def send_post_request(prompt):
+    url = "http://localhost:19194/"  # Update this to your actual server URL
+    payload = prompt.title()
+    response = requests.post(url, json=payload)
+    json = response.json()
+    if "response" in json:
+        return json["response"]
+    else:
+        return json
+
+st.title('Ask Maargdarshak')
+
+if 'messages' not in st.session_state:
+    st.session_state.messages = []
+
+# Load chat history from session state
+for message in st.session_state.messages:
+    st.chat_message(message['role']).markdown(message['content'])
+
+# Prompt input
+prompt = st.chat_input("Ask your question here")
+
+if prompt:
+    # Display the user prompt in the chat UI
+    st.chat_message('user').markdown(prompt)
+    st.session_state.messages.append({'role': 'user', 'content': prompt})
+
+    # Get the bot's response via a POST request
+    with st.spinner("Fetching data..."):
+        try:
+            response = send_post_request(prompt)
+        except Exception as e:
+            response = f"An error occurred: {e}"
+        st.success("Data fetched successfully!")
+
+    # Display the bot's response in the chat UI
+    st.chat_message('bot').markdown(response)
+    st.session_state.messages.append({'role': 'bot', 'content': response})
+
+    # Update the user's chat history in the JSON file
+    user_data[user_id]["messages"] = st.session_state.messages
+    save_user_data(user_data)
+
+    # Save updated history to cookies (optional)
+    cookies['history'] = st.session_state.messages
+    cookies.save()

requirements.txt

@@ -0,0 +1,115 @@
+aiohappyeyeballs==2.4.3
+aiohttp==3.10.10
+aiosignal==1.3.1
+altair==5.4.1
+annotated-types==0.7.0
+anyio==4.6.0
+appnope==0.1.4
+asttokens==2.4.1
+async-timeout==4.0.3
+attrs==24.2.0
+backcall==0.2.0
+beautifulsoup4==4.12.3
+bleach==6.1.0
+blinker==1.8.2
+bs4==0.0.2
+cachetools==5.5.0
+certifi==2024.8.30
+cffi==1.17.1
+charset-normalizer==3.4.0
+click==8.1.7
+cryptography==43.0.1
+decorator==5.1.1
+defusedxml==0.7.1
+dnspython==2.7.0
+docopt==0.6.2
+exceptiongroup==1.2.2
+executing==2.1.0
+fastjsonschema==2.20.0
+frozenlist==1.4.1
+gitdb==4.0.11
+GitPython==3.1.43
+h11==0.14.0
+httpcore==1.0.6
+httpx==0.27.2
+idna==3.10
+importlib_metadata==8.5.0
+ipython==8.12.3
+jedi==0.19.1
+Jinja2==3.1.4
+jsonpatch==1.33
+jsonpointer==3.0.0
+jsonschema==4.23.0
+jsonschema-specifications==2024.10.1
+jupyter_client==8.6.3
+jupyter_core==5.7.2
+jupyterlab_pygments==0.3.0
+langchain==0.3.3
+langchain-core==0.3.10
+langchain-text-splitters==0.3.0
+langsmith==0.1.134
+Markdown==3.7
+markdown-it-py==3.0.0
+MarkupSafe==3.0.1
+matplotlib-inline==0.1.7
+mdurl==0.1.2
+mistune==3.0.2
+multidict==6.1.0
+narwhals==1.9.3
+nbclient==0.10.0
+nbconvert==7.16.4
+nbformat==5.10.4
+numpy==1.26.4
+orjson==3.10.7
+packaging==24.1
+pandas==2.2.3
+pandocfilters==1.5.1
+parso==0.8.4
+pexpect==4.9.0
+pickleshare==0.7.5
+pillow==10.4.0
+pipreqs==0.5.0
+platformdirs==4.3.6
+prompt_toolkit==3.0.48
+propcache==0.2.0
+protobuf==5.28.2
+ptyprocess==0.7.0
+pure_eval==0.2.3
+pyarrow==17.0.0
+pycparser==2.22
+pydantic==2.9.2
+pydantic_core==2.23.4
+pydeck==0.9.1
+Pygments==2.18.0
+pymongo==4.10.1
+python-dateutil==2.9.0.post0
+pytz==2024.2
+PyYAML==6.0.2
+pyzmq==26.2.0
+referencing==0.35.1
+requests==2.32.3
+requests-toolbelt==1.0.0
+rich==13.9.2
+rpds-py==0.20.0
+six==1.16.0
+smmap==5.0.1
+sniffio==1.3.1
+soupsieve==2.6
+SQLAlchemy==2.0.35
+stack-data==0.6.3
+streamlit==1.39.0
+streamlit-cookies-manager==0.2.0
+tenacity==8.5.0
+tinycss2==1.3.0
+toml==0.10.2
+tornado==6.4.1
+traitlets==5.14.3
+typing_extensions==4.12.2
+tzdata==2024.2
+urllib3==2.2.3
+watchdog==5.0.3
+wcwidth==0.2.13
+webencodings==0.5.1
+yarg==0.1.9
+yarl==1.15.0
+zipp==3.20.2