Update app.py

app.py

@@ -26,11 +26,11 @@ def interactive():
     print("Hi! I am a simple AI chatbot built using Hugging Face and Streamlit.")
     query = ""
     while query != "quit":
-        if 'current_query' not in st.session_state or st.session_state['current_query'] != query:
+        if ('current_query' not in st.session_state) or (st.session_state['current_query'] != query):
             st.session_state['current_query'] = query
-            st.text_input("Ask me something or type 'quit' to exit:", "", key='my_input', on_change=update_displayed_query)
-            
-        if 'shown_query' not in st.session_state or st.session_state['shown_query'] != query:
+            st.text_input("Ask me something or type 'quit' to exit:", "", key='my_input', on_change=lambda x: update_displayed_query(x["new"]))
+
+        if ('shown_query' not in st.session_state) or (st.session_state['shown_query'] != query):
             st.session_state['shown_query'] = query
             if len(query) > 0 and query != 'quit':
                 try:
@@ -40,6 +40,8 @@ def interactive():
                 except Exception as e:
                     st.write(f"Error occurred: {str(e)}")
 
+        query = st.session_state.get('current_query', '')
+
 if __name__ == "__main__":
     st.set_page_config(layout="wide")
     st.title('AI Chatbot Built Using Hugging Face and Streamlit')