Update app.py

app.py

@@ -2,21 +2,21 @@ import streamlit as st
 from api_call_module import get_answer
 
 processes = ["Praksa", "Izrada završnog rada"]
-file_name = "ans.txt"
+answers_key = "answers"
 
-# Read answers from ans.txt
-answers = []
-try:
-    answers = [line.replace("\n", "") for line in open(file_name, "r")]
-except FileNotFoundError:
-    open(file_name, "w")
+# Session init
+if answers_key not in st.session_state:
+    st.session_state[answers_key] = []
 
 
-# Save the answer to ans.txt
+# Save the answer to session
 def save_answer(answer: str) -> None:
-    f = open(file_name, "a")
-    f.write(f"{answer}\n")
-    f.close()
+    st.session_state[answers_key].append(answer)
+
+
+# Clears the session
+def clear_session() -> None:
+    st.session_state[answers_key] = []
 
 
 # Submits the query and saves the answer
@@ -32,12 +32,6 @@ def submit_query() -> None:
     save_answer("")
 
 
-# Clears the chat
-def clear() -> None:
-    open(file_name, 'w').close()
-    pass
-
-
 # Sidebar
 with st.sidebar:
     st.title("Set up your chatbot!")
@@ -47,14 +41,14 @@ with st.sidebar:
 
 # Main
 st.title("Welcome to the process answering chatbot!")
-query = st.text_input("Ask me a question", placeholder="e.g. Koliko traje proces praksa?")
+query = st.text_input("Ask me a question", placeholder="e.g. Koliko traje proces prakse?")
 
 # Buttons
 col1, col2 = st.columns([6, 1])
 col1.button("Send", on_click=submit_query)
-col2.button("Clear chat", on_click=clear)
-
+col2.button("Clear chat", on_click=clear_session)
 
+# Chat history
 with st.expander("Chat History", expanded=True):
-    for index, ans in enumerate(answers[::-1]):
-        st.write(ans if index % 4 != 1 else ans.format((len(answers) - index) // 4 + 1))
+    for index, ans in enumerate(st.session_state[answers_key][::-1]):
+        st.write(ans if index % 4 != 1 else ans.format((len(st.session_state[answers_key]) - index) // 4 + 1))