Update app.py

app.py

@@ -1,29 +1,17 @@
-import streamlit as st
 from transformers import pipeline, Conversation
+import gradio as gr
 
 chatbot = pipeline(model="facebook/blenderbot-400M-distill")
 
+message_list = []
+response_list = []
+
 def vanilla_chatbot(message, history):
-    conversation = Conversation(text=message, past_user_inputs=history)
+    conversation = Conversation(text=message, past_user_inputs=message_list, generated_responses=response_list)
     conversation = chatbot(conversation)
 
     return conversation.generated_responses[-1]
 
-# Replace the gradio interface with Streamlit components
-st.title("Streamlit Chatbot")
-st.write("Enter text to start chatting.")
-
-# Add a text input and button for the user to submit their message
-user_input = st.text_input("")
-submit_button = st.button("Submit")
-
-# Store the chat history in a list
-history = []
+demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Vanilla Chatbot", description="Enter text to start chatting.")
 
-# Handle the user's message submission
-if submit_button:
-    message = user_input
-    response = vanilla_chatbot(message, history)
-    st.write(f"Bot: {response}")
-    history.append(message)
-    history.append(response)
+demo_chatbot.launch()