Update app.py

app.py

@@ -54,28 +54,17 @@ model = GPT2LMHeadModel.from_pretrained(model_name)
 
 
 
-# Set the title of the web app
-st.title("Simple Chatbot")
 
-    # Initialize session state to store chat history
-if 'chat_history' not in st.session_state:
-    st.session_state.chat_history = []
+prompt = st.text_input("Say Something!", key=None, max_chars=None, disabled=False)
 
-    # Chat input
-user_input = st.chat_input(placeholder="Say Something!", key="chat_input")
-
-    # Check if there is an input
-if user_input:
-        # Generate a response based on the user input
-    response = generate_response(user_input)
+if prompt:
+    with st.container():
+        # Displaying the user's input question.
+        st.markdown(prompt)
         
-        # Update chat history with the user input and bot response
-    st.session_state.chat_history.append(("You", user_input))
-    st.session_state.chat_history.append(("Bot", response))
-
-        # Display chat history
-    for author, text in st.session_state.chat_history:
-        st.chat_message(text=text, author=author)
+        # Generating and displaying the response.
+        response = generate_response(prompt)
+        st.markdown(generate_response(prompt))