using text input

app.py

@@ -164,5 +164,6 @@ What trades did client {client} do in May 2022
 with st.sidebar:
     st.markdown(sidebar_text)
 
+prompt = chat.get_promt("Ask IDF Anything")
 with st.container():
-    chat.get_input(get_response, llm)
+    chat.get_input(prompt, get_response, llm)

chat.py

@@ -1,12 +1,15 @@
+from sqlalchemy import label
 import streamlit as st
 from typing import Callable
 
 
 
 RESPONSE_LABEL = 'chat_response'
-PROMPT_LABEL = 'chat_user_input'
+PROMPT_LABEL = 'chat_prompt'
+TEXT_INPUT_LABEL = "chat_input"
 
 class Chat:
+
     def __init__(self): 
         if RESPONSE_LABEL not in st.session_state:
             st.session_state[RESPONSE_LABEL] = []
@@ -14,7 +17,18 @@ class Chat:
         if PROMPT_LABEL not in st.session_state:
             st.session_state[PROMPT_LABEL] = []
 
-    def get_input(self, process_prompt: Callable, *args):
+        if TEXT_INPUT_LABEL not in st.session_state:
+            st.session_state[TEXT_INPUT_LABEL] = ''
+
+    def get_promt(self, placeholder: str):
+        return st.text_input(
+            label="ChatIDF",
+            placeholder=placeholder,
+            key="chat_widget",
+        )
+        # return st.chat_input(placeholder=placeholder)
+
+    def get_input(self, prompt: str, process_prompt: Callable, *args):
         """
         process_prompt(promt: str, *args) -> tuple(Any, Callable)
             callback to process the chat promt, it takes the promt for input
@@ -30,7 +44,7 @@ class Chat:
                 on_render(response)
 
         # Compute prompt
-        if prompt:= st.chat_input(placeholder="Ask IDF Anything"):
+        if prompt:
             st.session_state[PROMPT_LABEL].append(prompt)
             (response, on_render) = process_prompt(prompt, *args)
             st.session_state[RESPONSE_LABEL].append((response, on_render))