File tree Expand file tree Collapse file tree 1 file changed +10
-1
lines changed
recipes/natural_language_processing/chatbot/app Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Original file line number Diff line number Diff line change @@ -46,10 +46,18 @@ def get_models():
46
46
with st .spinner ("Checking Model Service Availability..." ):
47
47
server = checking_model_service ()
48
48
49
+ def enableInput ():
50
+ st .session_state ["input_disabled" ] = False
51
+
52
+ def disableInput ():
53
+ st .session_state ["input_disabled" ] = True
54
+
49
55
st .title ("💬 Chatbot" )
50
56
if "messages" not in st .session_state :
51
57
st .session_state ["messages" ] = [{"role" : "assistant" ,
52
58
"content" : "How can I help you?" }]
59
+ if "input_disabled" not in st .session_state :
60
+ enableInput ()
53
61
54
62
for msg in st .session_state .messages :
55
63
st .chat_message (msg ["role" ]).write (msg ["content" ])
@@ -86,10 +94,11 @@ def memory():
86
94
verbose = False ,
87
95
memory = memory ())
88
96
89
- if prompt := st .chat_input ():
97
+ if prompt := st .chat_input (disabled = st . session_state [ "input_disabled" ], on_submit = disableInput ):
90
98
st .session_state .messages .append ({"role" : "user" , "content" : prompt })
91
99
st .chat_message ("user" ).markdown (prompt )
92
100
response = chain .invoke (prompt )
93
101
st .chat_message ("assistant" ).markdown (response ["text" ])
94
102
st .session_state .messages .append ({"role" : "assistant" , "content" : response ["text" ]})
103
+ enableInput ()
95
104
st .rerun ()
You can’t perform that action at this time.
0 commit comments