Upload app.py

app.py

@@ -0,0 +1,60 @@
+import streamlit as st
+from api_call_module import get_answer
+
+processes = ["Praksa", "Izrada završnog rada"]
+file_name = "ans.txt"
+
+# Read answers from ans.txt
+answers = []
+try:
+    answers = [line.replace("\n", "") for line in open(file_name, "r")]
+except FileNotFoundError:
+    open(file_name, "w")
+
+
+# Save the answer to ans.txt
+def save_answer(answer: str) -> None:
+    f = open(file_name, "a")
+    f.write(f"{answer}\n")
+    f.close()
+
+
+# Submits the query and saves the answer
+def submit_query() -> None:
+    if query == "":
+        return
+
+    ans = get_answer(query, process)
+
+    save_answer("A: " + ans)
+    save_answer("Q: " + query)
+    save_answer("**Question {}**")
+    save_answer("")
+
+
+# Clears the chat
+def clear() -> None:
+    open(file_name, 'w').close()
+    pass
+
+
+# Sidebar
+with st.sidebar:
+    st.title("Set up your chatbot!")
+    process = st.selectbox("Select a process", processes)
+    st.caption("You can find this chatbot implementation [here](https://huggingface.co/spaces/rkrstacic/"
+               "Software-module-for-answering-questions-on-processes/tree/main)")
+
+# Main
+st.title("Welcome to the process answering chatbot!")
+query = st.text_input("Ask me a question", placeholder="e.g. Koliko traje proces praksa?")
+
+# Buttons
+col1, col2 = st.columns([6, 1])
+col1.button("Send", on_click=submit_query)
+col2.button("Clear chat", on_click=clear)
+
+
+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))