Update app.py

app.py

@@ -44,38 +44,38 @@ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)
 
 # Chat loop
-print("Chatbot is ready. Type 'quit' to exit.")
-while True:
-    user_input = input("You: ")
-    if user_input.lower() == "quit":
-        break
-    response = generate_response(user_input)
-    print("Chatbot:", response)
+#print("Chatbot is ready. Type 'quit' to exit.")
+#while True:
+    #user_input = input("You: ")
+    #if user_input.lower() == "quit":
+        #break
+    #response = generate_response(user_input)
+    #print("Chatbot:", response)
 
 
 
-st.title("Simple Streamlit Chatbot")
+# Set the title of the web app
+st.title("Simple Chatbot")
 
-    # User input text box
-    user_input = st.text_input("You: ", key="user_input")
+    # Initialize session state to store chat history
+if 'chat_history' not in st.session_state:
+        st.session_state.chat_history = []
 
-    # Button to send the message
-    if st.button("Send"):
-        # Generating a response
-        response = get_response(user_input)
-        
-        # Displaying the conversation
-        # Here, we use st.session_state to keep track of the conversation
-        if 'conversation' not in st.session_state:
-            st.session_state.conversation = []
+    # Chat input
+user_input = st.chat_input("You", 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)
         
-        # Append the user input and bot response to the conversation
-        st.session_state.conversation.append("You: " + user_input)
-        st.session_state.conversation.append("Bot: " + response)
+        # 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 each line in the conversation
-        for line in st.session_state.conversation:
-            st.text(line)
+        # Display chat history
+        for author, text in st.session_state.chat_history:
+            st.chat_message(text=text, author=author)