Update app.py

app.py

@@ -278,12 +278,16 @@ if __name__=="__main__":
 
     user_question = st.chat_input('What do you want to ask ..')
 
-    st.chat_message("AI").write(" ")
     
     # streamlit chat
     if user_question is not None and user_question!="":
         with st.chat_message("Human", avatar="🧔🏻"):
             st.write(user_question)
+            
+        if response:
+            # to empty response container after first pass
+            st.chat_message("AI", avatar="🏛️").write(" ")
+            
         response = generate_response(user_question,  model, all_templates)   
         with st.chat_message("AI", avatar="🏛️"):
             st.write(response)