add init

chat.py

@@ -0,0 +1,43 @@
+import streamlit as st
+from typing import Callable
+
+
+
+RESPONSE_LABEL = 'chat_response'
+PROMPT_LABEL = 'chat_user_input'
+
+class Chat:
+    def __init__(self): 
+        if RESPONSE_LABEL not in st.session_state:
+            st.session_state[RESPONSE_LABEL] = []
+
+        if PROMPT_LABEL not in st.session_state:
+            st.session_state[PROMPT_LABEL] = []
+
+    def get_input(self, process_prompt: Callable, *args):
+        """
+        process_prompt(promt: str, *args) -> tuple(Any, Callable)
+            callback to process the chat promt, it takes the promt for input
+            and returns a tuple with the response and a render callback
+        """
+
+        # Render history
+        messages = zip(st.session_state[PROMPT_LABEL], st.session_state[RESPONSE_LABEL])
+        for prompt, (response, on_render) in messages:
+            with st.chat_message("user"):
+                st.markdown(prompt)
+            with st.chat_message("assistant"):
+                on_render(response)
+
+        # Compute prompt
+        if prompt:= st.chat_input(placeholder="Ask Me Anything", key='chat_text_input'):
+            st.session_state[PROMPT_LABEL].append(prompt)
+            (response, on_render) = process_prompt(prompt, *args)
+            st.session_state[RESPONSE_LABEL].append((response, on_render))
+
+            with st.chat_message("user"):
+                st.markdown(prompt)
+
+            with st.chat_message("assistant"):
+                on_render(response)
+