clean: rename callback && format

app.py

@@ -59,8 +59,8 @@ def initialize_session_state():
         )
 
 
-def on_click_callback():
-    with st.spinner('Wait for it...'):
+def send_message_callback():
+    with st.spinner("Wait for it..."):
         with get_openai_callback() as cb:
             human_prompt = st.session_state.human_prompt.strip()
             if len(human_prompt) == 0:
@@ -69,7 +69,9 @@ def on_click_callback():
             st.session_state.history.append(Message("human", human_prompt))
             st.session_state.history.append(
                 Message(
-                    "ai", llm_response["answer"], documents=llm_response["source_documents"]
+                    "ai",
+                    llm_response["answer"],
+                    documents=llm_response["source_documents"],
                 )
             )
             st.session_state.token_count += cb.total_tokens
@@ -134,7 +136,7 @@ with chat_column:
         cols[1].form_submit_button(
             "Submit",
             type="primary",
-            on_click=on_click_callback,
+            on_click=send_message_callback,
         )
 
     if st.session_state.token_count == 0: